def test_select(self, add_models_from_fields, add_fields):
self._query = PostgreSQLQuery(0)
query = self.query.select()
add_fields.return_value = None
self.assertTrue(add_fields.called)
self.assertTrue(add_models_from_fields.called)
self.assertIs(query, self.query)
def test_save(self, add_insertion_data):
query = PostgreSQLQuery(INSERT_QUERY_TYPE)
query.save(TestModel(test_field_one=1))
self.assertEqual(add_insertion_data.call_count, 1)
def setUp(self):
self.query = PostgreSQLQuery(SELECT_QUERY_TYPE)
class TestPostrgreSQLQuery(unittest.TestCase):
prefix = 'elysium.backends.postgresql.builder.'
def setUp(self):
self.query = PostgreSQLQuery(SELECT_QUERY_TYPE)
@mock.patch(prefix + 'PostgreSQLSelectQueryBuilder.add_fields')
@mock.patch(prefix + 'PostgreSQLSelectQueryBuilder.add_models_from_fields')
def test_select(self, add_models_from_fields, add_fields):
self._query = PostgreSQLQuery(0)
query = self.query.select()
add_fields.return_value = None
self.assertTrue(add_fields.called)
self.assertTrue(add_models_from_fields.called)
self.assertIs(query, self.query)
@mock.patch(prefix + 'PostgreSQLSelectQueryBuilder.add_conditions')
def test_filter(self, add_conditions):
query = self.query.filter(
TestModel.test_field_one == TestModel.test_field_two)
add_conditions.return_value = None
self.assertTrue(add_conditions.called)
self.assertIs(query, self.query)
@mock.patch('elysium.backends.postgresql.query.PostgreSQLQuery._join')
def test_outer_join(self, _join):
_join.return_value = None
query = self.query.outer_join(TestModel)
self.assertTrue(_join.called)
self.assertIs(query, self.query)
@mock.patch('elysium.backends.postgresql.query.PostgreSQLQuery._join')
def test_join(self, _join):
_join.return_value = None
query = self.query.join(TestModel)
self.assertTrue(_join.called)
self.assertIs(query, self.query)
@mock.patch('elysium.backends.postgresql.query.PostgreSQLQuery._join')
def test_left_join(self, _join):
_join.return_value = None
query = self.query.left_join(TestModel)
self.assertTrue(_join.called)
self.assertIs(query, self.query)
@mock.patch('elysium.backends.postgresql.query.PostgreSQLQuery._join')
def test_right_join(self, _join):
_join.return_value = None
query = self.query.left_join(TestModel)
self.assertTrue(_join.called)
self.assertIs(query, self.query)
@mock.patch(prefix + 'PostgreSQLSelectQueryBuilder.add_ordering_fields')
def test_order_by(self, add_ordering_fields):
query = self.query.order_by(TestModel.test_field_two)
add_ordering_fields.return_value = None
self.assertTrue(add_ordering_fields.called)
self.assertEqual(query, self.query)
# TODO FIX after adding logic
def test_all(self):
query = self.query.all()
self.assertIs(query, self.query)
# TODO FIX after adding logic
def test_first(self):
query = self.query.first()
self.assertIs(query, self.query)
# TODO FIX after adding logic
def test_last(self):
query = self.query.last()
self.assertIs(query, self.query)
@mock.patch(prefix + 'PostgreSQLInsertQueryBuilder.add_insertion_data')
def test_save(self, add_insertion_data):
query = PostgreSQLQuery(INSERT_QUERY_TYPE)
query.save(TestModel(test_field_one=1))
self.assertEqual(add_insertion_data.call_count, 1)
@mock.patch(prefix + 'PostgreSQLSelectQueryBuilder.build')
def test_sql(self, build):
build.return_value = 'test'
query = self.query.sql()
self.assertEqual(query, 'test')
def get_query(self, query_type):
from elysium.backends.postgresql.query import PostgreSQLQuery
return PostgreSQLQuery(query_type)
def test_get_initial_query(self, set_model, get_query):
get_query.return_value = PostgreSQLQuery(1)
model_cls = ModelMetaClass('Model', (), {'field': Field()})
query = model_cls.get_initial_query(1)
self.assertEqual(get_query.call_count, 1)
self.assertEqual(set_model.call_count, 1)
def test_filter(self, filter, get_initial_query):
get_initial_query.return_value = PostgreSQLQuery(1)
model_cls = ModelMetaClass('Model', (), {'field': Field()})
model_cls.filter()
self.assertEqual(get_initial_query.call_count, 1)
self.assertEqual(filter.call_count, 1)
def test_save(self, save, get_initial_query):
get_initial_query.return_value = PostgreSQLQuery(1)
instance = self.model_class(field_one=1, field_two='test')
instance.save()
self.assertEqual(save.call_count, 1)