summaryrefslogtreecommitdiff
path: root/ick2/workapi_tests.py
diff options
context:
space:
mode:
Diffstat (limited to 'ick2/workapi_tests.py')
-rw-r--r--ick2/workapi_tests.py16
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'