From 8de4326c727994760143e086dc369afe43d55f0f Mon Sep 17 00:00:00 2001 From: Jackson Goerner Date: Fri, 25 Oct 2024 13:12:28 +1100 Subject: [PATCH] Fix unit tests for databuilder. Signed-off-by: Jackson Goerner --- .../tests/unit/extractor/test_mysql_search_data_extractor.py | 3 ++- .../tests/unit/models/test_user_elasticsearch_document.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/databuilder/tests/unit/extractor/test_mysql_search_data_extractor.py b/databuilder/tests/unit/extractor/test_mysql_search_data_extractor.py index 5a2d25419..7e6c8641b 100644 --- a/databuilder/tests/unit/extractor/test_mysql_search_data_extractor.py +++ b/databuilder/tests/unit/extractor/test_mysql_search_data_extractor.py @@ -137,7 +137,8 @@ def test_user_search(self, manager = User(rk='test_manager_key', email='test_manager@email.com') user.manager = manager - expected_dict = dict(email='test_user@email.com', + expected_dict = dict(name='test_first_name test_last_name', + email='test_user@email.com', first_name='test_first_name', last_name='test_last_name', full_name='test_full_name', diff --git a/databuilder/tests/unit/models/test_user_elasticsearch_document.py b/databuilder/tests/unit/models/test_user_elasticsearch_document.py index 7ac379a17..6bf0da82c 100644 --- a/databuilder/tests/unit/models/test_user_elasticsearch_document.py +++ b/databuilder/tests/unit/models/test_user_elasticsearch_document.py @@ -13,7 +13,8 @@ def test_to_json(self) -> None: """ Test string generated from to_json method """ - test_obj = UserESDocument(email='test@email.com', + test_obj = UserESDocument(name='test_firstname test_lastname', + email='test@email.com', first_name='test_firstname', last_name='test_lastname', full_name='full_name',