def test_get_tasks(self):
"""
test get_tasks
"""
expected_ret = [(u'foo', u'bar')]
with patch('taskcluster.Index') as MockClass:
instance = MockClass.return_value
instance.listTasks.return_value = {u'tasks': [{u'data': {u'test': u'data'}, u'expires': u'2015-09-09T19:19:15.879Z', u'namespace': u'foo', u'rank': 1, u'taskId': u'bar'}]}
f = TaskFinder()
ret = f.get_tasks('foo')
self.assertEqual(ret, expected_ret)
def test_get_taskid_by_namespace(self):
"""
test get_taskid_by_namespace
"""
expected_ret = u'foobar_taskid'
with patch('taskcluster.Index') as MockClass:
instance = MockClass.return_value
instance.findTask.return_value = {u'data': {}, u'expires': u'2016-08-30T20:26:07.196Z', u'namespace': u'foo', u'rank': 999, u'taskId': u'foobar_taskid'}
f = TaskFinder()
ret = f.get_taskid_by_namespace('foo')
self.assertEqual(ret, expected_ret)
def test_get_namespaces(self):
"""
test get_namespaces
"""
expected_ret = [u'foo.test', u'foo.v1', u'foo.bar']
with patch('taskcluster.Index') as MockClass:
instance = MockClass.return_value
instance.listNamespaces.return_value = {u'namespaces': [{u'expires': u'2016-03-30T00:00:00.000Z', u'namespace': u'foo.test', u'name': u'test'}, {u'expires': u'2016-08-31T00:00:00.000Z', u'namespace': u'foo.v1', u'name': u'v1'}, {u'expires': u'2016-08-31T00:00:00.000Z', u'namespace': u'foo.bar', u'name': u'bar'}]}
f = TaskFinder()
ret = f.get_namespaces('foo')
self.assertEqual(ret, expected_ret)
def test_get_namespaces_and_tasks(self):
"""
test get_namespaces_and_tasks
"""
expected_ret = {'node': 'foo', 'tasks': [(u'foo', u'bar')], 'namespaces': [u'foo.test', u'foo.v1', u'foo.bar']}
with patch('taskcluster.Index') as MockClass:
instance = MockClass.return_value
instance.listNamespaces.return_value = {u'namespaces': [{u'expires': u'2016-03-30T00:00:00.000Z', u'namespace': u'foo.test', u'name': u'test'}, {u'expires': u'2016-08-31T00:00:00.000Z', u'namespace': u'foo.v1', u'name': u'v1'}, {u'expires': u'2016-08-31T00:00:00.000Z', u'namespace': u'foo.bar', u'name': u'bar'}]}
instance.listTasks.return_value = {u'tasks': [{u'data': {u'test': u'data'}, u'expires': u'2015-09-09T19:19:15.879Z', u'namespace': u'foo', u'rank': 1, u'taskId': u'bar'}]}
f = TaskFinder()
ret = f.get_namespaces_and_tasks('foo')
self.assertEqual(ret, expected_ret)
def test_get_parent_namespace(self):
"""
test get_parent_namespace
"""
f = TaskFinder()
expected_ret = 'foo'
ret = f.get_parent_namespace('foo.bar')
self.assertEqual(ret, expected_ret)
expected_ret = ''
ret = f.get_parent_namespace('root')
self.assertEqual(ret, expected_ret)
expected_ret = 'root.foo.bar.test_v1.moz'
ret = f.get_parent_namespace('root.foo.bar.test_v1.moz.node')
self.assertEqual(ret, expected_ret)
