-
Notifications
You must be signed in to change notification settings - Fork 7
/
test_upd_employee_audit.py
68 lines (51 loc) · 2.35 KB
/
test_upd_employee_audit.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
import sys, unittest
import logic_bank_utils.util as logic_bank_utils
from datetime import datetime
from decimal import Decimal
(did_fix_path, sys_env_info) = \
logic_bank_utils.add_python_path(project_dir="LogicBank", my_file=__file__)
if __name__ == '__main__':
print("\nStarted from cmd line - launch unittest and exit\n")
sys.argv = [sys.argv[0]]
unittest.main(module="examples.nw.tests.test_upd_employee_audit")
exit(0)
else:
print("Started from unittest: " + __name__)
from examples.nw import tests
tests.copy_gold_over_db()
import examples.nw.db.models as models
from examples.nw.logic import session, engine # opens db, activates rules <--
# activate rules: LogicBank.activate(session=session, activator=declare_logic)
print("\n" + sys_env_info + "\n\n")
class Test(unittest.TestCase):
def setUp(self): # banner
self.started_at = str(datetime.now())
tests.setUp(file=__file__)
def tearDown(self):
tests.tearDown(file=__file__, started_at=self.started_at, engine=engine, session=session)
def test_run(self):
"""
Test 1 - alter City, ensure no EmployeeAudit created
"""
print("\n\nTest 1 - alter Salary, ensure no EmployeeAudit created\n")
test_emp = session.query(models.Employee).filter(models.Employee.Id == 1).one()
test_emp.City = "don't audit this"
session.commit()
test_emp_audit = session.query(models.EmployeeAudit).all()
if test_emp_audit:
self.fail("Failure - audit row created for just City change")
"""
Test 2 - alter Salary, ensure EmployeeAudit created
"""
print("\n\nTest 2 - alter Salary, ensure EmployeeAudit created (from NWRuleExtension.nw_copy_row)")
test_emp = session.query(models.Employee).filter(models.Employee.Id == 1).one()
test_emp.Salary = test_emp.Salary * Decimal(1.5)
session.commit()
print("")
# test_emp_audit = session.query(models.EmployeeAudit).one()
query = session.query(models.EmployeeAudit)
print(f" .. test_emp_audit - did query, count = {query.count()}")
test_emp_audit = query.one()
print(f" .. test_emp_audit {test_emp_audit.LastName}")
if False and test_emp_audit is None:
self.fail("Failure - audit row not created on Salary change")