diff --git a/tests/data/graph/querybuilder/sample_data/case_insensitive_prop_ref.py b/tests/data/graph/querybuilder/sample_data/case_insensitive_prop_ref.py index 21e53c6fd..79ebc78b5 100644 --- a/tests/data/graph/querybuilder/sample_data/case_insensitive_prop_ref.py +++ b/tests/data/graph/querybuilder/sample_data/case_insensitive_prop_ref.py @@ -46,3 +46,21 @@ 'github_username': 'mbsimp-son', # pure lowercase }, ] + +FAKE_EMPLOYEE2_DATA = [ + { + 'id': 123, + 'email': 'hjsimpson@example.com', + 'first_name': 'Homer', + 'last_name': 'Simpson', + 'name': 'Homer Simpson', + 'github_username': 'jsimpso', # substring + }, + { + 'id': 456, + 'email': 'mbsimpson@example.com', + 'first_name': 'Marge', + 'last_name': 'Simpson', + 'github_username': 'mbsimp', # substring + }, +] diff --git a/tests/integration/cartography/graph/test_querybuilder_fuzzy_case_insensitive.py b/tests/integration/cartography/graph/test_querybuilder_fuzzy_case_insensitive.py index d81b06d3d..4d624a2a1 100644 --- a/tests/integration/cartography/graph/test_querybuilder_fuzzy_case_insensitive.py +++ b/tests/integration/cartography/graph/test_querybuilder_fuzzy_case_insensitive.py @@ -1,6 +1,6 @@ from cartography.client.core.tx import load from cartography.intel.github.users import load_organization_users -from tests.data.graph.querybuilder.sample_data.case_insensitive_prop_ref import FAKE_EMPLOYEE_DATA +from tests.data.graph.querybuilder.sample_data.case_insensitive_prop_ref import FAKE_EMPLOYEE2_DATA from tests.data.graph.querybuilder.sample_data.case_insensitive_prop_ref import FAKE_GITHUB_ORG_DATA from tests.data.graph.querybuilder.sample_data.case_insensitive_prop_ref import FAKE_GITHUB_USER_DATA from tests.data.graph.querybuilder.sample_models.fake_emps_githubusers_fuzzy import FakeEmp2Schema @@ -19,7 +19,7 @@ def test_load_team_members_data_fuzzy(neo4j_session): ) # Act: Create team members - load(neo4j_session, FakeEmp2Schema(), FAKE_EMPLOYEE_DATA, lastupdated=TEST_UPDATE_TAG) + load(neo4j_session, FakeEmp2Schema(), FAKE_EMPLOYEE2_DATA, lastupdated=TEST_UPDATE_TAG) # Assert we can create relationships using a fuzzy, case insensitive match assert check_rels(neo4j_session, 'FakeEmployee2', 'email', 'GitHubUser', 'username', 'IDENTITY_GITHUB') == {