fix tests

Author: zprobst

nodestream_plugin_neo4j/ingest_query_builder.py

@@ -114,6 +114,8 @@ def _make_relationship(
     set_properties_query = f"SET {RELATIONSHIP_REF_NAME} += params.{generate_prefixed_param_name(PROPERTIES_PARAM_NAME, RELATIONSHIP_REF_NAME)}"
     if creation_rule == RelationshipCreationRule.CREATE:
         create_rel_query = str(merge_rel_query).replace("MERGE", "CREATE")
+        if set_first_ingested_at:
+            set_properties_query = f"{set_properties_query} SET {RELATIONSHIP_REF_NAME}.first_ingested_at = params.{generate_prefixed_param_name(PROPERTIES_PARAM_NAME, RELATIONSHIP_REF_NAME)}['last_ingested_at']"
         return f"{create_rel_query} {set_properties_query}"
 
     if set_first_ingested_at:

tests/unit/test_ingest_query_builder.py

@@ -394,7 +394,7 @@ def test_node_update_with_created_timestamp_generates_exepected_queries(
 
 RELATIONSHIP_BETWEEN_TWO_NODES_EXPECTED_QUERY_WITH_MULTI_KEY_AND_CREATE_AND_TIMESTAMP = QueryBatch(
     """MATCH (from_node: TestType) WHERE from_node.id = params.__from_node_id MATCH (to_node: ComplexType) WHERE to_node.id_part1 = params.__to_node_id_part1 AND to_node.id_part2 = params.__to_node_id_part2
-    CREATE (from_node)-[rel: RELATED_TO]->(to_node) ON CREATE SET rel.first_ingested_at = params.__rel_properties['last_ingested_at']  SET rel += params.__rel_properties""",
+    CREATE (from_node)-[rel: RELATED_TO]->(to_node) SET rel += params.__rel_properties SET rel.first_ingested_at = params.__rel_properties['last_ingested_at']""",
     [
         {
             "__from_node_id": "foo",