1 files changed, 59 insertions, 13 deletions

diff --git a/common-logging/src/test/java/org/onap/policy/common/logging/flexlogger/PropertyUtilTest.java b/common-logging/src/test/java/org/onap/policy/common/logging/flexlogger/PropertyUtilTest.java
index d97df4ce..b4fcfc18 100644
--- a/common-logging/src/test/java/org/onap/policy/common/logging/flexlogger/PropertyUtilTest.java
+++ b/common-logging/src/test/java/org/onap/policy/common/logging/flexlogger/PropertyUtilTest.java
@@ -2,7 +2,7 @@
  * ============LICENSE_START=======================================================
  * ONAP-Logging
  * ================================================================================
- * Copyright (C) 2018 Ericsson. All rights reserved.
+ * Copyright (C) 2018 Ericsson, AT&T. All rights reserved.
  * ================================================================================
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -21,29 +21,70 @@
 package org.onap.policy.common.logging.flexlogger;
 
 import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
 import static org.junit.Assert.assertTrue;
-
+import static org.mockito.Matchers.any;
+import static org.mockito.Matchers.anyLong;
+import static org.mockito.Mockito.doAnswer;
+import static org.mockito.Mockito.mock;
 import java.io.File;
 import java.io.FileOutputStream;
 import java.io.IOException;
 import java.util.Properties;
 import java.util.Set;
-
+import java.util.Timer;
+import java.util.TimerTask;
+import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.TimeUnit;
 import org.junit.After;
+import org.junit.AfterClass;
 import org.junit.Before;
+import org.junit.BeforeClass;
 import org.junit.Test;
 import org.onap.policy.common.logging.flexlogger.PropertyUtil.Listener;
+import org.powermock.reflect.Whitebox;
 
 public class PropertyUtilTest {
 
+    /**
+     * 
+     */
+    private static final String TIMER_FIELD = "timer";
     private static final File FILE = new File("target/test.properties");
-    private TestListener testListener = new TestListener();
+    private static Timer saveTimer;
+    
+    private TimerTask task;
+    private Timer timer;
+    private TestListener testListener;
+    
+    @BeforeClass
+    public static void setUpBeforeClass() {
+        saveTimer = Whitebox.getInternalState(PropertyUtil.LazyHolder.class, TIMER_FIELD);
+        
+    }
+    
+    @AfterClass
+    public static void tearDownAfterClass() {
+        Whitebox.setInternalState(PropertyUtil.LazyHolder.class, TIMER_FIELD, saveTimer);
+        
+    }
 
     /**
      * Perform test case set up.
      */
     @Before
     public void setUp() throws IOException {
+        task = null;
+        timer = mock(Timer.class);
+        Whitebox.setInternalState(PropertyUtil.LazyHolder.class, TIMER_FIELD, timer);
+        
+        doAnswer(args -> {
+            task = args.getArgumentAt(0, TimerTask.class);
+            return null;
+        }).when(timer).schedule(any(TimerTask.class), anyLong(), anyLong());
+        
+        testListener = new TestListener();
+        
         FileOutputStream fileOutputStream = new FileOutputStream(FILE);
         Properties properties = new Properties();
         properties.put("testProperty", "testValue");
@@ -56,6 +97,11 @@ public class PropertyUtilTest {
         PropertyUtil.stopListening(FILE, testListener);
         FILE.delete();
     }
+    
+    @Test
+    public void testTimer() {
+        assertNotNull(saveTimer);
+    }
 
     @Test
     public void testGetProperties() throws IOException {
@@ -78,6 +124,8 @@ public class PropertyUtilTest {
         newProperties.put("testProperty", "testValueNew");
         newProperties.store(fileOutputStream, "");
 
+        // fire task and verify that it notifies the listener
+        task.run();
         assertTrue(testListener.isPropertiesChangedInvoked());
 
     }
@@ -104,23 +152,21 @@ public class PropertyUtilTest {
         assertEquals("testValueNew", readProperties.getProperty("testProperty"));
     }
 
+    /**
+     * The {@link #propertiesChanged(Properties, Set)} method is invoked via a background
+     * thread, thus we have to use a latch to wait for it to be invoked.
+     */
     private class TestListener implements Listener {
 
-        boolean propertiesChangedInvoked = false;
+        private CountDownLatch latch = new CountDownLatch(1);
 
         @Override
         public void propertiesChanged(Properties properties, Set<String> changedKeys) {
-            propertiesChangedInvoked = true;
+            latch.countDown();
         }
 
         public boolean isPropertiesChangedInvoked() throws InterruptedException {
-            for (int i = 0; i < 20; i++) {
-                if (propertiesChangedInvoked) {
-                    return true;
-                }
-                Thread.sleep(1000);
-            }
-            return false;
+            return latch.await(5, TimeUnit.SECONDS);
         }
     }