From c578cbff8a931d8bec3c3e026c8908cb61bf71d4 Mon Sep 17 00:00:00 2001 From: Bertrand Bordage Date: Sun, 18 Jan 2015 18:11:21 +0100 Subject: [PATCH] Fixes a test. --- cachalot/tests/read.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/cachalot/tests/read.py b/cachalot/tests/read.py index 3234068..1d61cdb 100644 --- a/cachalot/tests/read.py +++ b/cachalot/tests/read.py @@ -530,7 +530,11 @@ class ReadTestCase(TransactionTestCase): Tests if queries executed from a DB cursor are not cached. """ - sql = 'SELECT * FROM %s;' % Test._meta.db_table + attname_column_list = [f.get_attname_column() + for f in Test._meta.fields] + attnames = [t[0] for t in attname_column_list] + columns = [t[1] for t in attname_column_list] + sql = 'SELECT %s FROM %s;' % (', '.join(columns), Test._meta.db_table) with self.assertNumQueries(1): cursor = connection.cursor() @@ -543,7 +547,7 @@ class ReadTestCase(TransactionTestCase): data2 = list(cursor.fetchall()) cursor.close() self.assertListEqual(data2, data1) - self.assertListEqual(data2, list(Test.objects.values_list())) + self.assertListEqual(data2, list(Test.objects.values_list(*attnames))) def test_missing_table_cache_key(self): with self.assertNumQueries(1):