def test_setup_data_model_no_schema(self, mock_get, mock_schema):
mock_get.return_value = None
mock_schema.return_value = None
with pytest.raises(Exception) as ex:
generators.setup_data_model('foo', 'Bar')
assert str(ex.value) == 'Missing schema for model `Bar`'
mock_get.assert_called_once_with('Bar')
mock_schema.assert_called_once_with('foo')
def test_setup_data_model_success(self, mock_get, mock_gen, mock_schema):
mock_get.return_value = None
mock_schema.return_value = {'field1': 'val1'}
model = generators.setup_data_model('foo', 'Bar')
mock_get.assert_called_once_with('Bar')
mock_schema.assert_called_once_with('foo')
mock_gen.assert_called_once_with(
schema={'field1': 'val1'},
model_name='Bar',
raml_resource='foo')
assert model == mock_gen()
def test_setup_data_model_existing_model(self, mock_get):
mock_get.return_value = 1
model, auth_model = generators.setup_data_model('foo', 'Bar')
assert not auth_model
assert model == 1
mock_get.assert_called_once_with('Bar')
