Add: TokenGetter

1 files changed, 15 insertions, 1 deletions

diff --git a/ick2/workapi_tests.py b/ick2/workapi_tests.py
index 72bcaf1..0ac3a86 100644
--- a/ick2/workapi_tests.py
+++ b/ick2/workapi_tests.py
@@ -36,7 +36,9 @@ class WorkAPITests(unittest.TestCase):
             ],
         }
 
+        getter = DummyTokenGetter()
         pipeapi = ick2.PipelineAPI(self.state)
+        pipeapi.set_token_getter(getter)
         pipeapi.create(pipeline)
 
         project = {
@@ -47,6 +49,7 @@ class WorkAPITests(unittest.TestCase):
             'pipelines': ['build'],
         }
         api = ick2.ProjectAPI(self.state)
+        api.set_token_getter(getter)
         api.create(project)
         return api
 
@@ -57,12 +60,17 @@ class WorkAPITests(unittest.TestCase):
         self.claims = {
             'aud': 'asterix',
         }
+        getter = DummyTokenGetter()
         api = ick2.WorkerAPI(self.state)
+        api.set_token_getter(getter)
         api.create(worker, claims=self.claims)
         return api
 
     def create_work_api(self):
-        return ick2.WorkAPI(self.state)
+        getter = DummyTokenGetter()
+        api = ick2.WorkAPI(self.state)
+        api.set_token_getter(getter)
+        return api
 
     def test_worker_gets_no_work_when_no_builds_have_been_triggered(self):
         self.create_project_api()
@@ -287,3 +295,9 @@ class WorkAPITests(unittest.TestCase):
 
         # Ask for work again.
         self.assertEqual(work.get_work(claims=self.claims), {})
+
+
+class DummyTokenGetter:
+
+    def get_token(self):
+        return 'DUMMY.TOKEN'