use diagram for screenshots

packages/dmn-editor/tests/e2e/drdArtifacts/changeGroupProperties.spec.ts

@@ -50,7 +50,7 @@ test.describe("Change Properties - Group", () => {
     expect(await groupPropertiesPanel.getDescription()).toBe("New Group Description");
   });
 
-  test("should change the Group node font", async ({ nodes, groupPropertiesPanel }) => {
+  test("should change the Group node font", async ({ diagram, nodes, groupPropertiesPanel }) => {
     await nodes.select({ name: DefaultNodeName.GROUP, position: NodePosition.TOP });
     await groupPropertiesPanel.setFont({
       fontSize: "40",
@@ -62,7 +62,7 @@ test.describe("Change Properties - Group", () => {
       fontFamily: "Verdana",
     });
 
-    await expect(nodes.get({ name: DefaultNodeName.GROUP })).toHaveScreenshot("change-group-font.png");
+    await expect(diagram.get()).toHaveScreenshot("change-group-font.png");
   });
 
   test("should reset the Group node font", async ({ nodes, groupPropertiesPanel }) => {

packages/dmn-editor/tests/e2e/drdArtifacts/changeTextAnnotationProperties.spec.ts

@@ -61,7 +61,7 @@ test.describe("Change Properties - Text Annotation", () => {
     expect(await textAnnotationPropertiesPanel.getDescription()).toBe("New Text Annotation Description");
   });
 
-  test("should change the Text Annotation node font", async ({ nodes, textAnnotationPropertiesPanel }) => {
+  test("should change the Text Annotation node font", async ({ diagram, nodes, textAnnotationPropertiesPanel }) => {
     await nodes.select({ name: DefaultNodeName.TEXT_ANNOTATION });
     await textAnnotationPropertiesPanel.setFont({
       fontSize: "40",
@@ -73,9 +73,7 @@ test.describe("Change Properties - Text Annotation", () => {
       fontFamily: "Verdana",
     });
 
-    await expect(nodes.get({ name: DefaultNodeName.TEXT_ANNOTATION })).toHaveScreenshot(
-      "change-text-annotation-font.png"
-    );
+    await expect(diagram.get()).toHaveScreenshot("change-text-annotation-font.png");
   });
 
   test("should reset the Text Annotation node font", async ({ nodes, textAnnotationPropertiesPanel }) => {

packages/dmn-editor/tests/e2e/drgElements/changeBkmProperties.spec.ts

@@ -75,7 +75,7 @@ test.describe("Change Properties - BKM", () => {
     expect(links[0]).toHaveAttribute("href", "http://link.test.com/");
   });
 
-  test("should change the BKM node font", async ({ nodes, bkmPropertiesPanel }) => {
+  test("should change the BKM node font", async ({ diagram, nodes, bkmPropertiesPanel }) => {
     await nodes.select({ name: DefaultNodeName.BKM });
     await bkmPropertiesPanel.setFont({
       fontSize: "40",
@@ -87,7 +87,7 @@ test.describe("Change Properties - BKM", () => {
       fontFamily: "Verdana",
     });
 
-    await expect(nodes.get({ name: DefaultNodeName.BKM })).toHaveScreenshot("change-bkm-font.png");
+    await expect(diagram.get()).toHaveScreenshot("change-bkm-font.png");
   });
 
   test("should reset the BKM node font", async ({ nodes, bkmPropertiesPanel }) => {

packages/dmn-editor/tests/e2e/drgElements/changeDecisionProperties.spec.ts

@@ -96,7 +96,7 @@ test.describe("Change Properties - Decision", () => {
     expect(links[0]).toHaveAttribute("href", "http://link.test.com/");
   });
 
-  test("should change the Decision node font", async ({ nodes, decisionPropertiesPanel }) => {
+  test("should change the Decision node font", async ({ diagram, nodes, decisionPropertiesPanel }) => {
     await nodes.select({ name: DefaultNodeName.DECISION });
     await decisionPropertiesPanel.setFont({
       fontSize: "40",
@@ -108,7 +108,7 @@ test.describe("Change Properties - Decision", () => {
       fontFamily: "Verdana",
     });
 
-    await expect(nodes.get({ name: DefaultNodeName.DECISION })).toHaveScreenshot("change-decision-font.png");
+    await expect(diagram.get()).toHaveScreenshot("change-decision-font.png");
   });
 
   test("should reset the Decision node font", async ({ nodes, decisionPropertiesPanel }) => {

packages/dmn-editor/tests/e2e/drgElements/changeDecisionServiceProperties.spec.ts

@@ -83,7 +83,7 @@ test.describe("Change Properties - Decision Service", () => {
     expect(links[0]).toHaveAttribute("href", "http://link.test.com/");
   });
 
-  test("should change the Decision Service node font", async ({ nodes, decisionServicePropertiesPanel }) => {
+  test("should change the Decision Service node font", async ({ diagram, nodes, decisionServicePropertiesPanel }) => {
     await nodes.select({ name: DefaultNodeName.DECISION_SERVICE, position: NodePosition.TOP });
     await decisionServicePropertiesPanel.setFont({
       fontSize: "40",
@@ -95,9 +95,7 @@ test.describe("Change Properties - Decision Service", () => {
       fontFamily: "Verdana",
     });
 
-    await expect(nodes.get({ name: DefaultNodeName.DECISION_SERVICE })).toHaveScreenshot(
-      "change-decision-service-font.png"
-    );
+    await expect(diagram.get()).toHaveScreenshot("change-decision-service-font.png");
   });
 
   test("should reset the Decision Service node font", async ({ nodes, decisionServicePropertiesPanel }) => {

packages/dmn-editor/tests/e2e/drgElements/changeInputDataProperties.spec.ts

@@ -76,7 +76,7 @@ test.describe("Change Properties - Input Data", () => {
     expect(links[0]).toHaveAttribute("href", "http://link.test.com/");
   });
 
-  test("should change the Input Data node font", async ({ nodes, inputDataPropertiesPanel }) => {
+  test("should change the Input Data node font", async ({ diagram, nodes, inputDataPropertiesPanel }) => {
     await nodes.select({ name: DefaultNodeName.INPUT_DATA });
     await inputDataPropertiesPanel.setFont({
       fontSize: "40",
@@ -88,7 +88,7 @@ test.describe("Change Properties - Input Data", () => {
       fontFamily: "Verdana",
     });
 
-    await expect(nodes.get({ name: DefaultNodeName.INPUT_DATA })).toHaveScreenshot("change-input-data-font.png");
+    await expect(diagram.get()).toHaveScreenshot("change-input-data-font.png");
   });
 
   test("should reset the Input Data node font", async ({ nodes, inputDataPropertiesPanel }) => {

packages/dmn-editor/tests/e2e/drgElements/changeKnowledgeSourceProperties.spec.ts

@@ -89,7 +89,7 @@ test.describe("Change Properties - Knowledge Source", () => {
     expect(links[0]).toHaveAttribute("href", "http://link.test.com/");
   });
 
-  test("should change the Knowledge Source node font", async ({ nodes, knowledgeSourcePropertiesPanel }) => {
+  test("should change the Knowledge Source node font", async ({ diagram, nodes, knowledgeSourcePropertiesPanel }) => {
     await nodes.select({ name: DefaultNodeName.KNOWLEDGE_SOURCE });
     await knowledgeSourcePropertiesPanel.setFont({
       fontSize: "40",
@@ -101,9 +101,7 @@ test.describe("Change Properties - Knowledge Source", () => {
       fontFamily: "Verdana",
     });
 
-    await expect(nodes.get({ name: DefaultNodeName.KNOWLEDGE_SOURCE })).toHaveScreenshot(
-      "change-knowledge-source-font.png"
-    );
+    await expect(diagram.get()).toHaveScreenshot("change-knowledge-source-font.png");
   });
 
   test("should reset the Knowledge Source node font", async ({ nodes, knowledgeSourcePropertiesPanel }) => {