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();
+    }
+}