diff options
Diffstat (limited to 'src/test/java/com/orbekk/protobuf/TimeoutManagerTest.java')
-rw-r--r-- | src/test/java/com/orbekk/protobuf/TimeoutManagerTest.java | 33 |
1 files changed, 33 insertions, 0 deletions
diff --git a/src/test/java/com/orbekk/protobuf/TimeoutManagerTest.java b/src/test/java/com/orbekk/protobuf/TimeoutManagerTest.java new file mode 100644 index 0000000..9e4896a --- /dev/null +++ b/src/test/java/com/orbekk/protobuf/TimeoutManagerTest.java @@ -0,0 +1,33 @@ +package com.orbekk.protobuf; + +import static org.mockito.Mockito.*; + +import java.io.Closeable; + +import org.junit.Before; +import org.junit.Test; + +import com.orbekk.protobuf.TimeoutManager.Environment; + +public class TimeoutManagerTest { + Closeable closeable = mock(Closeable.class); + Environment environment = mock(Environment.class); + TimeoutManager timeoutManager = new TimeoutManager(environment); + + @Before public void setUp() { + } + + @Test public void closesExpiredEntries() throws Exception { + when(environment.currentTimeMillis()).thenReturn(1000l); + timeoutManager.addEntry(1000l, closeable); + timeoutManager.performWork(); + verify(closeable).close(); + } + + @Test public void doesNotCloseUnexpiredEntry() throws Exception { + when(environment.currentTimeMillis()).thenReturn(123456l); + timeoutManager.addEntry(123457l, closeable); + timeoutManager.performWork(); + verify(closeable, never()).close(); + } +} |