RootResource should return 503 during startup (#699)

Return 503 code while RootResource is in MODE_STARTUP

1 files changed, 16 insertions, 0 deletions

diff --git a/service/test/unit/resources/test_root_resource.py b/service/test/unit/resources/test_root_resource.py
index f2d7a951..b90e89eb 100644
--- a/service/test/unit/resources/test_root_resource.py
+++ b/service/test/unit/resources/test_root_resource.py
@@ -68,6 +68,21 @@ class TestRootResource(unittest.TestCase):
         self.root_resource._mode = MODE_RUNNING
         self._test_should_renew_xsrf_cookie()
 
+    def test_should_unavailable_child_resource_during_startup(self):
+        self.root_resource._mode = MODE_STARTUP
+
+        request = DummyRequest(['/child'])
+        request.getCookie = MagicMock(return_value='irrelevant -- stubbed')
+
+        d = self.web.get(request)
+
+        def assert_unavailable(_):
+            self.assertEqual(503, request.responseCode)
+            self.assertEqual("Service Unavailable", request.written[0])
+
+        d.addCallback(assert_unavailable)
+        return d
+
     def _mock_ajax_csrf(self, request, csrf_token):
         request.requestHeaders.setRawHeaders('x-requested-with', ['XMLHttpRequest'])
         request.requestHeaders.setRawHeaders('x-xsrf-token', [csrf_token])
@@ -92,6 +107,7 @@ class TestRootResource(unittest.TestCase):
 
         request.getCookie = MagicMock(return_value='irrelevant -- stubbed')
         self.root_resource._child_resources.add('features', FeaturesResource())
+        self.root_resource._mode = MODE_RUNNING
 
         d = self.web.get(request)