From 022eedce9ac35443f0ad9cfd487b9775551cd79b Mon Sep 17 00:00:00 2001
From: Oliver Palmer <oliverpalmer@opalmer.com>
Date: Thu, 5 Mar 2015 23:36:40 -0500
Subject: [PATCH] remaining tests for reannounce

---
 tests/test_agent/test_service.py | 113 ++++++++++++++++++++++++++++++-
 1 file changed, 112 insertions(+), 1 deletion(-)

diff --git a/tests/test_agent/test_service.py b/tests/test_agent/test_service.py
index 3d401c73..23f24f4a 100644
--- a/tests/test_agent/test_service.py
+++ b/tests/test_agent/test_service.py
@@ -909,8 +909,8 @@ def test_should_not_reannounce(self):
     @inlineCallbacks
     def test_reannounce_force(self):
         self.fake_api.code = OK
-
         agent = Agent()
+
         with nested(
             patch.object(agent, "should_reannounce", return_value=False),
             patch.object(agent.reannouce_lock, "acquire"),
@@ -939,3 +939,114 @@ def test_ok(self):
         self.assertEqual(result, self.normal_result)
         acquire_lock.assert_called_once()
         release_lock.assert_called_once()
+
+    @inlineCallbacks
+    def test_internal_server_error_shutting_down(self):
+        self.fake_api.code = INTERNAL_SERVER_ERROR
+        agent = Agent()
+        agent.shutting_down = True
+
+        with nested(
+            patch.object(agent.reannouce_lock, "acquire"),
+            patch.object(agent.reannouce_lock, "release"),
+            patch.object(svclog, "warning")
+        ) as (acquire_lock, release_lock, warning_log):
+            result = yield agent.reannounce(force=True)
+
+        self.assertEqual(result, self.normal_result)
+        acquire_lock.assert_called_once()
+        release_lock.assert_called_once()
+        warning_log.assert_called_once_with(
+            "Could not announce to master. Not retrying because of pending "
+            "shutdown."
+        )
+
+    @inlineCallbacks
+    def test_internal_server_error_retry(self):
+        self.fake_api.code = INTERNAL_SERVER_ERROR
+        agent = Agent()
+        agent.shutting_down = False
+
+        def shutdown():
+            agent.shutting_down = True
+
+        reactor.callLater(3, shutdown)
+
+        with nested(
+            patch.object(agent.reannouce_lock, "acquire"),
+            patch.object(agent.reannouce_lock, "release"),
+            patch.object(svclog, "warning")
+        ) as (acquire_lock, release_lock, warning_log):
+            result = yield agent.reannounce(force=True)
+
+        self.assertEqual(result, self.normal_result)
+        acquire_lock.assert_called_once()
+        release_lock.assert_called_once()
+        warning_log.assert_any_call(
+            "Could not announce self to the master server, internal server "
+            "error: %s.  Retrying in %s seconds.", self.normal_result, 1.0
+        )
+        self.assertGreaterEqual(warning_log.call_count, 2)
+
+    @inlineCallbacks
+    def test_not_found(self):
+        self.fake_api.code = NOT_FOUND
+        agent = Agent()
+
+        post_deferred = Deferred()
+        post_deferred.callback(None)
+        with nested(
+            patch.object(agent.reannouce_lock, "acquire"),
+            patch.object(agent.reannouce_lock, "release"),
+            patch.object(
+                    agent, "post_agent_to_master", return_value=post_deferred),
+        ) as (acquire_lock, release_lock, post_agent):
+            result = yield agent.reannounce(force=True)
+
+        self.assertEqual(result, self.normal_result)
+        acquire_lock.assert_called_once()
+        release_lock.assert_called_once()
+        post_agent.assert_called_once()
+
+    @inlineCallbacks
+    def test_bad_request(self):
+        self.fake_api.code = BAD_REQUEST
+        agent = Agent()
+
+        with nested(
+            patch.object(agent.reannouce_lock, "acquire"),
+            patch.object(agent.reannouce_lock, "release"),
+            patch.object(svclog, "error")
+        ) as (acquire_lock, release_lock, error_log):
+            result = yield agent.reannounce(force=True)
+
+        self.assertEqual(result, self.normal_result)
+        acquire_lock.assert_called_once()
+        release_lock.assert_called_once()
+        error_log.assert_called_once_with(
+            "Failed to announce self to the master, bad "
+            "request: %s.  This request will not be retried.",
+            self.normal_result
+        )
+
+    @inlineCallbacks
+    def test_other(self):
+        self.fake_api.code = 42
+        agent = Agent()
+
+        with nested(
+            patch.object(agent.reannouce_lock, "acquire"),
+            patch.object(agent.reannouce_lock, "release"),
+            patch.object(svclog, "error")
+        ) as (acquire_lock, release_lock, error_log):
+            result = yield agent.reannounce(force=True)
+
+        self.assertEqual(result, self.normal_result)
+        acquire_lock.assert_called_once()
+        release_lock.assert_called_once()
+        error_log.assert_called_once_with(
+            "Unhandled error when posting self to the "
+            "master: %s (code: %s).  This request will not be "
+            "retried.",
+            self.normal_result, 42
+        )
\ No newline at end of file