Have another shot at fixing the user data tests.

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

Hide file tree

Showing 2 changed files with 3 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,7 @@ def test_user_search(self,
         manager = User(rk='test_manager_key', email='[email protected]')
         user.manager = manager
 
-        expected_dict = dict(name='test_first_name test_last_name',
+        expected_dict = dict(name='test_full_name',
                              email='[email protected]',
                              first_name='test_first_name',
                              last_name='test_last_name',

diff --git a/databuilder/tests/unit/models/test_user_elasticsearch_document.py b/databuilder/tests/unit/models/test_user_elasticsearch_document.py
@@ -29,7 +29,8 @@ def test_to_json(self) -> None:
                                   total_own=3,
                                   total_follow=1)
 
-        expected_document_dict = {"first_name": "test_firstname",
+        expected_document_dict = {"name": "test_firstname test_lastname",
+                                  "first_name": "test_firstname",
                                   "last_name": "test_lastname",
                                   "full_name": "full_name",
                                   "team_name": "team",