ORM/test_mysql.py at master · SimeonRolev/ORM · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
import unittest
from mock import Mock
from model import Model, CharField, IntegerField
from database import MySQLDatabase
from database import *
from query import *


class School(Model):
    # These fields get declared in the class.
    # When you make an instance, it only saves the value
    name = CharField()
    age = IntegerField()
    students_count = IntegerField()


class TestMySql(unittest.TestCase):

    def setUp(self):
        self.ms = MySQLDatabase('MySqlSchools')
        self.ms.create_tables(School)
        self.cursor = Mock()
        self.ms.cursor = self.cursor
        self.s = School(name="Stefan Karadja", age=100, students_count=150)

    def test_setup(self):
        self.ms.create_tables(School)
        self.cursor.execute.assert_called_with(
            'CREATE TABLE IF NOT EXISTS School ('
            'rowid MEDIUMINT NOT NULL AUTO_INCREMENT, PRIMARY KEY (rowid), '
            'name text, age real, students_count real)')

    def test_placeholders(self):
        self.assertEqual(self.ms.placeholders(School), "(%s, %s, %s)")

    def test_row_values(self):
        self.assertEqual(row_values(self.s), ("Stefan Karadja", '100', '150'))

    def test_insert(self):
        self.ms.insert('School', self.s)
        self.cursor.execute.assert_called_with(
            'INSERT INTO School (name, age, students_count) VALUES (%s, %s, %s)', ('Stefan Karadja', '100', '150'))

    def test_where(self):
        self.ms.create_tables(School)
        q = self.s.select().where(School.age > 5)
        self.assertEqual(q.query, "SELECT * FROM School WHERE age>%s ")

    def test_contains(self):
        self.ms.create_tables(School)
        q = self.s.select().where(School.age.contains("Last"))
        self.assertEqual(q.query, 'SELECT * FROM School WHERE age LIKE %s ')
        q.limit(10).get()
        self.cursor.execute.assert_called_with(
            'SELECT * FROM School WHERE age LIKE %s ', ['%Last%'])

    def test_starts_with(self):
        self.ms.create_tables(School)
        q = self.s.select().where(School.age.startswith("Last"))
        self.assertEqual(q.query, 'SELECT * FROM School WHERE age LIKE %s ')
        q.limit(10).get()
        self.cursor.execute.assert_called_with(
            'SELECT * FROM School WHERE age LIKE %s ', ['Last%'])

    def test_ends_with(self):
        self.ms.create_tables(School)
        q = self.s.select().where(School.age.endswith("Last"))
        self.assertEqual(q.query, 'SELECT * FROM School WHERE age LIKE %s ')
        q.limit(10).get()
        self.cursor.execute.assert_called_with(
            'SELECT * FROM School WHERE age LIKE %s ', ['%Last'])

    def test_in(self):
        self.ms.create_tables(School)
        q = self.s.select().where(School.age > 5).where(School.age.in_('120', '50'))
        q.get()
        self.cursor.execute.assert_called_with(
            'SELECT * FROM School WHERE age>%s AND age IN (%s,%s) ', [5, '120', '50'])

        q = self.s.select().where(School.age.in_('120', '50'))
        q.get()
        self.cursor.execute.assert_called_with(
            'SELECT * FROM School WHERE age IN (%s,%s) ', ['120', '50'])

    def test_or(self):
        self.ms.create_tables(School)
        q = self.s.select().where(and_(School.age > 5, School.name == "Gosho"))
        q.get()
        self.cursor.execute.assert_called_with(
            'SELECT * FROM School WHERE (age>%s AND name=%s) ', [5, 'Gosho'])


if __name__ == '__main__':
    unittest.main()