@Test public void addDistributionGroupIdToLogsTest() { String distributionGroupId = UUID.randomUUID().toString(); DistributeInfoTracker distributeInfoTracker = new DistributeInfoTracker(distributionGroupId); /* Distribution group ID field is added to logs. */ { Log log = newLog(); distributeInfoTracker.onPreparingLog(log, TEST_GROUP); assertNotNull(log.getDistributionGroupId()); assertEquals(distributionGroupId, log.getDistributionGroupId()); } }
@Test public void setNewDistributionGroupIdTest() { DistributeInfoTracker distributeInfoTracker = new DistributeInfoTracker(null); /* No distribution group ID field is added to logs because value is null. */ { Log log = newLog(); distributeInfoTracker.onPreparingLog(log, TEST_GROUP); assertNull(log.getDistributionGroupId()); } /* Distribution group ID field is added to logs after the value was fetched and saved. */ { String distributionGroupId = UUID.randomUUID().toString(); distributeInfoTracker.updateDistributionGroupId(distributionGroupId); Log log = newLog(); distributeInfoTracker.onPreparingLog(log, TEST_GROUP); assertEquals(distributionGroupId, log.getDistributionGroupId()); } /* No distribution group ID field is added after the value was removed. */ { distributeInfoTracker.removeDistributionGroupId(); Log log = newLog(); distributeInfoTracker.onPreparingLog(log, TEST_GROUP); assertNull(log.getDistributionGroupId()); } }