Change: worker-manager gets APT server from controller

1 files changed, 14 insertions, 0 deletions

diff --git a/ick2/client_tests.py b/ick2/client_tests.py
index adba2dc..24b9227 100644
--- a/ick2/client_tests.py
+++ b/ick2/client_tests.py
@@ -174,6 +174,20 @@ class ControllerClientTests(unittest.TestCase):
         self.session.response = FakeResponse(200)
         self.assertEqual(self.controller.report_work(work), None)
 
+    def test_get_apt_server_raises_exception_on_error(self):
+        self.session.response = FakeResponse(400)
+        with self.assertRaises(ick2.HttpError):
+            self.controller.get_apt_server()
+
+    def test_get_apt_server_succeeds(self):
+        server = 'apt.example.com'
+        version = {
+            'apt-server': server,
+        }
+        self.session.response = FakeResponse(
+            200, body=json.dumps(version), content_type=json_type)
+        self.assertEqual(self.controller.get_apt_server(), server)
+
     def test_get_artifact_store_url_raises_exception_on_error(self):
         self.session.response = FakeResponse(400)
         with self.assertRaises(ick2.HttpError):