test: update the test cases

2022-11-11 10:44:59 +08:00
parent 3170f875e5
commit 1993734a99
3 changed files with 54 additions and 11 deletions
@@ -2,6 +2,7 @@ import ReactFlow, { EdgeProps } from 'reactflow';

 import ControlledFlow from '../../support/ControlledFlow';
 import * as simpleflow from '../../fixtures/simpleflow';
+import { RefReactFlow } from '../../support/RefReactFlow';

 describe('<ReactFlow />: Basic Props', () => {
  describe('uses defaultNodes and defaultEdges', () => {
@@ -194,4 +195,32 @@ describe('<ReactFlow />: Basic Props', () => {
    cy.mount(<ControlledFlow className="custom" />);
    cy.get('.react-flow').should('have.class', 'custom');
  });
+
+  it('uses function ref', () => {
+    let pResolve: ((element: HTMLDivElement | PromiseLike<HTMLDivElement> | null) => void) | null = null;
+    const promise: Promise<HTMLDivElement | null> = new Promise((resolve) => {
+      pResolve = resolve;
+    });
+
+    function ref(root: HTMLDivElement | null) {
+      pResolve && pResolve(root);
+    }
+
+    cy.mount(<ReactFlow ref={ref} />);
+    cy.wrap(promise).should('be.instanceOf', HTMLDivElement);
+  });
+
+  it('use mutable ref', () => {
+    let pResolve: ((element: HTMLDivElement | PromiseLike<HTMLDivElement> | null) => void) | null = null;
+    const promise: Promise<HTMLDivElement | null> = new Promise((resolve) => {
+      pResolve = resolve;
+    });
+
+    function onGetRef(root: HTMLDivElement | null) {
+      pResolve && pResolve(root);
+    }
+
+    cy.mount(<RefReactFlow onGetRef={onGetRef} />);
+    cy.wrap(promise).should('be.instanceOf', HTMLDivElement);
+  });
 });