Просмотр исходного кода

Updated unit tests with latest database content.

Taddeus Kroes 13 лет назад
Родитель
Сommit
9d16d57d21
1 измененных файлов с 4 добавлено и 3 удалено
  1. 4 3
      tests/test_generate.py

+ 4 - 3
tests/test_generate.py

@@ -8,7 +8,7 @@ class GenerateTest(TestCase):
     conn = mysql.connect('localhost', 'root', 'mysql12#$', 'test')
     conn = mysql.connect('localhost', 'root', 'mysql12#$', 'test')
 
 
     def setUp(self):
     def setUp(self):
-        self.model = Model('payments')
+        self.model = Model('payments', {})
 
 
     def test_php_value_dict(self):
     def test_php_value_dict(self):
         self.assertEqual(php_value({}), 'array()')
         self.assertEqual(php_value({}), 'array()')
@@ -51,12 +51,13 @@ class GenerateTest(TestCase):
         self.assertEqual(flatten([['a'], ['b'], ('c', 'd')]), list('abcd'))
         self.assertEqual(flatten([['a'], ['b'], ('c', 'd')]), list('abcd'))
 
 
     def test_read_tables(self):
     def test_read_tables(self):
-        self.assertEqual(read_tables(self.conn), ['payments', 'users'])
+        self.assertEqual(read_tables(self.conn),
+                ['orders', 'payments', 'receipts', 'users'])
 
 
     def test_read_fields(self):
     def test_read_fields(self):
         fields = read_fields(self.conn, 'payments')
         fields = read_fields(self.conn, 'payments')
         self.assertEqual([field['Field'] for field in fields],
         self.assertEqual([field['Field'] for field in fields],
-                         ['id', 'amount', 'user_id'])
+                         ['id', 'amount', 'user_id', 'order_id'])
 
 
     def test_model_read_attributes(self):
     def test_model_read_attributes(self):
         self.model.read_attributes(self.conn)
         self.model.read_attributes(self.conn)