Fix unit tests for databuilder.

Signed-off-by: Jackson Goerner <[email protected]>
amundsen-io · Oct 25, 2024 · 8de4326 · 8de4326
1 parent b134aac
commit 8de4326
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 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
@@ -137,7 +137,8 @@ def test_user_search(self,
         manager = User(rk='test_manager_key', email='[email protected]')
         user.manager = manager
 
-        expected_dict = dict(email='[email protected]',
+        expected_dict = dict(name='test_first_name test_last_name',
+                             email='[email protected]',
                              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
@@ -13,7 +13,8 @@ def test_to_json(self) -> None:
         """
         Test string generated from to_json method
         """
-        test_obj = UserESDocument(email='[email protected]',
+        test_obj = UserESDocument(name='test_firstname test_lastname',
+                                  email='[email protected]',
                                   first_name='test_firstname',
                                   last_name='test_lastname',
                                   full_name='full_name',