def test_find_cids_by_ocns_none(create_test_db):
db_conn_str = create_test_db['db_conn_str']
db = prepare_database(db_conn_str)
engine = db["engine"]
session = db["session"]
CidMintingStore = db["table"]
ocns_list = []
expected = {
'inquiry_ocns': [],
'matched_cids': [],
'min_cid': None,
'num_of_cids': 0
}
results = find_cids_by_ocns(engine, ocns_list)
assert results == expected
ocns_list = ['1234567890']
expected = {
'inquiry_ocns': ['1234567890'],
'matched_cids': [],
'min_cid': None,
'num_of_cids': 0
}
results = find_cids_by_ocns(engine, ocns_list)
assert results == expected
def test_find_by_identifier(create_test_db):
db_conn_str = create_test_db['db_conn_str']
db = prepare_database(db_conn_str)
engine = db["engine"]
session = db["session"]
CidMintingStore = db["table"]
record = find_by_identifier(CidMintingStore, session, 'ocn', '8727632')
print(record)
assert [record.type, record.identifier,
record.cid] == ['ocn', '8727632', '002492721']
record = find_by_identifier(CidMintingStore, session, 'ocn', '1234567890')
assert record == None
record = find_by_identifier(CidMintingStore, session, 'ocn', '')
assert record == None
record = find_by_identifier(CidMintingStore, session, 'sysid', 'pur215476')
print(record)
assert [record.type, record.identifier,
record.cid] == ['sysid', 'pur215476', '002492721']
record = find_by_identifier(CidMintingStore, session, 'sysid', 'xyz12345')
assert record == None
record = find_by_identifier(CidMintingStore, session, 'sysid', '')
assert record == None
def test_find_cids_by_ocns(create_test_db):
db_conn_str = create_test_db['db_conn_str']
db = prepare_database(db_conn_str)
engine = db["engine"]
session = db["session"]
CidMintingStore = db["table"]
ocns_list = ['8727632', '32882115']
expected_results = {
'inquiry_ocns': ['8727632', '32882115'],
'matched_cids': [{
'cid': '002492721'
}, {
'cid': '011323405'
}],
'min_cid': '002492721',
'num_of_cids': 2,
}
results = find_cids_by_ocns(engine, ocns_list)
print(results)
assert results['min_cid'] == expected_results['min_cid']
assert results['num_of_cids'] == expected_results['num_of_cids']
for result in results['inquiry_ocns']:
assert any(expected_result == result
for expected_result in expected_results['inquiry_ocns'])
for result in results['matched_cids']:
assert any(expected_result == result
for expected_result in expected_results['matched_cids'])
def test_insert_a_record(caplog, create_test_db):
caplog.set_level(logging.DEBUG)
db_conn_str = create_test_db['db_conn_str']
db = prepare_database(db_conn_str)
engine = db["engine"]
session = db["session"]
CidMintingStore = db["table"]
# before insert a record
results = find_all(CidMintingStore, session)
assert len(results) == 5
record = CidMintingStore(type='ocn',
identifier='30461866',
cid='011323406')
insert_a_record(session, record)
# after insert a record
results = find_all(CidMintingStore, session)
assert len(results) == 6
assert any([record.type, record.identifier, record.cid] ==
['ocn', '30461866', '011323406'] for record in results)
# insert the same record
record = CidMintingStore(type='ocn',
identifier='30461866',
cid='011323406')
insert_a_record(session, record)
assert "IntegrityError adding record" in caplog.text
results = find_all(CidMintingStore, session)
assert len(results) == 6
def test_find_all(create_test_db):
db_conn_str = create_test_db['db_conn_str']
db = prepare_database(db_conn_str)
engine = db["engine"]
session = db["session"]
CidMintingStore = db["table"]
results = find_all(CidMintingStore, session)
print(type(results))
assert len(results) == 5
assert any([record.type, record.identifier, record.cid] ==
['ocn', '8727632', '002492721'] for record in results)
assert any([record.type, record.identifier, record.cid] ==
['sysid', 'pur864352', '011323405'] for record in results)
def test_find_query(create_test_db):
""" the 'create_test_db' argument here is matched to the name of the
fixture above
"""
db_conn_str = create_test_db['db_conn_str']
db = prepare_database(db_conn_str)
engine = db["engine"]
session = db["session"]
CidMintingStore = db["table"]
select_sql = "select type, identifier, cid from cid_minting_store"
results = find_query(engine, select_sql)
print(results)
assert len(results) == 5
# expected results:
expected_results = [{
'type': 'ocn',
'identifier': '8727632',
'cid': '002492721'
}, {
'type': 'sysid',
'identifier': 'pur215476',
'cid': '002492721'
}, {
'type': 'ocn',
'identifier': '32882115',
'cid': '011323405'
}, {
'type': 'sysid',
'identifier': 'pur864352',
'cid': '011323405'
}, {
'type': 'sysid',
'identifier': 'uc1234567',
'cid': '011323405'
}]
for record in results:
assert any(expected_result == record
for expected_result in expected_results)
def test_find_cid_by_sysid(create_test_db):
db_conn_str = create_test_db['db_conn_str']
db = prepare_database(db_conn_str)
engine = db["engine"]
session = db["session"]
CidMintingStore = db["table"]
sysid = ""
expected = {}
result = find_cid_by_sysid(CidMintingStore, session, sysid)
print(result)
assert result == expected
sysid = "xyz123"
expected = {}
result = find_cid_by_sysid(CidMintingStore, session, sysid)
print(result)
assert result == expected
sysid = "uc1234567"
expected = {'inquiry_sys_id': 'uc1234567', 'matched_cid': '011323405'}
result = find_cid_by_sysid(CidMintingStore, session, sysid)
print(result)
assert result == expected