def test_add_both(self): queryset1 = Column.objects.all() columns = cache.get_many([1]) column_orders = [(cache.get(x), y) for x, y in [(1, 'desc')]] queryset2 = utils.add_columns(queryset1, columns) queryset2 = utils.add_ordering(queryset2, column_orders) self.assertEqual(str(queryset2.query), 'SELECT "avocado_column"."id", "avocado_column"."name", "avocado_column"."keywords" FROM "avocado_column" ORDER BY "avocado_column"."name" DESC, "avocado_column"."keywords" DESC') queryset3 = utils.add_ordering(queryset1, column_orders) queryset3 = utils.add_columns(queryset3, columns) self.assertEqual(str(queryset3.query), 'SELECT "avocado_column"."id", "avocado_column"."name", "avocado_column"."keywords" FROM "avocado_column" ORDER BY "avocado_column"."name" DESC, "avocado_column"."keywords" DESC') self.assertEqual(str(queryset2.query), str(queryset3.query)) columns = cache.get_many([2, 1]) column_orders = [(cache.get(x), y) for x, y in [(2, 'desc'), (1, 'asc')]] queryset4 = utils.add_columns(queryset1, columns) queryset4 = utils.add_ordering(queryset4, column_orders) self.assertEqual(str(queryset4.query), 'SELECT "avocado_column"."id", "avocado_field"."name", "avocado_field"."field_name", "avocado_column"."name", "avocado_column"."keywords" FROM "avocado_column" LEFT OUTER JOIN "avocado_columnfield" ON ("avocado_column"."id" = "avocado_columnfield"."concept_id") LEFT OUTER JOIN "avocado_field" ON ("avocado_columnfield"."field_id" = "avocado_field"."id") ORDER BY "avocado_field"."name" DESC, "avocado_field"."field_name" DESC, "avocado_column"."name" ASC, "avocado_column"."keywords" ASC') queryset5 = utils.add_ordering(queryset1, column_orders) queryset5 = utils.add_columns(queryset5, columns) self.assertEqual(str(queryset5.query), 'SELECT "avocado_column"."id", "avocado_field"."name", "avocado_field"."field_name", "avocado_column"."name", "avocado_column"."keywords" FROM "avocado_column" LEFT OUTER JOIN "avocado_columnfield" ON ("avocado_column"."id" = "avocado_columnfield"."concept_id") LEFT OUTER JOIN "avocado_field" ON ("avocado_columnfield"."field_id" = "avocado_field"."id") ORDER BY "avocado_field"."name" DESC, "avocado_field"."field_name" DESC, "avocado_column"."name" ASC, "avocado_column"."keywords" ASC') self.assertEqual(str(queryset4.query), str(queryset5.query))
def test_add_columns(self): queryset1 = Column.objects.all() columns = cache.get_many([1]) queryset2 = utils.add_columns(queryset1, columns) self.assertEqual(str(queryset2.query), 'SELECT "avocado_column"."id", "avocado_column"."name", "avocado_column"."keywords" FROM "avocado_column" ORDER BY "avocado_column"."name" ASC') columns = cache.get_many([1, 2]) queryset3 = utils.add_columns(queryset1, columns) self.assertEqual(str(queryset3.query), 'SELECT "avocado_column"."id", "avocado_column"."name", "avocado_column"."keywords", "avocado_field"."name", "avocado_field"."field_name" FROM "avocado_column" LEFT OUTER JOIN "avocado_columnfield" ON ("avocado_column"."id" = "avocado_columnfield"."concept_id") LEFT OUTER JOIN "avocado_field" ON ("avocado_columnfield"."field_id" = "avocado_field"."id") ORDER BY "avocado_column"."name" ASC')
def test_get_many(self): concept_ids = [1, 2] concepts = list(cache.get_many(concept_ids)) self.assertEqual([x.id for x in concepts], concept_ids) for i, x in enumerate(concept_ids): key = cache.id_key % x self.assertEqual(djcache.get(key), concepts[i])