adding tests for callback_shutting_down_changed

pyfarm · Feb 26, 2015 · d1e0b87 · d1e0b87
1 parent 9536c6c
commit d1e0b87
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/tests/test_agent/test_service.py b/tests/test_agent/test_service.py
@@ -141,6 +141,24 @@ def fake_schedule_call(*args, **kwargs):
         self.assertEqual(
             schedule_call_args[0][0][1], agent.reannounce)
 
+    def test_callback_shutting_down_true(self):
+        config["agent_shutdown_timeout"] = 4242
+        agent = Agent()
+        self.assertIsNone(agent.shutdown_timeout)
+        agent.shutting_down = True
+        self.assertIsInstance(agent.shutdown_timeout, datetime)
+        expected = timedelta(
+            seconds=config["agent_shutdown_timeout"]) + datetime.utcnow()
+        self.assertDateAlmostEqual(agent.shutdown_timeout, expected)
+
+    def test_callback_shutting_down_false(self):
+        config["agent_shutdown_timeout"] = 4242
+        agent = Agent()
+        self.assertIsNone(agent.shutdown_timeout)
+        agent.shutting_down = True
+        agent.shutting_down = False
+        self.assertIsNone(agent.shutdown_timeout)
+
     def test_shutting_down_getter(self):
         config["shutting_down"] = 42
         agent = Agent()