def test_get_does_not_eager_relations_when_no_results_are_returned(self):
flexmock(Builder)
builder = Builder(self.get_mock_query_builder())
builder.should_receive('get_models').with_args(['foo']).and_return(['bar'])
builder.should_receive('eager_load_relations').with_args(['bar']).and_return([])
builder.set_model(self.get_mock_model())
builder.get_model().new_collection = mock.MagicMock(return_value=Collection([]))
results = builder.get(['foo'])
self.assertEqual([], results.all())
builder.get_model().new_collection.assert_called_with([])
def test_first(self):
model = self.get_mock_model()
builder = Builder(self.get_mock_query_builder())
builder.set_model(model)
builder.take = mock.MagicMock(return_value=builder)
builder.get = mock.MagicMock(return_value=Collection(["bar"]))
result = builder.first()
self.assertEqual("bar", result)
builder.take.assert_called_once_with(1)
builder.get.assert_called_once_with(["*"])
def test_find_or_fail_with_many_raises_model_not_found_exception(self):
model = self.get_mock_model()
builder = Builder(self.get_mock_query_builder())
builder.set_model(model)
builder.get_query().where_in = mock.MagicMock()
builder.get = mock.MagicMock(return_value=Collection([1]))
self.assertRaises(ModelNotFound, builder.find_or_fail, [1, 2],
["column"])
builder.get_query().where_in.assert_called_once_with(
"foo_table.foo", [1, 2])
builder.get.assert_called_once_with(["column"])
def test_get_loads_models_and_hydrates_eager_relations(self):
flexmock(Builder)
builder = Builder(self.get_mock_query_builder())
builder.should_receive("get_models").with_args(["foo"
]).and_return(["bar"])
builder.should_receive("eager_load_relations").with_args(
["bar"]).and_return(["bar", "baz"])
builder.set_model(self.get_mock_model())
builder.get_model().new_collection = mock.MagicMock(
return_value=Collection(["bar", "baz"]))
results = builder.get(["foo"])
self.assertEqual(["bar", "baz"], results.all())
builder.get_model().new_collection.assert_called_with(["bar", "baz"])
def test_find_with_many(self):
model = self.get_mock_model()
builder = Builder(self.get_mock_query_builder())
builder.set_model(model)
builder.get_query().where_in = mock.MagicMock()
builder.get = mock.MagicMock(return_value="baz")
result = builder.find([1, 2], ["column"])
self.assertEqual("baz", result)
builder.get_query().where_in.assert_called_once_with(
"foo_table.foo", [1, 2])
builder.get.assert_called_once_with(["column"])
def test_chunk(self):
builder = Builder(self.get_mock_query_builder())
results = [['foo1', 'foo2'], ['foo3'], []]
builder.for_page = mock.MagicMock(return_value=builder)
builder.get = mock.MagicMock(side_effect=results)
i = 0
for result in builder.chunk(2):
self.assertEqual(result, results[i])
i += 1
builder.for_page.assert_has_calls(
[mock.call(1, 2),
mock.call(2, 2),
mock.call(3, 2)])
def test_chunk(self):
builder = Builder(self.get_mock_query_builder())
results = [['foo1', 'foo2'], ['foo3'], []]
builder.for_page = mock.MagicMock(return_value=builder)
builder.get = mock.MagicMock(side_effect=results)
i = 0
for result in builder.chunk(2):
self.assertEqual(result, results[i])
i += 1
builder.for_page.assert_has_calls([
mock.call(1, 2),
mock.call(2, 2),
mock.call(3, 2)
])
def test_first(self):
model = self.get_mock_model()
builder = Builder(self.get_mock_query_builder())
builder.set_model(model)
builder.take = mock.MagicMock(return_value=builder)
builder.get = mock.MagicMock(return_value=Collection(['bar']))
result = builder.first()
self.assertEqual('bar', result)
builder.take.assert_called_once_with(
1
)
builder.get.assert_called_once_with(
['*']
)
def test_find_with_many(self):
model = self.get_mock_model()
builder = Builder(self.get_mock_query_builder())
builder.set_model(model)
builder.get_query().where_in = mock.MagicMock()
builder.get = mock.MagicMock(return_value='baz')
result = builder.find([1, 2], ['column'])
self.assertEqual('baz', result)
builder.get_query().where_in.assert_called_once_with(
'foo_table.foo', [1, 2]
)
builder.get.assert_called_once_with(
['column']
)
def test_find_or_fail_with_many_raises_model_not_found_exception(self):
model = self.get_mock_model()
builder = Builder(self.get_mock_query_builder())
builder.set_model(model)
builder.get_query().where_in = mock.MagicMock()
builder.get = mock.MagicMock(return_value=Collection([1]))
self.assertRaises(
ModelNotFound,
builder.find_or_fail,
[1, 2],
['column']
)
builder.get_query().where_in.assert_called_once_with(
'foo_table.foo', [1, 2]
)
builder.get.assert_called_once_with(
['column']
)
