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)