def test_del_DB_no_transac(self):
expected_db = SimpleDB()
self.test_db.delete('test_get_key')
self.assertEqual(self.test_db.getDB(), expected_db.getDB())
with self.assertRaises(Exception):
self.test_db.delete('test_bad_key')
with self.assertRaises(Exception):
self.test_db.delete(123)
def test_put_DB_no_transac(self):
expected_db = SimpleDB({
'test_get_key': 'test_val_1',
'test_key': 'test_value'
})
self.test_db.put('test_key', 'test_value')
self.assertEqual(self.test_db.getDB(), expected_db.getDB())
with self.assertRaises(Exception):
self.test_db.put('test_bad_key')
# Test bad input
with self.assertRaises(Exception):
self.test_db.put('test_bad_key', {'key': 123})
# Test bad key
with self.assertRaises(Exception):
self.test_db.put(123, '123')
import PySimpleGUI as sg
from service import PersionServiceImpl as psi
from service import DocFileServiceImpl as dsi
from create_doc_ui import CreateDocUI
from create_file_ui import CreateFileUI
from create_person_ui import CreatePersonUI
from person_detail_ui import PersonDetailUI
from doc_detail_ui import DocDetailUI
from db import SimpleDB
# read all person data from database
simple_db = SimpleDB()
simple_db.open(r"doc_management_sqlite3.db")
person_service = psi(simple_db)
doc_file_service = dsi(simple_db)
people = person_service.get_all_by_page(1, 40)
header = ["序号", "姓名", "性别", "身份证", "创建时间", "更新时间"]
col_widths = [8, 25, 6, 18, 25, 25]
skip_len = 0
for col_width in col_widths:
skip_len = skip_len + col_width
skip_len = skip_len - 45
data = []
for person in people:
elem = [
person.id, person.name, person.sex, person.id_card, person.update_time,
person.create_time
]
data.append(elem)
def setUp(self):
self.test_db = SimpleDB({
'test_get_key': 'test_val_1',
})
class TestSimpleDB(unittest.TestCase):
def setUp(self):
self.test_db = SimpleDB({
'test_get_key': 'test_val_1',
})
def test_getDB_helper(self):
expected_db = {
'test_get_key': 'test_val_1',
}
self.assertEqual(self.test_db.getDB(), expected_db)
def test_get_insert_transaction(self):
expected_trans = {
'test_get_transactionID': {
'value': {
'test_get_key': 'test_val'
},
'transaction_uuid': {
'test_get_key': 'test_uuid'
}
}
}
self.test_db.insert_transaction(expected_trans)
self.assertEqual(self.test_db.getTransaction(), expected_trans)
def test_check_commit_immediately(self):
commit_immediately, transID = self.test_db.check_commit_immediately(
None)
self.assertTrue(commit_immediately)
self.assertTrue(transID != None)
commit_immediately, transID = self.test_db.check_commit_immediately(
'uuid')
self.assertFalse(commit_immediately)
self.assertEqual(transID, 'uuid')
def test_get_DB_no_transac(self):
expected_val = 'test_val_1'
receive_val = self.test_db.get('test_get_key')
self.assertEqual(receive_val, expected_val)
with self.assertRaises(Exception):
self.test_db.get('test_bad_key')
with self.assertRaises(Exception):
self.test_db.get(123)
def test_put_DB_no_transac(self):
expected_db = SimpleDB({
'test_get_key': 'test_val_1',
'test_key': 'test_value'
})
self.test_db.put('test_key', 'test_value')
self.assertEqual(self.test_db.getDB(), expected_db.getDB())
with self.assertRaises(Exception):
self.test_db.put('test_bad_key')
# Test bad input
with self.assertRaises(Exception):
self.test_db.put('test_bad_key', {'key': 123})
# Test bad key
with self.assertRaises(Exception):
self.test_db.put(123, '123')
def test_del_DB_no_transac(self):
expected_db = SimpleDB()
self.test_db.delete('test_get_key')
self.assertEqual(self.test_db.getDB(), expected_db.getDB())
with self.assertRaises(Exception):
self.test_db.delete('test_bad_key')
with self.assertRaises(Exception):
self.test_db.delete(123)
def test_get_DB_transac(self):
self.test_db.insert_transaction({
'test_get_transactionID': {
'value': {
'test_get_key': 'test_val'
},
'transaction_uuid': {
'test_get_key': uuid.uuid4()
}
}
})
expected_val = 'test_val'
receive_val = self.test_db.get(
'test_get_key', 'test_get_transactionID')
self.assertEqual(receive_val, expected_val)
with self.assertRaises(Exception):
self.test_db.get('test_bad_key', 'bad_transID')
with self.assertRaises(Exception):
self.test_db.get('test_bad_key', 123)
def test_put_DB_transac(self):
# Test transaction value have been populated
# adding non exisit value via transaction hence transaction uuid should be None
expected_val = {
'test_put_transID': {
'value': {
'test_put_key': 'test_put_val'
},
'transaction_uuid': {
'test_put_key': None
}
}
}
trans_ID = 'test_put_transID'
test_key = 'test_put_key'
test_val = 'test_put_val'
self.test_db.createTransaction(trans_ID)
self.test_db.put(test_key, test_val, trans_ID)
receive_val = self.test_db.getTransaction()
self.assertTrue(trans_ID in receive_val)
self.assertTrue(test_key in receive_val[trans_ID]['transaction_uuid'])
self.assertTrue(receive_val[trans_ID]
['transaction_uuid'][test_key] == None)
self.assertEqual(receive_val[trans_ID]['value'],
expected_val[trans_ID]['value'])
# adding exisit value via transaction hence transaction uuid should be None
expected_val = {
'test_put_transID_2': {
'value': {
'test_get_key': 'test_put_val'
},
'transaction_uuid': {
'test_get_key': None
}
}
}
trans_ID = 'test_put_transID_2'
test_key = 'test_get_key'
test_val = 'test_put_val'
self.test_db.createTransaction(trans_ID)
self.test_db.put(test_key, test_val, trans_ID)
receive_val = self.test_db.getTransaction()
self.assertTrue(trans_ID in receive_val)
self.assertTrue(test_key in receive_val[trans_ID]['transaction_uuid'])
self.assertTrue(receive_val[trans_ID]
['transaction_uuid'][test_key] != None)
self.assertEqual(receive_val[trans_ID]['value'],
expected_val[trans_ID]['value'])
# Raise error on non exist transaction id
with self.assertRaises(Exception):
self.test_db.put('test_bad_key', 'test_bad_val', 'bad_transID')
with self.assertRaises(Exception):
self.test_db.put('test_bad_key', 'test_bad_val', 123)
def test_del_DB_transac(self):
expected_trans = {
'test_del_transactionID': {
'value': {
'test_del_key_2': 'test_val_2',
},
'transaction_uuid': {
'test_del_key_2': 'uuid_2',
}
}
}
self.test_db.insert_transaction({
'test_del_transactionID': {
'value': {
'test_del_key_1': 'test_val_1',
'test_del_key_2': 'test_val_2',
},
'transaction_uuid': {
'test_del_key_1': 'uuid_1',
'test_del_key_2': 'uuid_2',
}
}
})
self.test_db.delete('test_del_key_1', 'test_del_transactionID')
self.assertEqual(self.test_db.getTransaction(), expected_trans)
with self.assertRaises(Exception):
self.test_db.delete('test_bad_key', 'bad_transID')
with self.assertRaises(Exception):
self.test_db.delete('test_bad_key', 123)
def test_createTransaction(self):
expected_trans = {
'Test_transID': {
'value': {},
'transaction_uuid': {}
}
}
self.test_db.createTransaction('Test_transID')
self.assertEqual(self.test_db.getTransaction(), expected_trans)
with self.assertRaises(Exception):
self.test_db.createTransaction('Test_transID')
with self.assertRaises(Exception):
self.test_db.createTransaction(123)
def test_rollbackTransaction(self):
self.test_db.insert_transaction({
'test_rollback_transID': {
'value': {
'test_key_1': 'test_val_1',
},
'transaction_uuid': {
'test_key_1': 'uuid_1',
}
}
})
self.test_db.rollbackTransaction('test_rollback_transID')
self.assertEqual(self.test_db.getTransaction(), {})
with self.assertRaises(Exception):
self.test_db.rollbackTransaction('bad_transID')
def commitTransaction(self):
expected_DB = {
'test_get_key': 'test_val_1',
'test_key_1': 'test_val',
}
self.test_db.insert_transaction({
'test_commit_transID': {
'value': {
'test_key_1': 'test_val',
},
'transaction_uuid': {
'test_key_1': 'uuid',
}
}
})
self.test_db.commitTransaction('test_commit_transID')
self.assertEqual(self.test_db.getTransaction(), {})
self.assertEqual(self.test_db.getDB(), expected_DB)
# Test dirty read
self.test_db.put('dirty_read_key', 'init_val')
self.test_db.createTransaction('dirty_read_trans')
self.test_db.put('dirty_read_key', 'dirty_val')
self.test_db.get('dirty_read_key')
with self.assertRaises(Exception):
self.test_db.commitTransaction('dirty_read_trans')
# Test delete will invalid transaction
self.test_db.put('delete_key', 'init_val')
self.test_db.createTransaction('delete_trans')
self.test_db.put('delete_key', 'dirty_val')
self.test_db.delete('delete_key')
with self.assertRaises(Exception):
self.test_db.commitTransaction('delete_trans')
# Test modify will invalid transaction
self.test_db.put('put_key', 'init_val')
self.test_db.createTransaction('put_trans')
self.test_db.put('put_key', 'put_val', 'put_trans')
self.test_db.put('put_key', 'dirty_put_val')
with self.assertRaises(Exception):
self.test_db.commitTransaction('put_trans')
with self.assertRaises(Exception):
self.test_db.commitTransaction('bad_transID')
def setUp(self):
self.test_db = SimpleDB()
class TestSimpleDB_Scenario(unittest.TestCase):
def setUp(self):
self.test_db = SimpleDB()
def test_flow_no_transaction(self):
self.test_db.put('example', 'foo')
receive_val = self.test_db.get('example')
self.assertEqual(receive_val, 'foo')
self.test_db.delete('example')
self.assertEqual(self.test_db.getDB(), {})
with self.assertRaises(Exception):
self.test_db.get('example')
with self.assertRaises(Exception):
self.test_db.delete('example')
def test_flow_transaction(self):
self.test_db.createTransaction('abc')
self.test_db.put('a', 'foo', 'abc')
# returns 'foo'
receive_val = self.test_db.get('a', 'abc')
self.assertEqual(receive_val, 'foo')
# returns null / error
with self.assertRaises(Exception):
self.test_db.get('a')
self.test_db.createTransaction('xyz')
self.test_db.put('a', 'bar', 'xyz')
# // returns 'bar'
receive_val = self.test_db.get('a', 'xyz')
self.assertEqual(receive_val, 'bar')
self.test_db.commitTransaction('xyz')
# // returns 'bar'
receive_val = self.test_db.get('a')
self.assertEqual(receive_val, 'bar')
# // failure
with self.assertRaises(Exception):
self.test_db.commitTransaction('abc')
# // returns 'bar'
receive_val = self.test_db.get('a')
self.assertEqual(receive_val, 'bar')
self.test_db.createTransaction('abc')
self.test_db.put('a', 'foo', 'abc')
# // returns 'bar'
receive_val = self.test_db.get('a')
self.assertEqual(receive_val, 'bar')
self.test_db.rollbackTransaction('abc')
# // failure
with self.assertRaises(Exception):
self.test_db.put('a', 'foo', 'abc')
# // returns 'bar'
receive_val = self.test_db.get('a')
self.assertEqual(receive_val, 'bar')
self.test_db.createTransaction('def')
self.test_db.put('b', 'foo', 'def')
'''
NOTE: in the example this is return as bar but base on the API descibtion
● String get(String key, String transactionId)
○ Returns the value associated with “key” within the transaction with ID
“transactionId”
○ Throws an exception or returns an error on failure
This following command get('a', 'def') should throw an exception since
the key 'a' was never assign any value within the transaction 'def'
'''
with self.assertRaises(Exception):
self.test_db.get('a', 'def')
# // returns 'foo'
receive_val = self.test_db.get('b', 'def')
self.assertEqual(receive_val, 'foo')
self.test_db.rollbackTransaction('def')
# // returns null
with self.assertRaises(Exception):
self.test_db.get('b')