Change: worker-manager gets APT server from controller

1 files changed, 5 insertions, 0 deletions

diff --git a/ick2/versionapi_tests.py b/ick2/versionapi_tests.py
index 76879ed..6dc8f54 100644
--- a/ick2/versionapi_tests.py
+++ b/ick2/versionapi_tests.py
@@ -22,18 +22,23 @@ import ick2
 class VersionAPITests(unittest.TestCase):
 
     def test_returns_version_correcly(self):
+        apt = 'apt.example.com'
         bloburl = 'https://blobs.example.com'
         idpurl = 'https://idp.example.com'
         notifyurl = 'https://notify.example.com'
+
         api = ick2.VersionAPI(None)
+        api.set_apt_server(apt)
         api.set_artifact_store_url(bloburl)
         api.set_auth_url(idpurl)
         api.set_notify_url(notifyurl)
+
         response = api.get_version()
         self.assertEqual(
             response,
             {
                 'version': ick2.__version__,
+                'apt_server': apt,
                 'artifact_store': bloburl,
                 'auth_url': idpurl,
                 'notify_url': notifyurl,