From 83d313150c789467cbc879d1055a28a6c7b05039 Mon Sep 17 00:00:00 2001 From: TrisNol Date: Tue, 17 Oct 2023 18:47:25 +0200 Subject: [PATCH] test: Update to new functions --- tests/utils/mongo/company_mongo_service_test.py | 15 ++++++++++++--- 1 file changed, 12 insertions(+), 3 deletions(-) diff --git a/tests/utils/mongo/company_mongo_service_test.py b/tests/utils/mongo/company_mongo_service_test.py index 52b58af..434198f 100644 --- a/tests/utils/mongo/company_mongo_service_test.py +++ b/tests/utils/mongo/company_mongo_service_test.py @@ -172,19 +172,28 @@ def test_get_where_malformed_yearly_results( "_id": "abc", "name": "Fielmann", "Hotel?": "Trivago", - "yearly_results": {"Vor Aeonen": 42, "2022": 4711}, + "yearly_results": { + "Vor Aeonen": {"auditors": [], 42: 1}, + "2022": {"auditors": [], 42: 1}, + }, }, { "_id": "abc", "name": "Fielmann", "Hotel?": "Trivago", - "yearly_results": {"1998": 42, "2022": 4711}, + "yearly_results": { + "1998": {"auditors": [], 42: 1}, + "2022": {"auditors": [], 42: 1}, + }, }, { "_id": "abc", "name": "Fielmann", "Hotel?": "Trivago", - "yearly_results": {"19": 42, "2022": 4711}, + "yearly_results": { + "19": {"auditors": [], 42: 1}, + "2022": {"auditors": [], 42: 1}, + }, }, ] mock_collection.find.return_value = mock_result