|
@@ -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)
|