def test_main_registerci(self):
response = mock.Mock()
response.json.return_value = {
'_items': [{
'id': 'id', 'name': 'name',
'created_at': 'created_at', 'updated_at': 'updated_at'}],
'id': 'a'}
response.status_code = 201
session = mock.Mock()
session.get.return_value = response
session.post.return_value = response
dciclient.client.requests.Session = mock.Mock(return_value=session)
setattr(dciclient, 'print', self._catch_print_call)
dciclient.main(args=['register-remoteci', '--name', 'bob'])
self.assertEqual([
"args: ['register-remoteci', '--name', 'bob']",
"RemoteCI 'bob' created successfully."], self.print_call)
def test_main_auto(self):
response = mock.Mock()
response.json.return_value = {
'id': 'bob',
'job_id': 'bobo',
'data': {'ksgen_args': {}},
'_status': 'OK'
}
response.status_code = 201
session = mock.Mock()
session.post.return_value = response
session.get.return_value = response
dciclient.client.requests.Session = mock.Mock(return_value=session)
popenobj = mock.Mock()
popenobj.returncode = 0
dciclient.client.subprocess = mock.Mock()
dciclient.client.subprocess.Popen.return_value = popenobj
dciclient.main(args=['auto', 'some-remoteci-id'])
self.assertEqual(self.print_call, [])
def test_main_list(self):
response = mock.Mock()
response.json.return_value = {'_items': [
{'id': 'id', 'name': 'name',
'created_at': 'created_at', 'updated_at': 'updated_at'}]}
session = mock.Mock()
session.get.return_value = response
dciclient.client.requests.Session = mock.Mock(
return_value=session)
setattr(dciclient, 'print', self._catch_print_call)
dciclient.main(args=['list', '--remotecis'])
self.assertEqual([
"args: ['list', '--remotecis']",
'+------------+------+------------+------------+\n'
'| identifier | name | created_at | updated_at |\n'
'+------------+------+------------+------------+\n'
'| id | name | created_at | updated_at |\n'
'+------------+------+------------+------------+'],
self.print_call)
