def test_add_create_with_result_dct_id_required_org_error(self, construct, execute): """Check command create when result has dct id key and organization is required but is not present """ execute.return_value = [{'id': 'foo', 'bar': 'bas'}] Base.command_requires_org = True with pytest.raises(CLIError): Base.create() assert 'create' == Base.command_sub construct.called_once_with({}) execute.called_once_with(construct.return_value, output_format='csv')
def test_add_create_with_empty_result(self, construct, execute): """Check command create when result is empty""" execute.return_value = [] assert execute.return_value == Base.create() assert 'create' == Base.command_sub construct.called_once_with({}) execute.called_once_with(construct.return_value, output_format='csv')
def test_add_create_with_result_dct_without_id(self, construct, execute, info): """Check command create when result has dct but dct hasn't id key""" execute.return_value = [{'not_id': 'foo'}] assert execute.return_value == Base.create() assert 'create' == Base.command_sub construct.called_once_with({}) execute.called_once_with(construct.return_value, output_format='csv') assert not info.called
def test_add_create_with_result_dct_without_id(self, construct, execute, info): """Check command create when result has dct but dct hasn't id key""" execute.return_value = [{'not_id': 'foo'}] self.assertEqual(execute.return_value, Base.create()) self.assertEqual('create', Base.command_sub) construct.called_once_with({}) execute.called_once_with(construct.return_value, output_format='csv') self.assertFalse(info.called)
def test_add_create_with_empty_result(self, construct, execute): """Check command create when result is empty""" execute.return_value = [] self.assertEqual( execute.return_value, Base.create() ) self.assertEqual('create', Base.command_sub) construct.called_once_with({}) execute.called_once_with(construct.return_value, output_format='csv')
def test_add_create_with_result_dct_with_id_required_org(self, construct, execute, info): """Check command create when result has dct id key and organization is required """ execute.return_value = [{'id': 'foo', 'bar': 'bas'}] Base.command_requires_org = True assert execute.return_value == Base.create({'organization-id': 'org-id'}) assert 'create' == Base.command_sub construct.called_once_with({}) execute.called_once_with(construct.return_value, output_format='csv') info.called_once_with({'id': 'foo', 'organization-id': 'org-id'})
def test_add_create_with_result_dct_without_id( self, construct, execute, info): """Check command create when result has dct but dct hasn't id key""" execute.return_value = [{'not_id': 'foo'}] self.assertEqual( execute.return_value, Base.create() ) self.assertEqual('create', Base.command_sub) construct.called_once_with({}) execute.called_once_with(construct.return_value, output_format='csv') self.assertFalse(info.called)
def test_add_create_with_result_dct_with_id_not_required_org( self, construct, execute, info): """Check command create when result has dct id key and organization is not required """ execute.return_value = [{'id': 'foo', 'bar': 'bas'}] Base.command_requires_org = False self.assertEqual(execute.return_value, Base.create()) self.assertEqual('create', Base.command_sub) construct.called_once_with({}) execute.called_once_with(construct.return_value, output_format='csv') info.called_once_with({'id': 'foo'})
def test_add_create_with_result_dct_with_id_required_org( self, construct, execute, info): """Check command create when result has dct id key and organization is required """ execute.return_value = [{'id': 'foo', 'bar': 'bas'}] Base.command_requires_org = True self.assertEqual( execute.return_value, Base.create({'organization-id': 'org-id'}) ) self.assertEqual('create', Base.command_sub) construct.called_once_with({}) execute.called_once_with(construct.return_value, output_format='csv') info.called_once_with({'id': 'foo', 'organization-id': 'org-id'})