def test_database_insert_one(data_three_filter_logicals_company_code):
mongo_mock = MongoDBMock()
mongo_handler = MongoDBHandler(mongo_mock)
mongo_handler.save_entry(data_three_filter_logicals_company_code)
assert mongo_mock.collection.find().count() == 1
def test_database_find_non_existing():
mongo_mock = MongoDBMock()
mongo_handler = MongoDBHandler(mongo_mock)
entry = mongo_handler.find_entry(ObjectId("5c61a1295f627d1db904dd39"))
assert not entry
def test_database_delete_non_existing(data_single_filter_logical_company_code):
mongo_mock = MongoDBMock()
mongo_handler = MongoDBHandler(mongo_mock)
deleted_num = mongo_handler.delete_entry(
ObjectId("5c61a1295f627d1db904dd39"))
assert deleted_num == 0
def test_database_find_existing(data_single_filter_logical_company_code):
mongo_mock = MongoDBMock()
mongo_mock.collection.insert_one(data_single_filter_logical_company_code)
mongo_handler = MongoDBHandler(mongo_mock)
entry = mongo_handler.find_entry(ObjectId("5c61a1295f627d1db904dd39"))
assert entry
def test_database_total_score_eleven(data_single_filter_logical_company_code,
data_three_filter_logicals_company_code):
mongo_mock = MongoDBMock()
mongo_mock.collection.insert_many([
data_single_filter_logical_company_code,
data_three_filter_logicals_company_code
])
mongo_handler = MongoDBHandler(mongo_mock)
total_score = mongo_handler.total_score()
assert total_score == 11
def test_database_total_entries_two(data_single_filter_logical_company_code,
data_three_filter_logicals_company_code):
mongo_mock = MongoDBMock()
mongo_mock.collection.insert_many([
data_single_filter_logical_company_code,
data_three_filter_logicals_company_code
])
mongo_handler = MongoDBHandler(mongo_mock)
total_entries = mongo_handler.total_entries()
assert total_entries == 2
def test_database_delete_collection(data_single_filter_logical_company_code,
data_three_filter_logicals_company_code):
mongo_mock = MongoDBMock()
mongo_mock.collection.insert_many([
data_single_filter_logical_company_code,
data_three_filter_logicals_company_code
])
mongo_handler = MongoDBHandler(mongo_mock)
mongo_handler.delete_collection()
assert mongo_mock.collection.find().count() == 0
def test_distinct_errorous_two_entities(
data_single_filter_logical_company_code_error,
data_search_output_set_error):
mongo_mock = MongoDBMock()
mongo_mock.collection.insert_many([
data_single_filter_logical_company_code_error,
data_search_output_set_error
])
mongo_handler = MongoDBHandler(mongo_mock)
distinct_entities = mongo_handler.find_distinct_errorous_entity_names()
assert set(distinct_entities) == set(
['C_CorrespondenceOutputSet', 'C_CorrespondenceCompanyCodeVH'])
def test_database_sample_filter_entries_no_exclude(
data_single_filter_logical_company_code,
data_two_filter_logicals_company_code,
data_three_filter_logicals_company_code, data_search_output_set):
mongo_mock = MongoDBMock()
mongo_mock.collection.insert_many([
data_single_filter_logical_company_code,
data_two_filter_logicals_company_code,
data_three_filter_logicals_company_code, data_search_output_set
])
mongo_handler = MongoDBHandler(mongo_mock)
entry = mongo_handler.sample_filter_entry('C_CorrespondenceCompanyCodeVH',
1)
assert entry == data_three_filter_logicals_company_code
def test_database_find_best_two(data_single_filter_logical_company_code,
data_single_filter_logical_company_code_error,
data_search_correspondence_company_code_error):
mongo_mock = MongoDBMock()
mongo_mock.collection.insert_many([
data_single_filter_logical_company_code,
data_single_filter_logical_company_code_error,
data_search_correspondence_company_code_error
])
mongo_handler = MongoDBHandler(mongo_mock)
best_two_entries = mongo_handler.find_best_entries(
'C_CorrespondenceCompanyCodeVH')
assert best_two_entries == [
data_single_filter_logical_company_code_error,
data_search_correspondence_company_code_error
]
