/
test.py
124 lines (83 loc) · 3.16 KB
/
test.py
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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
import os
import unittest
from models import Base, Book, Author, Publisher, engine
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
Base.metadata.bind = engine
DBSession = sessionmaker(bind=engine)
session = DBSession()
class DBTestCases(unittest.TestCase):
def test_source_insert_1(self):
s = Book(id='1', title = 'TESTONE')
session.add(s)
session.commit()
r = session.query(Book).filter_by(id = '1').one()
self.assertEqual(str(r.id), '1')
session.query(Book).filter_by(id = '1').delete()
session.commit()
def test_source_insert_2(self):
s = Book(id='2', title = 'TESTTWO')
session.add(s)
session.commit()
r = session.query(Book).filter_by(id = '2').one()
self.assertEqual(str(r.id), '2')
session.query(Book).filter_by(id = '2').delete()
session.commit()
def test_source_insert_3(self):
s = Book(id='3', title = 'TESTTHREE')
session.add(s)
session.commit()
r = session.query(Book).filter_by(id = '3').one()
self.assertEqual(str(r.id), '3')
session.query(Book).filter_by(id = '3').delete()
session.commit()
def test_source_insert_4(self):
s = Author(id='1', name = 'TESTONE')
session.add(s)
session.commit()
r = session.query(Author).filter_by(id = '1').one()
self.assertEqual(str(r.id), '1')
session.query(Author).filter_by(id = '1').delete()
session.commit()
def test_source_insert_5(self):
s = Author(id='2', name = 'TESTTWO')
session.add(s)
session.commit()
r = session.query(Author).filter_by(id = '2').one()
self.assertEqual(str(r.id), '2')
session.query(Author).filter_by(id = '2').delete()
session.commit()
def test_source_insert_6(self):
s = Author(id='3', name = 'TESTTHREE')
session.add(s)
session.commit()
r = session.query(Author).filter_by(id = '3').one()
self.assertEqual(str(r.id), '3')
session.query(Author).filter_by(id = '3').delete()
session.commit()
def test_source_insert_7(self):
s = Publisher(id='1', name = 'TESTONE')
session.add(s)
session.commit()
r = session.query(Publisher).filter_by(id = '1').one()
self.assertEqual(str(r.id), '1')
session.query(Publisher).filter_by(id = '1').delete()
session.commit()
def test_source_insert_8(self):
s = Publisher(id='2', name = 'TESTTWO')
session.add(s)
session.commit()
r = session.query(Publisher).filter_by(id = '2').one()
self.assertEqual(str(r.id), '2')
session.query(Publisher).filter_by(id = '2').delete()
session.commit()
def test_source_insert_9(self):
s = Publisher(id='3', name = 'TESTTHREE')
session.add(s)
session.commit()
r = session.query(Publisher).filter_by(id = '3').one()
self.assertEqual(str(r.id), '3')
session.query(Publisher).filter_by(id = '3').delete()
session.commit()
if __name__ == '__main__':
unittest.main()