Lineage IT test.

jpmorganchase · Jan 14, 2025 · 3bb3d21 · 3bb3d21
1 parent a1b10ad
commit 3bb3d21
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 5 deletions.
diff --git a/src/it/java/io/github/jpmorganchase/fusion/packaging/DatasetOperationsIT.java b/src/it/java/io/github/jpmorganchase/fusion/packaging/DatasetOperationsIT.java
@@ -136,15 +136,17 @@ public void testUpdateDataset() {
     @Test
     public void testUpdateDatasetLineage() {
         // Given
-        wireMockRule.stubFor(WireMock.put(WireMock.urlEqualTo("/catalogs/common/datasets/SD0004"))
-                .withRequestBody(equalToJson(TestUtils.loadJsonForIt("dataset/dataset-SD0004-update-request.json")))
+        wireMockRule.stubFor(WireMock.post(WireMock.urlEqualTo("/catalogs/common/datasets/SD0002/lineage"))
+                .withRequestBody(equalToJson(TestUtils.loadJsonForIt("dataset/dataset-SD0002-lineage-create-request.json")))
                 .withHeader("Content-Type", WireMock.equalTo("application/json"))
                 .willReturn(WireMock.aResponse()
                         .withHeader("Content-Type", "application/json")
-                        .withStatus(200)
-                        .withBodyFile("dataset/dataset-update-response.json")));
+                        .withStatus(200)));
 
-        Dataset dataset = getSdk().builders().dataset().build();
+        Dataset dataset = getSdk().builders().dataset()
+                .identifier("SD0002")
+                .catalogIdentifier("common")
+                .build();
 
 
         // When & Then

diff --git a/src/test/resources/__files/dataset/dataset-SD0002-lineage-create-request.json b/src/test/resources/__files/dataset/dataset-SD0002-lineage-create-request.json
@@ -0,0 +1,15 @@
+{
+  "source" : [ {
+    "catalog" : "foo",
+    "dataset" : "d1"
+  }, {
+    "catalog" : "foo",
+    "dataset" : "d2"
+  }, {
+    "catalog" : "bar",
+    "dataset" : "d1"
+  }, {
+    "catalog" : "bar",
+    "dataset" : "d3"
+  } ]
+}