diff options
Diffstat (limited to 'ick2/client_tests.py')
-rw-r--r-- | ick2/client_tests.py | 14 |
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): |