@finos/legend-application-studio 27.0.0 → 27.1.1

package/lib/stores/editor/editor-state/element-editor-state/testable/TestAssertionState.js

@@ -14,13 +14,14 @@
  * limitations under the License.
  */
 import { AssertFail, TestExecuted, TestError, EqualToJson, ExternalFormatData, EqualToJsonAssertFail, MultiExecutionServiceTestResult, AssertPass, TestExecutionStatus, } from '@finos/legend-graph';
-import { ActionState, assertErrorThrown, ContentType, UnsupportedOperationError, } from '@finos/legend-shared';
-import { action, flow, makeObservable, observable } from 'mobx';
+import { ActionState, assertErrorThrown, ContentType, UnsupportedOperationError, assertTrue, isNonNullable, IllegalStateError, guaranteeNonNullable, guaranteeType, } from '@finos/legend-shared';
+import { action, flow, flowResult, makeObservable, observable } from 'mobx';
 import { externalFormatData_setData } from '../../../../graph-modifier/DSL_Data_GraphModifierHelper.js';
 import { getTestableResultFromAssertionStatus, TESTABLE_RESULT, } from '../../../sidebar-state/testable/GlobalTestRunnerState.js';
+import { isTestPassing } from '../../../utils/TestableUtils.js';
 export var TEST_ASSERTION_TAB;
 (function (TEST_ASSERTION_TAB) {
-    TEST_ASSERTION_TAB["SETUP"] = "SETUP";
+    TEST_ASSERTION_TAB["EXPECTED"] = "EXPECTED";
     TEST_ASSERTION_TAB["RESULT"] = "RESULT";
 })(TEST_ASSERTION_TAB || (TEST_ASSERTION_TAB = {}));
 export class TestAssertionStatusState {
@@ -157,11 +158,16 @@ export class EqualToJsonAssertionState extends TestAssertionState {
     setExpectedValue(val) {
         externalFormatData_setData(this.assertion.expected, val);
     }
+    get supportsGeneratingAssertion() {
+        return true;
+    }
     generateExpected(status) {
         if (status instanceof EqualToJsonAssertFail) {
             const expected = status.actual;
             this.setExpectedValue(expected);
+            return true;
         }
+        return false;
     }
     generateBare() {
         const bareAssertion = new EqualToJson();
@@ -175,11 +181,14 @@ export class EqualToJsonAssertionState extends TestAssertionState {
     }
 }
 export class UnsupportedAssertionState extends TestAssertionState {
+    get supportsGeneratingAssertion() {
+        return false;
+    }
     generateBare() {
         throw new UnsupportedOperationError();
     }
     generateExpected(status) {
-        return;
+        return false;
     }
     label() {
         return 'Unsupported';
@@ -191,7 +200,7 @@ export class TestAssertionEditorState {
     assertionState;
     assertionResultState;
     assertion;
-    selectedTab = TEST_ASSERTION_TAB.SETUP;
+    selectedTab = TEST_ASSERTION_TAB.EXPECTED;
     generatingExpectedAction = ActionState.create();
     constructor(editorStore, assertion, testState) {
         makeObservable(this, {
@@ -211,18 +220,52 @@ export class TestAssertionEditorState {
     }
     *generateExpected() {
         try {
+            assertTrue(this.assertionState.supportsGeneratingAssertion, 'Assertion does not support generation');
             this.generatingExpectedAction.inProgress();
-            const bare = this.assertionState.generateBare();
-            bare.parentTest = this.assertion.parentTest;
-            const assertFail = (yield this.editorStore.graphManagerState.graphManager.generateExpectedResult(this.testState.testable, this.testState.test, bare, this.editorStore.graphManagerState.graph));
-            this.assertionState.generateExpected(assertFail);
+            const result = (yield flowResult(this.testState.fetchTestResult()));
+            let testExecuted;
+            if (result instanceof TestExecuted) {
+                testExecuted = result;
+            }
+            else if (result instanceof MultiExecutionServiceTestResult) {
+                testExecuted = guaranteeNonNullable(Array.from(result.keyIndexedTestResults.values())
+                    .map((testResult) => {
+                    if (testResult instanceof TestExecuted) {
+                        return testResult;
+                    }
+                    else if (testResult instanceof TestError) {
+                        throw new IllegalStateError(testResult.error);
+                    }
+                    return undefined;
+                })
+                    .filter(isNonNullable)[0], 'Unable to derive expected result from test result');
+            }
+            else {
+                throw new UnsupportedOperationError('Unable to derive expected result from test result');
+            }
+            // if test is passing, update UI and return
+            // if test errors report error
+            if (isTestPassing(testExecuted)) {
+                this.testState.handleTestResult(testExecuted);
+                return;
+            }
+            else if (testExecuted instanceof TestError) {
+                throw new IllegalStateError(testExecuted.error);
+            }
+            const assertionStatus = testExecuted.assertStatuses.find((aStatus) => aStatus.assertion.id === this.assertion.id &&
+                aStatus instanceof AssertFail);
+            const assertFail = guaranteeType(assertionStatus, AssertFail, 'Unable to derive expected result from test result');
+            const generated = this.assertionState.generateExpected(assertFail);
+            if (generated) {
+                this.setSelectedTab(TEST_ASSERTION_TAB.EXPECTED);
+            }
             this.generatingExpectedAction.complete();
             this.editorStore.applicationStore.notificationService.notifySuccess(`Expected results generated!`);
         }
         catch (error) {
             assertErrorThrown(error);
             this.editorStore.applicationStore.notificationService.notifyError(`Error generating expected result, please check data input: ${error.message}.`);
-            this.setSelectedTab(TEST_ASSERTION_TAB.SETUP);
+            this.setSelectedTab(TEST_ASSERTION_TAB.EXPECTED);
             this.generatingExpectedAction.fail();
         }
     }

package/lib/stores/editor/editor-state/element-editor-state/testable/TestAssertionState.js.map

@@ -1 +1 @@
-{"version":3,"file":"TestAssertionState.js","sourceRoot":"","sources":["../../../../../../src/stores/editor/editor-state/element-editor-state/testable/TestAssertionState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAGL,UAAU,EAEV,YAAY,EACZ,SAAS,EACT,WAAW,EACX,kBAAkB,EAClB,qBAAqB,EACrB,+BAA+B,EAC/B,UAAU,EACV,mBAAmB,GACpB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAEL,WAAW,EACX,iBAAiB,EACjB,WAAW,EACX,yBAAyB,GAC1B,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,MAAM,CAAC;AAEhE,OAAO,EAAE,0BAA0B,EAAE,MAAM,4DAA4D,CAAC;AACxG,OAAO,EACL,oCAAoC,EACpC,eAAe,GAChB,MAAM,0DAA0D,CAAC;AAGlE,MAAM,CAAN,IAAY,kBAGX;AAHD,WAAY,kBAAkB;IAC5B,qCAAe,CAAA;IACf,uCAAiB,CAAA;AACnB,CAAC,EAHW,kBAAkB,KAAlB,kBAAkB,QAG7B;AAED,MAAM,OAAgB,wBAAwB;IAC5C,WAAW,CAA2B;IACtC,MAAM,CAAkB;IAExB,YAAY,WAAqC,EAAE,MAAuB;QACxE,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;CACF;AAED,MAAM,OAAO,eAAgB,SAAQ,wBAAwB;IAG3D,YAAY,WAAqC,EAAE,MAAkB;QACnE,KAAK,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,cAAc,CAAC,IAAI,EAAE;YACnB,MAAM,EAAE,UAAU;SACnB,CAAC,CAAC;IACL,CAAC;CACF;AAED,MAAM,OAAO,0BAA2B,SAAQ,eAAe;IAE7D,SAAS,GAAG,KAAK,CAAC;IAElB,YACE,WAAqC,EACrC,MAA6B;QAE7B,KAAK,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,cAAc,CAAC,IAAI,EAAE;YACnB,SAAS,EAAE,UAAU;YACrB,YAAY,EAAE,MAAM;SACrB,CAAC,CAAC;IACL,CAAC;IAED,YAAY,CAAC,GAAY;QACvB,IAAI,CAAC,SAAS,GAAG,GAAG,CAAC;IACvB,CAAC;CACF;AAED,MAAM,OAAO,+BAAgC,SAAQ,wBAAwB;CAAG;AAEhF,MAAM,OAAO,wBAAwB;IACnC,UAAU,CAAyB;IACnC,WAAW,CAGG;IACL,WAAW,CAAc;IACzB,cAAc,CAA2B;IAClD,YACE,WAAwB,EACxB,cAAwC;QAExC,cAAc,CAAC,IAAI,EAAE;YACnB,UAAU,EAAE,UAAU;YACtB,aAAa,EAAE,MAAM;YACrB,WAAW,EAAE,UAAU;SACxB,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;IACvC,CAAC;IAED,aAAa,CAAC,GAA2B;QACvC,IAAI,CAAC,UAAU,GAAG,GAAG,CAAC;QACtB,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;QAC7B,IAAI,GAAG,YAAY,YAAY,EAAE;YAC/B,MAAM,MAAM,GAAG,GAAG,CAAC,cAAc,CAAC,IAAI,CACpC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,SAAS,KAAK,IAAI,CAAC,cAAc,CAAC,SAAS,CACjE,CAAC;YACF,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;SAC7C;aAAM,IAAI,GAAG,YAAY,+BAA+B,EAAE;YACzD,MAAM,SAAS,GAAG,IAAI,GAAG,EAAoC,CAAC;YAC9D,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,qBAAqB,CAAC,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE;gBACtE,MAAM,WAAW,GAAG,IAAI,wBAAwB,CAC9C,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,cAAc,CACpB,CAAC;gBACF,WAAW,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC1C,SAAS,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC;YAC7C,CAAC,CAAC,CAAC;YACH,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;SAC9B;IACH,CAAC;IAED,WAAW,CACT,GAAgC;QAEhC,IAAI,GAAG,EAAE;YACP,IAAI,GAAG,YAAY,qBAAqB,EAAE;gBACxC,OAAO,IAAI,0BAA0B,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;aAClD;YACD,IAAI,GAAG,YAAY,UAAU,EAAE;gBAC7B,OAAO,IAAI,eAAe,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;aACvC;YACD,OAAO,IAAI,+BAA+B,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;SACvD;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,MAAM;QACR,IAAI,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,iBAAiB,CAAC,YAAY,EAAE;YAChE,OAAO,eAAe,CAAC,WAAW,CAAC;SACpC;QACD,IAAI,IAAI,CAAC,UAAU,YAAY,SAAS,EAAE;YACxC,OAAO,eAAe,CAAC,KAAK,CAAC;SAC9B;aAAM,IACL,IAAI,CAAC,UAAU,YAAY,YAAY;YACvC,IAAI,CAAC,UAAU,CAAC,mBAAmB,KAAK,mBAAmB,CAAC,IAAI,EAChE;YACA,OAAO,eAAe,CAAC,MAAM,CAAC;SAC/B;aAAM,IACL,IAAI,CAAC,UAAU,YAAY,YAAY;YACvC,IAAI,CAAC,UAAU,CAAC,mBAAmB,KAAK,mBAAmB,CAAC,IAAI;YAChE,IAAI,CAAC,WAAW,YAAY,wBAAwB,EACpD;YACA,OAAO,oCAAoC,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;SACtE;aAAM,IAAI,IAAI,CAAC,UAAU,YAAY,+BAA+B,EAAE;YACrE,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CACvB,IAAI,CAAC,UAAU,CAAC,qBAAqB,CAAC,OAAO,EAAE,CAChD,CAAC,KAAK,CAAC,CAAC,SAAS,EAAE,EAAE;gBACpB,MAAM,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;gBAC5B,IACE,MAAM,YAAY,YAAY;oBAC9B,MAAM,CAAC,mBAAmB,KAAK,mBAAmB,CAAC,IAAI,EACvD;oBACA,OAAO,IAAI,CAAC;iBACb;gBACD,IACE,MAAM,YAAY,YAAY;oBAC9B,MAAM,CAAC,mBAAmB,KAAK,mBAAmB,CAAC,IAAI,EACvD;oBACA,MAAM,MAAM,GAAG,MAAM,CAAC,cAAc,CAAC,IAAI,CACvC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,SAAS,KAAK,IAAI,CAAC,cAAc,CAAC,SAAS,CACjE,CAAC;oBACF,IAAI,MAAM,YAAY,UAAU,EAAE;wBAChC,OAAO,IAAI,CAAC;qBACb;iBACF;gBACD,OAAO,KAAK,CAAC;YACf,CAAC,CAAC,CAAC;YACH,IAAI,MAAM,EAAE;gBACV,OAAO,eAAe,CAAC,MAAM,CAAC;aAC/B;YACD,MAAM,eAAe,GAAG,KAAK,CAAC,IAAI,CAChC,IAAI,CAAC,UAAU,CAAC,qBAAqB,CAAC,MAAM,EAAE,CAC/C,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,YAAY,SAAS,CAAC,CAAC;YACtC,IAAI,eAAe,EAAE;gBACnB,OAAO,eAAe,CAAC,KAAK,CAAC;aAC9B;YACD,OAAO,eAAe,CAAC,MAAM,CAAC;SAC/B;QACD,OAAO,eAAe,CAAC,WAAW,CAAC;IACrC,CAAC;CACF;AACD,MAAM,OAAgB,kBAAkB;IAC7B,WAAW,CAAc;IAClC,SAAS,CAAgB;IACzB,MAAM,CAA2B;IAEjC,YACE,WAAwB,EACxB,cAAwC;QAExC,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,cAAc,CAAC,SAAS,CAAC;QAC1C,IAAI,CAAC,MAAM,GAAG,IAAI,wBAAwB,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC;IAC1E,CAAC;CAOF;AAED,MAAM,OAAO,yBAA0B,SAAQ,kBAAkB;IAG/D,gBAAgB,CAAC,GAAW;QAC1B,0BAA0B,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;IAC3D,CAAC;IAED,gBAAgB,CAAC,MAAkB;QACjC,IAAI,MAAM,YAAY,qBAAqB,EAAE;YAC3C,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC;YAC/B,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;SACjC;IACH,CAAC;IACD,YAAY;QACV,MAAM,aAAa,GAAG,IAAI,WAAW,EAAE,CAAC;QACxC,aAAa,CAAC,QAAQ,GAAG,IAAI,kBAAkB,EAAE,CAAC;QAClD,aAAa,CAAC,QAAQ,CAAC,WAAW,GAAG,WAAW,CAAC,gBAAgB,CAAC;QAClE,aAAa,CAAC,QAAQ,CAAC,IAAI,GAAG,EAAE,CAAC;QACjC,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,KAAK;QACH,OAAO,aAAa,CAAC;IACvB,CAAC;CACF;AAED,MAAM,OAAO,yBAA0B,SAAQ,kBAAkB;IAC/D,YAAY;QACV,MAAM,IAAI,yBAAyB,EAAE,CAAC;IACxC,CAAC;IACD,gBAAgB,CAAC,MAAkB;QACjC,OAAO;IACT,CAAC;IAED,KAAK;QACH,OAAO,aAAa,CAAC;IACvB,CAAC;CACF;AAED,MAAM,OAAO,wBAAwB;IAC1B,WAAW,CAAc;IACzB,SAAS,CAA0B;IAC5C,cAAc,CAAqB;IACnC,oBAAoB,CAA2B;IAC/C,SAAS,CAAgB;IACzB,WAAW,GAAG,kBAAkB,CAAC,KAAK,CAAC;IACvC,wBAAwB,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC;IAChD,YACE,WAAwB,EACxB,SAAwB,EACxB,SAAkC;QAElC,cAAc,CAAC,IAAI,EAAE;YACnB,WAAW,EAAE,UAAU;YACvB,oBAAoB,EAAE,UAAU;YAChC,cAAc,EAAE,MAAM;YACtB,gBAAgB,EAAE,IAAI;SACvB,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAC1D,IAAI,CAAC,oBAAoB,GAAG,IAAI,wBAAwB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IAC9E,CAAC;IAED,cAAc,CAAC,GAAuB;QACpC,IAAI,CAAC,WAAW,GAAG,GAAG,CAAC;IACzB,CAAC;IAED,CAAC,gBAAgB;QACf,IAAI;YACF,IAAI,CAAC,wBAAwB,CAAC,UAAU,EAAE,CAAC;YAC3C,MAAM,IAAI,GAAG,IAAI,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;YAChD,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC;YAC5C,MAAM,UAAU,GACd,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,YAAY,CAAC,sBAAsB,CAC3E,IAAI,CAAC,SAAS,CAAC,QAAQ,EACvB,IAAI,CAAC,SAAS,CAAC,IAAI,EAEnB,IAAI,EACJ,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,KAAK,CACzC,CAAe,CAAC;YACnB,IAAI,CAAC,cAAc,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;YACjD,IAAI,CAAC,wBAAwB,CAAC,QAAQ,EAAE,CAAC;YACzC,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,aAAa,CACjE,6BAA6B,CAC9B,CAAC;SACH;QAAC,OAAO,KAAK,EAAE;YACd,iBAAiB,CAAC,KAAK,CAAC,CAAC;YACzB,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,WAAW,CAC/D,8DAA8D,KAAK,CAAC,OAAO,GAAG,CAC/E,CAAC;YACF,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;YAC9C,IAAI,CAAC,wBAAwB,CAAC,IAAI,EAAE,CAAC;SACtC;IACH,CAAC;IAED,mBAAmB,CAAC,SAAwB;QAC1C,IAAI,SAAS,YAAY,WAAW,EAAE;YACpC,OAAO,IAAI,yBAAyB,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;SAC9D;QACD,OAAO,IAAI,yBAAyB,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IAC/D,CAAC;CACF"}
+{"version":3,"file":"TestAssertionState.js","sourceRoot":"","sources":["../../../../../../src/stores/editor/editor-state/element-editor-state/testable/TestAssertionState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAGL,UAAU,EAEV,YAAY,EACZ,SAAS,EACT,WAAW,EACX,kBAAkB,EAClB,qBAAqB,EACrB,+BAA+B,EAC/B,UAAU,EACV,mBAAmB,GACpB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAEL,WAAW,EACX,iBAAiB,EACjB,WAAW,EACX,yBAAyB,EACzB,UAAU,EACV,aAAa,EACb,iBAAiB,EACjB,oBAAoB,EACpB,aAAa,GACd,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,MAAM,CAAC;AAE5E,OAAO,EAAE,0BAA0B,EAAE,MAAM,4DAA4D,CAAC;AACxG,OAAO,EACL,oCAAoC,EACpC,eAAe,GAChB,MAAM,0DAA0D,CAAC;AAElE,OAAO,EAAE,aAAa,EAAE,MAAM,iCAAiC,CAAC;AAEhE,MAAM,CAAN,IAAY,kBAGX;AAHD,WAAY,kBAAkB;IAC5B,2CAAqB,CAAA;IACrB,uCAAiB,CAAA;AACnB,CAAC,EAHW,kBAAkB,KAAlB,kBAAkB,QAG7B;AAED,MAAM,OAAgB,wBAAwB;IAC5C,WAAW,CAA2B;IACtC,MAAM,CAAkB;IAExB,YAAY,WAAqC,EAAE,MAAuB;QACxE,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;CACF;AAED,MAAM,OAAO,eAAgB,SAAQ,wBAAwB;IAG3D,YAAY,WAAqC,EAAE,MAAkB;QACnE,KAAK,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,cAAc,CAAC,IAAI,EAAE;YACnB,MAAM,EAAE,UAAU;SACnB,CAAC,CAAC;IACL,CAAC;CACF;AAED,MAAM,OAAO,0BAA2B,SAAQ,eAAe;IAE7D,SAAS,GAAG,KAAK,CAAC;IAElB,YACE,WAAqC,EACrC,MAA6B;QAE7B,KAAK,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC;QAC3B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,cAAc,CAAC,IAAI,EAAE;YACnB,SAAS,EAAE,UAAU;YACrB,YAAY,EAAE,MAAM;SACrB,CAAC,CAAC;IACL,CAAC;IAED,YAAY,CAAC,GAAY;QACvB,IAAI,CAAC,SAAS,GAAG,GAAG,CAAC;IACvB,CAAC;CACF;AAED,MAAM,OAAO,+BAAgC,SAAQ,wBAAwB;CAAG;AAEhF,MAAM,OAAO,wBAAwB;IACnC,UAAU,CAAyB;IACnC,WAAW,CAGG;IACL,WAAW,CAAc;IACzB,cAAc,CAA2B;IAClD,YACE,WAAwB,EACxB,cAAwC;QAExC,cAAc,CAAC,IAAI,EAAE;YACnB,UAAU,EAAE,UAAU;YACtB,aAAa,EAAE,MAAM;YACrB,WAAW,EAAE,UAAU;SACxB,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;IACvC,CAAC;IAED,aAAa,CAAC,GAA2B;QACvC,IAAI,CAAC,UAAU,GAAG,GAAG,CAAC;QACtB,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;QAC7B,IAAI,GAAG,YAAY,YAAY,EAAE;YAC/B,MAAM,MAAM,GAAG,GAAG,CAAC,cAAc,CAAC,IAAI,CACpC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,SAAS,KAAK,IAAI,CAAC,cAAc,CAAC,SAAS,CACjE,CAAC;YACF,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;SAC7C;aAAM,IAAI,GAAG,YAAY,+BAA+B,EAAE;YACzD,MAAM,SAAS,GAAG,IAAI,GAAG,EAAoC,CAAC;YAC9D,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,qBAAqB,CAAC,OAAO,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,WAAW,EAAE,EAAE;gBACtE,MAAM,WAAW,GAAG,IAAI,wBAAwB,CAC9C,IAAI,CAAC,WAAW,EAChB,IAAI,CAAC,cAAc,CACpB,CAAC;gBACF,WAAW,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC1C,SAAS,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC;YAC7C,CAAC,CAAC,CAAC;YACH,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;SAC9B;IACH,CAAC;IAED,WAAW,CACT,GAAgC;QAEhC,IAAI,GAAG,EAAE;YACP,IAAI,GAAG,YAAY,qBAAqB,EAAE;gBACxC,OAAO,IAAI,0BAA0B,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;aAClD;YACD,IAAI,GAAG,YAAY,UAAU,EAAE;gBAC7B,OAAO,IAAI,eAAe,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;aACvC;YACD,OAAO,IAAI,+BAA+B,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;SACvD;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,IAAI,MAAM;QACR,IAAI,IAAI,CAAC,cAAc,CAAC,SAAS,CAAC,iBAAiB,CAAC,YAAY,EAAE;YAChE,OAAO,eAAe,CAAC,WAAW,CAAC;SACpC;QACD,IAAI,IAAI,CAAC,UAAU,YAAY,SAAS,EAAE;YACxC,OAAO,eAAe,CAAC,KAAK,CAAC;SAC9B;aAAM,IACL,IAAI,CAAC,UAAU,YAAY,YAAY;YACvC,IAAI,CAAC,UAAU,CAAC,mBAAmB,KAAK,mBAAmB,CAAC,IAAI,EAChE;YACA,OAAO,eAAe,CAAC,MAAM,CAAC;SAC/B;aAAM,IACL,IAAI,CAAC,UAAU,YAAY,YAAY;YACvC,IAAI,CAAC,UAAU,CAAC,mBAAmB,KAAK,mBAAmB,CAAC,IAAI;YAChE,IAAI,CAAC,WAAW,YAAY,wBAAwB,EACpD;YACA,OAAO,oCAAoC,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;SACtE;aAAM,IAAI,IAAI,CAAC,UAAU,YAAY,+BAA+B,EAAE;YACrE,MAAM,MAAM,GAAG,KAAK,CAAC,IAAI,CACvB,IAAI,CAAC,UAAU,CAAC,qBAAqB,CAAC,OAAO,EAAE,CAChD,CAAC,KAAK,CAAC,CAAC,SAAS,EAAE,EAAE;gBACpB,MAAM,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;gBAC5B,IACE,MAAM,YAAY,YAAY;oBAC9B,MAAM,CAAC,mBAAmB,KAAK,mBAAmB,CAAC,IAAI,EACvD;oBACA,OAAO,IAAI,CAAC;iBACb;gBACD,IACE,MAAM,YAAY,YAAY;oBAC9B,MAAM,CAAC,mBAAmB,KAAK,mBAAmB,CAAC,IAAI,EACvD;oBACA,MAAM,MAAM,GAAG,MAAM,CAAC,cAAc,CAAC,IAAI,CACvC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,SAAS,KAAK,IAAI,CAAC,cAAc,CAAC,SAAS,CACjE,CAAC;oBACF,IAAI,MAAM,YAAY,UAAU,EAAE;wBAChC,OAAO,IAAI,CAAC;qBACb;iBACF;gBACD,OAAO,KAAK,CAAC;YACf,CAAC,CAAC,CAAC;YACH,IAAI,MAAM,EAAE;gBACV,OAAO,eAAe,CAAC,MAAM,CAAC;aAC/B;YACD,MAAM,eAAe,GAAG,KAAK,CAAC,IAAI,CAChC,IAAI,CAAC,UAAU,CAAC,qBAAqB,CAAC,MAAM,EAAE,CAC/C,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,YAAY,SAAS,CAAC,CAAC;YACtC,IAAI,eAAe,EAAE;gBACnB,OAAO,eAAe,CAAC,KAAK,CAAC;aAC9B;YACD,OAAO,eAAe,CAAC,MAAM,CAAC;SAC/B;QACD,OAAO,eAAe,CAAC,WAAW,CAAC;IACrC,CAAC;CACF;AACD,MAAM,OAAgB,kBAAkB;IAC7B,WAAW,CAAc;IAClC,SAAS,CAAgB;IACzB,MAAM,CAA2B;IAEjC,YACE,WAAwB,EACxB,cAAwC;QAExC,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,cAAc,CAAC,SAAS,CAAC;QAC1C,IAAI,CAAC,MAAM,GAAG,IAAI,wBAAwB,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC;IAC1E,CAAC;CASF;AAED,MAAM,OAAO,yBAA0B,SAAQ,kBAAkB;IAG/D,gBAAgB,CAAC,GAAW;QAC1B,0BAA0B,CAAC,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;IAC3D,CAAC;IAED,IAAa,2BAA2B;QACtC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,gBAAgB,CAAC,MAAkB;QACjC,IAAI,MAAM,YAAY,qBAAqB,EAAE;YAC3C,MAAM,QAAQ,GAAG,MAAM,CAAC,MAAM,CAAC;YAC/B,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC;YAChC,OAAO,IAAI,CAAC;SACb;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IACD,YAAY;QACV,MAAM,aAAa,GAAG,IAAI,WAAW,EAAE,CAAC;QACxC,aAAa,CAAC,QAAQ,GAAG,IAAI,kBAAkB,EAAE,CAAC;QAClD,aAAa,CAAC,QAAQ,CAAC,WAAW,GAAG,WAAW,CAAC,gBAAgB,CAAC;QAClE,aAAa,CAAC,QAAQ,CAAC,IAAI,GAAG,EAAE,CAAC;QACjC,OAAO,aAAa,CAAC;IACvB,CAAC;IAED,KAAK;QACH,OAAO,aAAa,CAAC;IACvB,CAAC;CACF;AAED,MAAM,OAAO,yBAA0B,SAAQ,kBAAkB;IAC/D,IAAa,2BAA2B;QACtC,OAAO,KAAK,CAAC;IACf,CAAC;IACD,YAAY;QACV,MAAM,IAAI,yBAAyB,EAAE,CAAC;IACxC,CAAC;IACD,gBAAgB,CAAC,MAAkB;QACjC,OAAO,KAAK,CAAC;IACf,CAAC;IAED,KAAK;QACH,OAAO,aAAa,CAAC;IACvB,CAAC;CACF;AAED,MAAM,OAAO,wBAAwB;IAC1B,WAAW,CAAc;IACzB,SAAS,CAA0B;IAC5C,cAAc,CAAqB;IACnC,oBAAoB,CAA2B;IAC/C,SAAS,CAAgB;IACzB,WAAW,GAAG,kBAAkB,CAAC,QAAQ,CAAC;IAC1C,wBAAwB,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC;IAChD,YACE,WAAwB,EACxB,SAAwB,EACxB,SAAkC;QAElC,cAAc,CAAC,IAAI,EAAE;YACnB,WAAW,EAAE,UAAU;YACvB,oBAAoB,EAAE,UAAU;YAChC,cAAc,EAAE,MAAM;YACtB,gBAAgB,EAAE,IAAI;SACvB,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;QAC1D,IAAI,CAAC,oBAAoB,GAAG,IAAI,wBAAwB,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IAC9E,CAAC;IAED,cAAc,CAAC,GAAuB;QACpC,IAAI,CAAC,WAAW,GAAG,GAAG,CAAC;IACzB,CAAC;IAED,CAAC,gBAAgB;QACf,IAAI;YACF,UAAU,CACR,IAAI,CAAC,cAAc,CAAC,2BAA2B,EAC/C,uCAAuC,CACxC,CAAC;YACF,IAAI,CAAC,wBAAwB,CAAC,UAAU,EAAE,CAAC;YAC3C,MAAM,MAAM,GAAG,CAAC,MAAM,UAAU,CAC9B,IAAI,CAAC,SAAS,CAAC,eAAe,EAAE,CACjC,CAAe,CAAC;YACjB,IAAI,YAA0B,CAAC;YAC/B,IAAI,MAAM,YAAY,YAAY,EAAE;gBAClC,YAAY,GAAG,MAAM,CAAC;aACvB;iBAAM,IAAI,MAAM,YAAY,+BAA+B,EAAE;gBAC5D,YAAY,GAAG,oBAAoB,CACjC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,qBAAqB,CAAC,MAAM,EAAE,CAAC;qBAC9C,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;oBAClB,IAAI,UAAU,YAAY,YAAY,EAAE;wBACtC,OAAO,UAAU,CAAC;qBACnB;yBAAM,IAAI,UAAU,YAAY,SAAS,EAAE;wBAC1C,MAAM,IAAI,iBAAiB,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;qBAC/C;oBACD,OAAO,SAAS,CAAC;gBACnB,CAAC,CAAC;qBACD,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,EAC3B,mDAAmD,CACpD,CAAC;aACH;iBAAM;gBACL,MAAM,IAAI,yBAAyB,CACjC,mDAAmD,CACpD,CAAC;aACH;YACD,2CAA2C;YAC3C,8BAA8B;YAC9B,IAAI,aAAa,CAAC,YAAY,CAAC,EAAE;gBAC/B,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,YAAY,CAAC,CAAC;gBAC9C,OAAO;aACR;iBAAM,IAAI,YAAY,YAAY,SAAS,EAAE;gBAC5C,MAAM,IAAI,iBAAiB,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;aACjD;YACD,MAAM,eAAe,GAAG,YAAY,CAAC,cAAc,CAAC,IAAI,CACtD,CAAC,OAAO,EAAE,EAAE,CACV,OAAO,CAAC,SAAS,CAAC,EAAE,KAAK,IAAI,CAAC,SAAS,CAAC,EAAE;gBAC1C,OAAO,YAAY,UAAU,CAChC,CAAC;YACF,MAAM,UAAU,GAAG,aAAa,CAC9B,eAAe,EACf,UAAU,EACV,mDAAmD,CACpD,CAAC;YACF,MAAM,SAAS,GAAG,IAAI,CAAC,cAAc,CAAC,gBAAgB,CAAC,UAAU,CAAC,CAAC;YAEnE,IAAI,SAAS,EAAE;gBACb,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;aAClD;YACD,IAAI,CAAC,wBAAwB,CAAC,QAAQ,EAAE,CAAC;YACzC,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,aAAa,CACjE,6BAA6B,CAC9B,CAAC;SACH;QAAC,OAAO,KAAK,EAAE;YACd,iBAAiB,CAAC,KAAK,CAAC,CAAC;YACzB,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,WAAW,CAC/D,8DAA8D,KAAK,CAAC,OAAO,GAAG,CAC/E,CAAC;YACF,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YACjD,IAAI,CAAC,wBAAwB,CAAC,IAAI,EAAE,CAAC;SACtC;IACH,CAAC;IAED,mBAAmB,CAAC,SAAwB;QAC1C,IAAI,SAAS,YAAY,WAAW,EAAE;YACpC,OAAO,IAAI,yBAAyB,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;SAC9D;QACD,OAAO,IAAI,yBAAyB,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;IAC/D,CAAC;CACF"}

package/lib/stores/editor/editor-state/element-editor-state/testable/TestableEditorState.d.ts

@@ -46,6 +46,7 @@ export declare class TestableTestEditorState {
     deleteAssertion(assertionState: TestAssertionEditorState): void;
     openAssertion(val: TestAssertion): void;
     runTest(): GeneratorFn<void>;
+    fetchTestResult(): Promise<TestResult>;
     resetResult(): void;
     handleTestResult(testResult: TestResult): void;
     get assertionCount(): number;
@@ -58,7 +59,7 @@ export declare class TestableTestSuiteEditorState {
     suite: TestSuite;
     isReadOnly: boolean;
     testStates: TestableTestEditorState[];
-    isRunningTest: ActionState;
+    runningSuiteState: ActionState;
     selectTestState: TestableTestEditorState | undefined;
     constructor(testable: Testable, suite: TestSuite, isReadOnly: boolean, editorStore: EditorStore);
     runSuite(): GeneratorFn<void>;

package/lib/stores/editor/editor-state/element-editor-state/testable/TestableEditorState.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"TestableEditorState.d.ts","sourceRoot":"","sources":["../../../../../../src/stores/editor/editor-state/element-editor-state/testable/TestableEditorState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,KAAK,UAAU,EACf,KAAK,QAAQ,EACb,KAAK,aAAa,EAClB,KAAK,UAAU,EAIf,SAAS,EAGV,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,KAAK,WAAW,EAIhB,WAAW,EAIZ,MAAM,sBAAsB,CAAC;AAE9B,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAI3D,OAAO,EACL,wBAAwB,EAEzB,MAAM,yBAAyB,CAAC;AAEjC,qBAAa,uBAAuB;IAClC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAClC,QAAQ,CAAC,SAAS,EAAE,uBAAuB,CAAC;IAC5C,MAAM,EAAE,UAAU,GAAG,SAAS,CAAC;gBAEnB,SAAS,EAAE,uBAAuB,EAAE,WAAW,EAAE,WAAW;IAUxE,SAAS,CAAC,GAAG,EAAE,UAAU,GAAG,SAAS,GAAG,IAAI;CAG7C;AAED,oBAAY,iBAAiB;IAC3B,KAAK,UAAU;IACf,SAAS,cAAc;CACxB;AAED,qBAAa,uBAAuB;IAClC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAClC,QAAQ,EAAE,QAAQ,CAAC;IACnB,IAAI,EAAE,UAAU,CAAC;IACjB,qBAAqB,EAAE,wBAAwB,GAAG,SAAS,CAAC;IAC5D,qBAAqB,EAAE,wBAAwB,EAAE,CAAM;IACvD,WAAW,oBAA+B;IAC1C,iBAAiB,EAAE,aAAa,GAAG,SAAS,CAAC;IAC7C,iBAAiB,cAAwB;IACzC,eAAe,EAAE,uBAAuB,CAAC;IACzC,UAAU,EAAE,OAAO,CAAC;gBAGlB,QAAQ,EAAE,QAAQ,EAClB,IAAI,EAAE,UAAU,EAChB,UAAU,EAAE,OAAO,EACnB,WAAW,EAAE,WAAW;IAc1B,cAAc,CAAC,GAAG,EAAE,iBAAiB,GAAG,IAAI;IAI5C,oBAAoB,CAAC,SAAS,EAAE,aAAa,GAAG,SAAS,GAAG,IAAI;IAIhE,YAAY,IAAI,IAAI;IAapB,eAAe,CAAC,cAAc,EAAE,wBAAwB,GAAG,IAAI;IAQ/D,aAAa,CAAC,GAAG,EAAE,aAAa,GAAG,IAAI;IAOtC,OAAO,IAAI,WAAW,CAAC,IAAI,CAAC;IA+B7B,WAAW,IAAI,IAAI;IAOnB,gBAAgB,CAAC,UAAU,EAAE,UAAU,GAAG,IAAI;IAQ9C,IAAI,cAAc,IAAI,MAAM,CAE3B;IAED,IAAI,eAAe,IAAI,MAAM,CAW5B;IAED,IAAI,eAAe,IAAI,MAAM,CAW5B;CACF;AAED,qBAAa,4BAA4B;IACvC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAClC,QAAQ,EAAE,QAAQ,CAAC;IACnB,KAAK,EAAE,SAAS,CAAC;IACjB,UAAU,EAAE,OAAO,CAAC;IACpB,UAAU,EAAE,uBAAuB,EAAE,CAAM;IAC3C,aAAa,cAAwB;IACrC,eAAe,EAAE,uBAAuB,GAAG,SAAS,CAAC;gBAGnD,QAAQ,EAAE,QAAQ,EAClB,KAAK,EAAE,SAAS,EAChB,UAAU,EAAE,OAAO,EACnB,WAAW,EAAE,WAAW;IAQzB,QAAQ,IAAI,WAAW,CAAC,IAAI,CAAC;IA4B7B,eAAe,IAAI,WAAW,CAAC,IAAI,CAAC;CAoCtC"}
+{"version":3,"file":"TestableEditorState.d.ts","sourceRoot":"","sources":["../../../../../../src/stores/editor/editor-state/element-editor-state/testable/TestableEditorState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EACL,KAAK,UAAU,EACf,KAAK,QAAQ,EACb,KAAK,aAAa,EAClB,KAAK,UAAU,EAIf,SAAS,EAGV,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACL,KAAK,WAAW,EAIhB,WAAW,EAIZ,MAAM,sBAAsB,CAAC;AAE9B,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAI3D,OAAO,EACL,wBAAwB,EAEzB,MAAM,yBAAyB,CAAC;AAEjC,qBAAa,uBAAuB;IAClC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAClC,QAAQ,CAAC,SAAS,EAAE,uBAAuB,CAAC;IAC5C,MAAM,EAAE,UAAU,GAAG,SAAS,CAAC;gBAEnB,SAAS,EAAE,uBAAuB,EAAE,WAAW,EAAE,WAAW;IAUxE,SAAS,CAAC,GAAG,EAAE,UAAU,GAAG,SAAS,GAAG,IAAI;CAG7C;AAED,oBAAY,iBAAiB;IAC3B,KAAK,UAAU;IACf,SAAS,cAAc;CACxB;AAED,qBAAa,uBAAuB;IAClC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAClC,QAAQ,EAAE,QAAQ,CAAC;IACnB,IAAI,EAAE,UAAU,CAAC;IACjB,qBAAqB,EAAE,wBAAwB,GAAG,SAAS,CAAC;IAC5D,qBAAqB,EAAE,wBAAwB,EAAE,CAAM;IACvD,WAAW,oBAA+B;IAC1C,iBAAiB,EAAE,aAAa,GAAG,SAAS,CAAC;IAC7C,iBAAiB,cAAwB;IACzC,eAAe,EAAE,uBAAuB,CAAC;IACzC,UAAU,EAAE,OAAO,CAAC;gBAGlB,QAAQ,EAAE,QAAQ,EAClB,IAAI,EAAE,UAAU,EAChB,UAAU,EAAE,OAAO,EACnB,WAAW,EAAE,WAAW;IAc1B,cAAc,CAAC,GAAG,EAAE,iBAAiB,GAAG,IAAI;IAI5C,oBAAoB,CAAC,SAAS,EAAE,aAAa,GAAG,SAAS,GAAG,IAAI;IAIhE,YAAY,IAAI,IAAI;IAapB,eAAe,CAAC,cAAc,EAAE,wBAAwB,GAAG,IAAI;IAQ/D,aAAa,CAAC,GAAG,EAAE,aAAa,GAAG,IAAI;IAOtC,OAAO,IAAI,WAAW,CAAC,IAAI,CAAC;IAiBvB,eAAe,IAAI,OAAO,CAAC,UAAU,CAAC;IAmB5C,WAAW,IAAI,IAAI;IAOnB,gBAAgB,CAAC,UAAU,EAAE,UAAU,GAAG,IAAI;IAQ9C,IAAI,cAAc,IAAI,MAAM,CAE3B;IAED,IAAI,eAAe,IAAI,MAAM,CAW5B;IAED,IAAI,eAAe,IAAI,MAAM,CAW5B;CACF;AAED,qBAAa,4BAA4B;IACvC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAClC,QAAQ,EAAE,QAAQ,CAAC;IACnB,KAAK,EAAE,SAAS,CAAC;IACjB,UAAU,EAAE,OAAO,CAAC;IACpB,UAAU,EAAE,uBAAuB,EAAE,CAAM;IAC3C,iBAAiB,cAAwB;IACzC,eAAe,EAAE,uBAAuB,GAAG,SAAS,CAAC;gBAGnD,QAAQ,EAAE,QAAQ,EAClB,KAAK,EAAE,SAAS,EAChB,UAAU,EAAE,OAAO,EACnB,WAAW,EAAE,WAAW;IAQzB,QAAQ,IAAI,WAAW,CAAC,IAAI,CAAC;IA4B7B,eAAe,IAAI,WAAW,CAAC,IAAI,CAAC;CAoCtC"}

package/lib/stores/editor/editor-state/element-editor-state/testable/TestableEditorState.js

@@ -15,7 +15,7 @@
  */
 import { TestExecuted, UniqueTestId, RunTestsTestableInput, TestSuite, TestExecutionStatus, TestError, } from '@finos/legend-graph';
 import { assertErrorThrown, assertTrue, guaranteeNonNullable, ActionState, addUniqueEntry, deleteEntry, isNonNullable, } from '@finos/legend-shared';
-import { action, makeObservable, observable } from 'mobx';
+import { action, flowResult, makeObservable, observable } from 'mobx';
 import { atomicTest_addAssertion } from '../../../../graph-modifier/Testable_GraphModifierHelper.js';
 import { createEmptyEqualToJsonAssertion } from '../../../utils/TestableUtils.js';
 import { TESTABLE_RESULT } from '../../../sidebar-state/testable/GlobalTestRunnerState.js';
@@ -92,14 +92,7 @@ export class TestableTestEditorState {
         try {
             this.resetResult();
             this.runningTestAction.inProgress();
-            const input = new RunTestsTestableInput(this.testable);
-            const suite = this.test.__parent instanceof TestSuite
-                ? this.test.__parent
-                : undefined;
-            input.unitTestIds.push(new UniqueTestId(suite, this.test));
-            const testResults = (yield this.editorStore.graphManagerState.graphManager.runTests([input], this.editorStore.graphManagerState.graph));
-            const result = guaranteeNonNullable(testResults[0]);
-            assertTrue(result.testable === this.testable && result.atomicTest === this.test, 'Unexpected test result');
+            const result = (yield flowResult(this.fetchTestResult()));
             this.handleTestResult(result);
             this.runningTestAction.complete();
         }
@@ -109,6 +102,17 @@ export class TestableTestEditorState {
             this.runningTestAction.fail();
         }
     }
+    // Fetches test results. Caller of test should catch the error
+    async fetchTestResult() {
+        const input = new RunTestsTestableInput(this.testable);
+        const suite = this.test.__parent instanceof TestSuite ? this.test.__parent : undefined;
+        input.unitTestIds.push(new UniqueTestId(suite, this.test));
+        const testResults = await this.editorStore.graphManagerState.graphManager.runTests([input], this.editorStore.graphManagerState.graph);
+        const result = guaranteeNonNullable(testResults[0]);
+        assertTrue(result.testable === this.testable &&
+            result.atomicTest.id === this.test.id, 'Unexpected test result');
+        return result;
+    }
     resetResult() {
         this.testResultState.setResult(undefined);
         this.assertionEditorStates.forEach((assertionState) => assertionState.assertionResultState.setTestResult(undefined));
@@ -146,7 +150,7 @@ export class TestableTestSuiteEditorState {
     suite;
     isReadOnly;
     testStates = [];
-    isRunningTest = ActionState.create();
+    runningSuiteState = ActionState.create();
     selectTestState;
     constructor(testable, suite, isReadOnly, editorStore) {
         this.testable = testable;
@@ -156,7 +160,7 @@ export class TestableTestSuiteEditorState {
     }
     *runSuite() {
         try {
-            this.isRunningTest.inProgress();
+            this.runningSuiteState.inProgress();
             this.testStates.forEach((t) => t.resetResult());
             this.testStates.forEach((t) => t.runningTestAction.inProgress());
             const input = new RunTestsTestableInput(this.testable);
@@ -166,12 +170,12 @@ export class TestableTestSuiteEditorState {
                 const state = this.testStates.find((t) => t.test === result.atomicTest);
                 state?.handleTestResult(result);
             });
-            this.isRunningTest.complete();
+            this.runningSuiteState.complete();
         }
         catch (error) {
             assertErrorThrown(error);
             this.editorStore.applicationStore.notificationService.notifyError(error);
-            this.isRunningTest.fail();
+            this.runningSuiteState.fail();
         }
         finally {
             this.testStates.forEach((t) => t.runningTestAction.complete());
@@ -179,7 +183,7 @@ export class TestableTestSuiteEditorState {
     }
     *runFailingTests() {
         try {
-            this.isRunningTest.inProgress();
+            this.runningSuiteState.inProgress();
             const input = new RunTestsTestableInput(this.testable);
             input.unitTestIds = this.testStates
                 .map((testState) => {
@@ -198,12 +202,12 @@ export class TestableTestSuiteEditorState {
                 const state = this.testStates.find((t) => t.test === result.atomicTest);
                 state?.handleTestResult(result);
             });
-            this.isRunningTest.complete();
+            this.runningSuiteState.complete();
         }
         catch (error) {
             assertErrorThrown(error);
             this.editorStore.applicationStore.notificationService.notifyError(error);
-            this.isRunningTest.fail();
+            this.runningSuiteState.fail();
         }
         finally {
             this.testStates.forEach((t) => t.runningTestAction.complete());

package/lib/stores/editor/editor-state/element-editor-state/testable/TestableEditorState.js.map

@@ -1 +1 @@
-{"version":3,"file":"TestableEditorState.js","sourceRoot":"","sources":["../../../../../../src/stores/editor/editor-state/element-editor-state/testable/TestableEditorState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAKL,YAAY,EACZ,YAAY,EACZ,qBAAqB,EACrB,SAAS,EACT,mBAAmB,EACnB,SAAS,GACV,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAEL,iBAAiB,EACjB,UAAU,EACV,oBAAoB,EACpB,WAAW,EACX,cAAc,EACd,WAAW,EACX,aAAa,GACd,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,MAAM,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,MAAM,CAAC;AAE1D,OAAO,EAAE,uBAAuB,EAAE,MAAM,4DAA4D,CAAC;AACrG,OAAO,EAAE,+BAA+B,EAAE,MAAM,iCAAiC,CAAC;AAClF,OAAO,EAAE,eAAe,EAAE,MAAM,0DAA0D,CAAC;AAC3F,OAAO,EACL,wBAAwB,EACxB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AAEjC,MAAM,OAAO,uBAAuB;IACzB,WAAW,CAAc;IACzB,SAAS,CAA0B;IAC5C,MAAM,CAAyB;IAE/B,YAAY,SAAkC,EAAE,WAAwB;QACtE,cAAc,CAAC,IAAI,EAAE;YACnB,MAAM,EAAE,UAAU;YAClB,SAAS,EAAE,MAAM;SAClB,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,SAAS,CAAC,GAA2B;QACnC,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC;IACpB,CAAC;CACF;AAED,MAAM,CAAN,IAAY,iBAGX;AAHD,WAAY,iBAAiB;IAC3B,oCAAe,CAAA;IACf,4CAAuB,CAAA;AACzB,CAAC,EAHW,iBAAiB,KAAjB,iBAAiB,QAG5B;AAED,MAAM,OAAO,uBAAuB;IACzB,WAAW,CAAc;IAClC,QAAQ,CAAW;IACnB,IAAI,CAAa;IACjB,qBAAqB,CAAuC;IAC5D,qBAAqB,GAA+B,EAAE,CAAC;IACvD,WAAW,GAAG,iBAAiB,CAAC,SAAS,CAAC;IAC1C,iBAAiB,CAA4B;IAC7C,iBAAiB,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC;IACzC,eAAe,CAA0B;IACzC,UAAU,CAAU;IAEpB,YACE,QAAkB,EAClB,IAAgB,EAChB,UAAmB,EACnB,WAAwB;QAExB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,eAAe,GAAG,IAAI,uBAAuB,CAAC,IAAI,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QAC3E,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAC9C,CAAC,SAAS,EAAE,EAAE,CACZ,IAAI,wBAAwB,CAAC,IAAI,CAAC,WAAW,EAAE,SAAS,EAAE,IAAI,CAAC,CAClE,CAAC;QACF,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC3D,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IAC/B,CAAC;IAED,cAAc,CAAC,GAAsB;QACnC,IAAI,CAAC,WAAW,GAAG,GAAG,CAAC;IACzB,CAAC;IAED,oBAAoB,CAAC,SAAoC;QACvD,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;IACrC,CAAC;IAED,YAAY;QACV,MAAM,SAAS,GAAG,+BAA+B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7D,uBAAuB,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QAC9C,MAAM,cAAc,GAAG,IAAI,wBAAwB,CACjD,IAAI,CAAC,WAAW,EAChB,SAAS,EACT,IAAI,CACL,CAAC;QACF,cAAc,CAAC,IAAI,CAAC,qBAAqB,EAAE,cAAc,CAAC,CAAC;QAC3D,IAAI,CAAC,qBAAqB,GAAG,cAAc,CAAC;QAC5C,IAAI,CAAC,WAAW,EAAE,CAAC;IACrB,CAAC;IAED,eAAe,CAAC,cAAwC;QACtD,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,cAAc,CAAC,SAAS,CAAC,CAAC;QAC5D,WAAW,CAAC,IAAI,CAAC,qBAAqB,EAAE,cAAc,CAAC,CAAC;QACxD,IAAI,IAAI,CAAC,qBAAqB,KAAK,cAAc,EAAE;YACjD,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;SAC5D;IACH,CAAC;IAED,aAAa,CAAC,GAAkB;QAC9B,MAAM,KAAK,GAAG,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,GAAG,CAAC,CAAC;QAC1E,IAAI,KAAK,EAAE;YACT,IAAI,CAAC,qBAAqB,GAAG,KAAK,CAAC;SACpC;IACH,CAAC;IAED,CAAC,OAAO;QACN,IAAI;YACF,IAAI,CAAC,WAAW,EAAE,CAAC;YACnB,IAAI,CAAC,iBAAiB,CAAC,UAAU,EAAE,CAAC;YACpC,MAAM,KAAK,GAAG,IAAI,qBAAqB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvD,MAAM,KAAK,GACT,IAAI,CAAC,IAAI,CAAC,QAAQ,YAAY,SAAS;gBACrC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ;gBACpB,CAAC,CAAC,SAAS,CAAC;YAChB,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAC3D,MAAM,WAAW,GACf,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,YAAY,CAAC,QAAQ,CAC7D,CAAC,KAAK,CAAC,EACP,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,KAAK,CACzC,CAAiB,CAAC;YACrB,MAAM,MAAM,GAAG,oBAAoB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;YACpD,UAAU,CACR,MAAM,CAAC,QAAQ,KAAK,IAAI,CAAC,QAAQ,IAAI,MAAM,CAAC,UAAU,KAAK,IAAI,CAAC,IAAI,EACpE,wBAAwB,CACzB,CAAC;YACF,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC9B,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC;SACnC;QAAC,OAAO,KAAK,EAAE;YACd,iBAAiB,CAAC,KAAK,CAAC,CAAC;YACzB,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,WAAW,CAC/D,uBAAuB,KAAK,CAAC,OAAO,EAAE,CACvC,CAAC;YACF,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,CAAC;SAC/B;IACH,CAAC;IAED,WAAW;QACT,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAC1C,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,cAAc,EAAE,EAAE,CACpD,cAAc,CAAC,oBAAoB,CAAC,aAAa,CAAC,SAAS,CAAC,CAC7D,CAAC;IACJ,CAAC;IAED,gBAAgB,CAAC,UAAsB;QACrC,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAC3C,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,cAAc,EAAE,EAAE;YACpD,cAAc,CAAC,oBAAoB,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;YAC9D,cAAc,CAAC,cAAc,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC;IAED,IAAI,cAAc;QAChB,OAAO,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC;IAC3C,CAAC;IAED,IAAI,eAAe;QACjB,IACE,IAAI,CAAC,eAAe,CAAC,MAAM,YAAY,YAAY;YACnD,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,mBAAmB;gBAC7C,mBAAmB,CAAC,IAAI,EAC1B;YACA,OAAO,IAAI,CAAC,cAAc,CAAC;SAC5B;QACD,OAAO,IAAI,CAAC,qBAAqB,CAAC,MAAM,CACtC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,oBAAoB,CAAC,MAAM,KAAK,eAAe,CAAC,MAAM,CACxE,CAAC,MAAM,CAAC;IACX,CAAC;IAED,IAAI,eAAe;QACjB,IACE,IAAI,CAAC,eAAe,CAAC,MAAM,YAAY,YAAY;YACnD,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,mBAAmB;gBAC7C,mBAAmB,CAAC,IAAI,EAC1B;YACA,OAAO,CAAC,CAAC;SACV;QACD,OAAO,IAAI,CAAC,qBAAqB,CAAC,MAAM,CACtC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,oBAAoB,CAAC,MAAM,KAAK,eAAe,CAAC,MAAM,CACxE,CAAC,MAAM,CAAC;IACX,CAAC;CACF;AAED,MAAM,OAAO,4BAA4B;IAC9B,WAAW,CAAc;IAClC,QAAQ,CAAW;IACnB,KAAK,CAAY;IACjB,UAAU,CAAU;IACpB,UAAU,GAA8B,EAAE,CAAC;IAC3C,aAAa,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC;IACrC,eAAe,CAAsC;IAErD,YACE,QAAkB,EAClB,KAAgB,EAChB,UAAmB,EACnB,WAAwB;QAExB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;IACjC,CAAC;IAED,CAAC,QAAQ;QACP,IAAI;YACF,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;YAChC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;YAChD,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,iBAAiB,CAAC,UAAU,EAAE,CAAC,CAAC;YACjE,MAAM,KAAK,GAAG,IAAI,qBAAqB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvD,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CACtC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,YAAY,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CACvC,CAAC;YACF,MAAM,WAAW,GACf,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,YAAY,CAAC,QAAQ,CAC7D,CAAC,KAAK,CAAC,EACP,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,KAAK,CACzC,CAAiB,CAAC;YACrB,WAAW,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,UAAU,CAAC,CAAC;gBACxE,KAAK,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAClC,CAAC,CAAC,CAAC;YACH,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,CAAC;SAC/B;QAAC,OAAO,KAAK,EAAE;YACd,iBAAiB,CAAC,KAAK,CAAC,CAAC;YACzB,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YACzE,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;SAC3B;gBAAS;YACR,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC,CAAC;SAChE;IACH,CAAC;IAED,CAAC,eAAe;QACd,IAAI;YACF,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;YAChC,MAAM,KAAK,GAAG,IAAI,qBAAqB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvD,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC,UAAU;iBAChC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE;gBACjB,MAAM,MAAM,GAAG,SAAS,CAAC,eAAe,CAAC,MAAM,CAAC;gBAChD,IACE,CAAC,MAAM,YAAY,YAAY;oBAC7B,MAAM,CAAC,mBAAmB,KAAK,mBAAmB,CAAC,IAAI,CAAC;oBAC1D,MAAM,YAAY,SAAS,EAC3B;oBACA,SAAS,CAAC,iBAAiB,CAAC,UAAU,EAAE,CAAC;oBACzC,OAAO,IAAI,YAAY,CAAC,IAAI,CAAC,KAAK,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC;iBACrD;gBACD,OAAO,SAAS,CAAC;YACnB,CAAC,CAAC;iBACD,MAAM,CAAC,aAAa,CAAC,CAAC;YACzB,MAAM,WAAW,GACf,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,YAAY,CAAC,QAAQ,CAC7D,CAAC,KAAK,CAAC,EACP,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,KAAK,CACzC,CAAiB,CAAC;YACrB,WAAW,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,UAAU,CAAC,CAAC;gBACxE,KAAK,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAClC,CAAC,CAAC,CAAC;YACH,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,CAAC;SAC/B;QAAC,OAAO,KAAK,EAAE;YACd,iBAAiB,CAAC,KAAK,CAAC,CAAC;YACzB,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YACzE,IAAI,CAAC,aAAa,CAAC,IAAI,EAAE,CAAC;SAC3B;gBAAS;YACR,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC,CAAC;SAChE;IACH,CAAC;CACF"}
+{"version":3,"file":"TestableEditorState.js","sourceRoot":"","sources":["../../../../../../src/stores/editor/editor-state/element-editor-state/testable/TestableEditorState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,EAKL,YAAY,EACZ,YAAY,EACZ,qBAAqB,EACrB,SAAS,EACT,mBAAmB,EACnB,SAAS,GACV,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAEL,iBAAiB,EACjB,UAAU,EACV,oBAAoB,EACpB,WAAW,EACX,cAAc,EACd,WAAW,EACX,aAAa,GACd,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAAE,MAAM,EAAE,UAAU,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,MAAM,CAAC;AAEtE,OAAO,EAAE,uBAAuB,EAAE,MAAM,4DAA4D,CAAC;AACrG,OAAO,EAAE,+BAA+B,EAAE,MAAM,iCAAiC,CAAC;AAClF,OAAO,EAAE,eAAe,EAAE,MAAM,0DAA0D,CAAC;AAC3F,OAAO,EACL,wBAAwB,EACxB,kBAAkB,GACnB,MAAM,yBAAyB,CAAC;AAEjC,MAAM,OAAO,uBAAuB;IACzB,WAAW,CAAc;IACzB,SAAS,CAA0B;IAC5C,MAAM,CAAyB;IAE/B,YAAY,SAAkC,EAAE,WAAwB;QACtE,cAAc,CAAC,IAAI,EAAE;YACnB,MAAM,EAAE,UAAU;YAClB,SAAS,EAAE,MAAM;SAClB,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC7B,CAAC;IAED,SAAS,CAAC,GAA2B;QACnC,IAAI,CAAC,MAAM,GAAG,GAAG,CAAC;IACpB,CAAC;CACF;AAED,MAAM,CAAN,IAAY,iBAGX;AAHD,WAAY,iBAAiB;IAC3B,oCAAe,CAAA;IACf,4CAAuB,CAAA;AACzB,CAAC,EAHW,iBAAiB,KAAjB,iBAAiB,QAG5B;AAED,MAAM,OAAO,uBAAuB;IACzB,WAAW,CAAc;IAClC,QAAQ,CAAW;IACnB,IAAI,CAAa;IACjB,qBAAqB,CAAuC;IAC5D,qBAAqB,GAA+B,EAAE,CAAC;IACvD,WAAW,GAAG,iBAAiB,CAAC,SAAS,CAAC;IAC1C,iBAAiB,CAA4B;IAC7C,iBAAiB,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC;IACzC,eAAe,CAA0B;IACzC,UAAU,CAAU;IAEpB,YACE,QAAkB,EAClB,IAAgB,EAChB,UAAmB,EACnB,WAAwB;QAExB,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;QAC/B,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,eAAe,GAAG,IAAI,uBAAuB,CAAC,IAAI,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QAC3E,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAC9C,CAAC,SAAS,EAAE,EAAE,CACZ,IAAI,wBAAwB,CAAC,IAAI,CAAC,WAAW,EAAE,SAAS,EAAE,IAAI,CAAC,CAClE,CAAC;QACF,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC3D,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;IAC/B,CAAC;IAED,cAAc,CAAC,GAAsB;QACnC,IAAI,CAAC,WAAW,GAAG,GAAG,CAAC;IACzB,CAAC;IAED,oBAAoB,CAAC,SAAoC;QACvD,IAAI,CAAC,iBAAiB,GAAG,SAAS,CAAC;IACrC,CAAC;IAED,YAAY;QACV,MAAM,SAAS,GAAG,+BAA+B,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7D,uBAAuB,CAAC,IAAI,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QAC9C,MAAM,cAAc,GAAG,IAAI,wBAAwB,CACjD,IAAI,CAAC,WAAW,EAChB,SAAS,EACT,IAAI,CACL,CAAC;QACF,cAAc,CAAC,IAAI,CAAC,qBAAqB,EAAE,cAAc,CAAC,CAAC;QAC3D,IAAI,CAAC,qBAAqB,GAAG,cAAc,CAAC;QAC5C,IAAI,CAAC,WAAW,EAAE,CAAC;IACrB,CAAC;IAED,eAAe,CAAC,cAAwC;QACtD,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,cAAc,CAAC,SAAS,CAAC,CAAC;QAC5D,WAAW,CAAC,IAAI,CAAC,qBAAqB,EAAE,cAAc,CAAC,CAAC;QACxD,IAAI,IAAI,CAAC,qBAAqB,KAAK,cAAc,EAAE;YACjD,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;SAC5D;IACH,CAAC;IAED,aAAa,CAAC,GAAkB;QAC9B,MAAM,KAAK,GAAG,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,GAAG,CAAC,CAAC;QAC1E,IAAI,KAAK,EAAE;YACT,IAAI,CAAC,qBAAqB,GAAG,KAAK,CAAC;SACpC;IACH,CAAC;IAED,CAAC,OAAO;QACN,IAAI;YACF,IAAI,CAAC,WAAW,EAAE,CAAC;YACnB,IAAI,CAAC,iBAAiB,CAAC,UAAU,EAAE,CAAC;YACpC,MAAM,MAAM,GAAG,CAAC,MAAM,UAAU,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC,CAAe,CAAC;YACxE,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAC9B,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC;SACnC;QAAC,OAAO,KAAK,EAAE;YACd,iBAAiB,CAAC,KAAK,CAAC,CAAC;YACzB,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,WAAW,CAC/D,uBAAuB,KAAK,CAAC,OAAO,EAAE,CACvC,CAAC;YACF,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,CAAC;SAC/B;IACH,CAAC;IAED,8DAA8D;IAC9D,KAAK,CAAC,eAAe;QACnB,MAAM,KAAK,GAAG,IAAI,qBAAqB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvD,MAAM,KAAK,GACT,IAAI,CAAC,IAAI,CAAC,QAAQ,YAAY,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,CAAC;QAC3E,KAAK,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,YAAY,CAAC,KAAK,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAC3D,MAAM,WAAW,GACf,MAAM,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,YAAY,CAAC,QAAQ,CAC5D,CAAC,KAAK,CAAC,EACP,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,KAAK,CACzC,CAAC;QACJ,MAAM,MAAM,GAAG,oBAAoB,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC;QACpD,UAAU,CACR,MAAM,CAAC,QAAQ,KAAK,IAAI,CAAC,QAAQ;YAC/B,MAAM,CAAC,UAAU,CAAC,EAAE,KAAK,IAAI,CAAC,IAAI,CAAC,EAAE,EACvC,wBAAwB,CACzB,CAAC;QACF,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,WAAW;QACT,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC;QAC1C,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,cAAc,EAAE,EAAE,CACpD,cAAc,CAAC,oBAAoB,CAAC,aAAa,CAAC,SAAS,CAAC,CAC7D,CAAC;IACJ,CAAC;IAED,gBAAgB,CAAC,UAAsB;QACrC,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAC3C,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,cAAc,EAAE,EAAE;YACpD,cAAc,CAAC,oBAAoB,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;YAC9D,cAAc,CAAC,cAAc,CAAC,kBAAkB,CAAC,MAAM,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC;IAED,IAAI,cAAc;QAChB,OAAO,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC;IAC3C,CAAC;IAED,IAAI,eAAe;QACjB,IACE,IAAI,CAAC,eAAe,CAAC,MAAM,YAAY,YAAY;YACnD,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,mBAAmB;gBAC7C,mBAAmB,CAAC,IAAI,EAC1B;YACA,OAAO,IAAI,CAAC,cAAc,CAAC;SAC5B;QACD,OAAO,IAAI,CAAC,qBAAqB,CAAC,MAAM,CACtC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,oBAAoB,CAAC,MAAM,KAAK,eAAe,CAAC,MAAM,CACxE,CAAC,MAAM,CAAC;IACX,CAAC;IAED,IAAI,eAAe;QACjB,IACE,IAAI,CAAC,eAAe,CAAC,MAAM,YAAY,YAAY;YACnD,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,mBAAmB;gBAC7C,mBAAmB,CAAC,IAAI,EAC1B;YACA,OAAO,CAAC,CAAC;SACV;QACD,OAAO,IAAI,CAAC,qBAAqB,CAAC,MAAM,CACtC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,oBAAoB,CAAC,MAAM,KAAK,eAAe,CAAC,MAAM,CACxE,CAAC,MAAM,CAAC;IACX,CAAC;CACF;AAED,MAAM,OAAO,4BAA4B;IAC9B,WAAW,CAAc;IAClC,QAAQ,CAAW;IACnB,KAAK,CAAY;IACjB,UAAU,CAAU;IACpB,UAAU,GAA8B,EAAE,CAAC;IAC3C,iBAAiB,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC;IACzC,eAAe,CAAsC;IAErD,YACE,QAAkB,EAClB,KAAgB,EAChB,UAAmB,EACnB,WAAwB;QAExB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,KAAK,GAAG,KAAK,CAAC;QACnB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;IACjC,CAAC;IAED,CAAC,QAAQ;QACP,IAAI;YACF,IAAI,CAAC,iBAAiB,CAAC,UAAU,EAAE,CAAC;YACpC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;YAChD,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,iBAAiB,CAAC,UAAU,EAAE,CAAC,CAAC;YACjE,MAAM,KAAK,GAAG,IAAI,qBAAqB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvD,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CACtC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,YAAY,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC,CACvC,CAAC;YACF,MAAM,WAAW,GACf,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,YAAY,CAAC,QAAQ,CAC7D,CAAC,KAAK,CAAC,EACP,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,KAAK,CACzC,CAAiB,CAAC;YACrB,WAAW,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,UAAU,CAAC,CAAC;gBACxE,KAAK,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAClC,CAAC,CAAC,CAAC;YACH,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC;SACnC;QAAC,OAAO,KAAK,EAAE;YACd,iBAAiB,CAAC,KAAK,CAAC,CAAC;YACzB,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YACzE,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,CAAC;SAC/B;gBAAS;YACR,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC,CAAC;SAChE;IACH,CAAC;IAED,CAAC,eAAe;QACd,IAAI;YACF,IAAI,CAAC,iBAAiB,CAAC,UAAU,EAAE,CAAC;YACpC,MAAM,KAAK,GAAG,IAAI,qBAAqB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACvD,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC,UAAU;iBAChC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE;gBACjB,MAAM,MAAM,GAAG,SAAS,CAAC,eAAe,CAAC,MAAM,CAAC;gBAChD,IACE,CAAC,MAAM,YAAY,YAAY;oBAC7B,MAAM,CAAC,mBAAmB,KAAK,mBAAmB,CAAC,IAAI,CAAC;oBAC1D,MAAM,YAAY,SAAS,EAC3B;oBACA,SAAS,CAAC,iBAAiB,CAAC,UAAU,EAAE,CAAC;oBACzC,OAAO,IAAI,YAAY,CAAC,IAAI,CAAC,KAAK,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC;iBACrD;gBACD,OAAO,SAAS,CAAC;YACnB,CAAC,CAAC;iBACD,MAAM,CAAC,aAAa,CAAC,CAAC;YACzB,MAAM,WAAW,GACf,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,YAAY,CAAC,QAAQ,CAC7D,CAAC,KAAK,CAAC,EACP,IAAI,CAAC,WAAW,CAAC,iBAAiB,CAAC,KAAK,CACzC,CAAiB,CAAC;YACrB,WAAW,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE;gBAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,UAAU,CAAC,CAAC;gBACxE,KAAK,EAAE,gBAAgB,CAAC,MAAM,CAAC,CAAC;YAClC,CAAC,CAAC,CAAC;YACH,IAAI,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC;SACnC;QAAC,OAAO,KAAK,EAAE;YACd,iBAAiB,CAAC,KAAK,CAAC,CAAC;YACzB,IAAI,CAAC,WAAW,CAAC,gBAAgB,CAAC,mBAAmB,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YACzE,IAAI,CAAC,iBAAiB,CAAC,IAAI,EAAE,CAAC;SAC/B;gBAAS;YACR,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,iBAAiB,CAAC,QAAQ,EAAE,CAAC,CAAC;SAChE;IACH,CAAC;CACF"}

package/lib/stores/editor/panel-group/SQLPlaygroundPanelState.d.ts

@@ -19,13 +19,16 @@ import { editor as monacoEditorAPI } from 'monaco-editor';
 import { type Schema, type Table, type Database, type PackageableConnection, Column } from '@finos/legend-graph';
 import type { EditorStore } from '../EditorStore.js';
 import type { CommandRegistrar } from '@finos/legend-application';
+import type { Entity } from '@finos/legend-storage';
 export declare abstract class DatabaseSchemaExplorerTreeNodeData implements TreeNodeData {
     isOpen?: boolean | undefined;
     id: string;
     label: string;
     parentId?: string | undefined;
     childrenIds?: string[] | undefined;
+    isChecked: boolean;
     constructor(id: string, label: string, parentId: string | undefined);
+    setChecked(val: boolean): void;
 }
 export declare class DatabaseSchemaExplorerTreeSchemaNodeData extends DatabaseSchemaExplorerTreeNodeData {
     schema: Schema;
@@ -51,12 +54,15 @@ export declare class SQLPlaygroundPanelState implements CommandRegistrar {
     isFetchingSchema: boolean;
     isExecutingRawSQL: boolean;
     connection?: PackageableConnection | undefined;
+    database?: Database | undefined;
     treeData?: DatabaseSchemaExplorerTreeData | undefined;
     readonly sqlEditorTextModel: monacoEditorAPI.ITextModel;
     sqlEditor?: monacoEditorAPI.IStandaloneCodeEditor | undefined;
     sqlEditorViewState?: monacoEditorAPI.ICodeEditorViewState | undefined;
     sqlText: string;
     sqlExecutionResult?: string | undefined;
+    isBuildingDatabase: boolean;
+    isUpdatingDatabase: boolean;
     constructor(editorStore: EditorStore);
     setTreeData(val?: DatabaseSchemaExplorerTreeData): void;
     setConnection(val: PackageableConnection | undefined): void;
@@ -68,6 +74,7 @@ export declare class SQLPlaygroundPanelState implements CommandRegistrar {
     deregisterCommands(): void;
     onNodeSelect(node: DatabaseSchemaExplorerTreeNodeData, treeData: DatabaseSchemaExplorerTreeData): GeneratorFn<void>;
     getChildNodes(node: DatabaseSchemaExplorerTreeNodeData, treeData: DatabaseSchemaExplorerTreeData): DatabaseSchemaExplorerTreeNodeData[] | undefined;
+    toggleCheckedNode(node: DatabaseSchemaExplorerTreeNodeData, treeData: DatabaseSchemaExplorerTreeData): void;
     fetchDatabaseMetadata(): GeneratorFn<void>;
     fetchSchemaMetadata(schemaNode: DatabaseSchemaExplorerTreeSchemaNodeData, treeData: DatabaseSchemaExplorerTreeData): GeneratorFn<void>;
     fetchTableMetadata(tableNode: DatabaseSchemaExplorerTreeTableNodeData, treeData: DatabaseSchemaExplorerTreeData): GeneratorFn<void>;
@@ -78,5 +85,7 @@ export declare class SQLPlaygroundPanelState implements CommandRegistrar {
      */
     private buildIntermediateDatabase;
     executeRawSQL(): GeneratorFn<void>;
+    generateDatabase(): GeneratorFn<Entity>;
+    updateDatabase(): GeneratorFn<void>;
 }
 //# sourceMappingURL=SQLPlaygroundPanelState.d.ts.map

package/lib/stores/editor/panel-group/SQLPlaygroundPanelState.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"SQLPlaygroundPanelState.d.ts","sourceRoot":"","sources":["../../../../src/stores/editor/panel-group/SQLPlaygroundPanelState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAChE,OAAO,EACL,KAAK,WAAW,EASjB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EAAE,MAAM,IAAI,eAAe,EAAE,MAAM,eAAe,CAAC;AAC1D,OAAO,EACL,KAAK,MAAM,EACX,KAAK,KAAK,EACV,KAAK,QAAQ,EACb,KAAK,qBAAqB,EAI1B,MAAM,EAIP,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAMrD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAIlE,8BAAsB,kCACpB,YAAW,YAAY;IAEvB,MAAM,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,WAAW,CAAC,EAAE,MAAM,EAAE,GAAG,SAAS,CAAC;gBAEvB,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,SAAS;CAKpE;AAED,qBAAa,wCAAyC,SAAQ,kCAAkC;IAC9F,MAAM,EAAE,MAAM,CAAC;gBAEH,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM;CAIvC;AAED,qBAAa,uCAAwC,SAAQ,kCAAkC;IACpF,QAAQ,EAAE,MAAM,CAAC;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,KAAK,CAAC;gBAED,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK;CAMtE;AAED,qBAAa,wCAAyC,SAAQ,kCAAkC;IACrF,QAAQ,EAAE,MAAM,CAAC;IAC1B,KAAK,EAAE,KAAK,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;gBAEH,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM;CAMvE;AAED,MAAM,WAAW,8BACf,SAAQ,QAAQ,CAAC,kCAAkC,CAAC;IACpD,QAAQ,EAAE,QAAQ,CAAC;CACpB;AAMD,qBAAa,uBAAwB,YAAW,gBAAgB;IAC9D,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAElC,gBAAgB,UAAS;IACzB,iBAAiB,UAAS;IAE1B,UAAU,CAAC,EAAE,qBAAqB,GAAG,SAAS,CAAC;IAC/C,QAAQ,CAAC,EAAE,8BAA8B,GAAG,SAAS,CAAC;IACtD,QAAQ,CAAC,kBAAkB,EAAE,eAAe,CAAC,UAAU,CAAC;IACxD,SAAS,CAAC,EAAE,eAAe,CAAC,qBAAqB,GAAG,SAAS,CAAC;IAC9D,kBAAkB,CAAC,EAAE,eAAe,CAAC,oBAAoB,GAAG,SAAS,CAAC;IACtE,OAAO,SAAoB;IAC3B,kBAAkB,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;gBAE5B,WAAW,EAAE,WAAW;IA8BpC,WAAW,CAAC,GAAG,CAAC,EAAE,8BAA8B,GAAG,IAAI;IAIvD,aAAa,CAAC,GAAG,EAAE,qBAAqB,GAAG,SAAS,GAAG,IAAI;IAK3D,UAAU,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAI7B,YAAY,CAAC,GAAG,EAAE,eAAe,CAAC,qBAAqB,GAAG,SAAS,GAAG,IAAI;IAW1E,QAAQ,IAAI,IAAI;IAMhB,qBAAqB,CACnB,GAAG,EAAE,eAAe,CAAC,oBAAoB,GAAG,SAAS,GACpD,IAAI;IAIP,gBAAgB,IAAI,IAAI;IAgBxB,kBAAkB,IAAI,IAAI;IAOzB,YAAY,CACX,IAAI,EAAE,kCAAkC,EACxC,QAAQ,EAAE,8BAA8B,GACvC,WAAW,CAAC,IAAI,CAAC;IAgBpB,aAAa,CACX,IAAI,EAAE,kCAAkC,EACxC,QAAQ,EAAE,8BAA8B,GACvC,kCAAkC,EAAE,GAAG,SAAS;IAMlD,qBAAqB,IAAI,WAAW,CAAC,IAAI,CAAC;IAoD1C,mBAAmB,CAClB,UAAU,EAAE,wCAAwC,EACpD,QAAQ,EAAE,8BAA8B,GACvC,WAAW,CAAC,IAAI,CAAC;IAyDnB,kBAAkB,CACjB,SAAS,EAAE,uCAAuC,EAClD,QAAQ,EAAE,8BAA8B,GACvC,WAAW,CAAC,IAAI,CAAC;IAkEpB;;;;OAIG;YACW,yBAAyB;IAoBtC,aAAa,IAAI,WAAW,CAAC,IAAI,CAAC;CAkCpC"}
+{"version":3,"file":"SQLPlaygroundPanelState.d.ts","sourceRoot":"","sources":["../../../../src/stores/editor/panel-group/SQLPlaygroundPanelState.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAEH,OAAO,KAAK,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAChE,OAAO,EACL,KAAK,WAAW,EAYjB,MAAM,sBAAsB,CAAC;AAE9B,OAAO,EAAE,MAAM,IAAI,eAAe,EAAE,MAAM,eAAe,CAAC;AAC1D,OAAO,EACL,KAAK,MAAM,EACX,KAAK,KAAK,EACV,KAAK,QAAQ,EACb,KAAK,qBAAqB,EAI1B,MAAM,EAMP,MAAM,qBAAqB,CAAC;AAC7B,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAMrD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAGlE,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,uBAAuB,CAAC;AAGpD,8BAAsB,kCACpB,YAAW,YAAY;IAEvB,MAAM,CAAC,EAAE,OAAO,GAAG,SAAS,CAAC;IAC7B,EAAE,EAAE,MAAM,CAAC;IACX,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC9B,WAAW,CAAC,EAAE,MAAM,EAAE,GAAG,SAAS,CAAC;IACnC,SAAS,UAAS;gBAEN,EAAE,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,SAAS;IAWnE,UAAU,CAAC,GAAG,EAAE,OAAO,GAAG,IAAI;CAG/B;AAED,qBAAa,wCAAyC,SAAQ,kCAAkC;IAC9F,MAAM,EAAE,MAAM,CAAC;gBAEH,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM;CAIvC;AAED,qBAAa,uCAAwC,SAAQ,kCAAkC;IACpF,QAAQ,EAAE,MAAM,CAAC;IAC1B,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,KAAK,CAAC;gBAED,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK;CAMtE;AAED,qBAAa,wCAAyC,SAAQ,kCAAkC;IACrF,QAAQ,EAAE,MAAM,CAAC;IAC1B,KAAK,EAAE,KAAK,CAAC;IACb,MAAM,EAAE,MAAM,CAAC;gBAEH,EAAE,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM;CAMvE;AAED,MAAM,WAAW,8BACf,SAAQ,QAAQ,CAAC,kCAAkC,CAAC;IACpD,QAAQ,EAAE,QAAQ,CAAC;CACpB;AAMD,qBAAa,uBAAwB,YAAW,gBAAgB;IAC9D,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAElC,gBAAgB,UAAS;IACzB,iBAAiB,UAAS;IAE1B,UAAU,CAAC,EAAE,qBAAqB,GAAG,SAAS,CAAC;IAC/C,QAAQ,CAAC,EAAE,QAAQ,GAAG,SAAS,CAAC;IAChC,QAAQ,CAAC,EAAE,8BAA8B,GAAG,SAAS,CAAC;IACtD,QAAQ,CAAC,kBAAkB,EAAE,eAAe,CAAC,UAAU,CAAC;IACxD,SAAS,CAAC,EAAE,eAAe,CAAC,qBAAqB,GAAG,SAAS,CAAC;IAC9D,kBAAkB,CAAC,EAAE,eAAe,CAAC,oBAAoB,GAAG,SAAS,CAAC;IACtE,OAAO,SAAoB;IAC3B,kBAAkB,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAExC,kBAAkB,UAAS;IAC3B,kBAAkB,UAAS;gBAEf,WAAW,EAAE,WAAW;IAmCpC,WAAW,CAAC,GAAG,CAAC,EAAE,8BAA8B,GAAG,IAAI;IAIvD,aAAa,CAAC,GAAG,EAAE,qBAAqB,GAAG,SAAS,GAAG,IAAI;IAQ3D,UAAU,CAAC,GAAG,EAAE,MAAM,GAAG,IAAI;IAI7B,YAAY,CAAC,GAAG,EAAE,eAAe,CAAC,qBAAqB,GAAG,SAAS,GAAG,IAAI;IAW1E,QAAQ,IAAI,IAAI;IAMhB,qBAAqB,CACnB,GAAG,EAAE,eAAe,CAAC,oBAAoB,GAAG,SAAS,GACpD,IAAI;IAIP,gBAAgB,IAAI,IAAI;IAgBxB,kBAAkB,IAAI,IAAI;IAOzB,YAAY,CACX,IAAI,EAAE,kCAAkC,EACxC,QAAQ,EAAE,8BAA8B,GACvC,WAAW,CAAC,IAAI,CAAC;IAgBpB,aAAa,CACX,IAAI,EAAE,kCAAkC,EACxC,QAAQ,EAAE,8BAA8B,GACvC,kCAAkC,EAAE,GAAG,SAAS;IAMnD,iBAAiB,CACf,IAAI,EAAE,kCAAkC,EACxC,QAAQ,EAAE,8BAA8B,GACvC,IAAI;IAwBN,qBAAqB,IAAI,WAAW,CAAC,IAAI,CAAC;IA4D1C,mBAAmB,CAClB,UAAU,EAAE,wCAAwC,EACpD,QAAQ,EAAE,8BAA8B,GACvC,WAAW,CAAC,IAAI,CAAC;IAyEnB,kBAAkB,CACjB,SAAS,EAAE,uCAAuC,EAClD,QAAQ,EAAE,8BAA8B,GACvC,WAAW,CAAC,IAAI,CAAC;IAkEpB;;;;OAIG;YACW,yBAAyB;IAoBtC,aAAa,IAAI,WAAW,CAAC,IAAI,CAAC;IAmClC,gBAAgB,IAAI,WAAW,CAAC,MAAM,CAAC;IAmEvC,cAAc,IAAI,WAAW,CAAC,IAAI,CAAC;CA2ErC"}

package/lib/stores/editor/panel-group/SQLPlaygroundPanelState.js

@@ -13,25 +13,34 @@
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
-import { assertErrorThrown, LogEvent, addUniqueEntry, isNonNullable, filterByType, ActionState, getNonNullableEntry, getNullableLastEntry, } from '@finos/legend-shared';
+import { assertErrorThrown, LogEvent, addUniqueEntry, isNonNullable, filterByType, ActionState, getNonNullableEntry, getNullableLastEntry, guaranteeNonNullable, IllegalStateError, } from '@finos/legend-shared';
 import { observable, makeObservable, flow, flowResult, action } from 'mobx';
 import { editor as monacoEditorAPI } from 'monaco-editor';
-import { DatabaseBuilderInput, DatabasePattern, TargetDatabase, Column, getSchema, guaranteeRelationalDatabaseConnection, GRAPH_MANAGER_EVENT, } from '@finos/legend-graph';
+import { DatabaseBuilderInput, DatabasePattern, TargetDatabase, Column, getSchema, guaranteeRelationalDatabaseConnection, GRAPH_MANAGER_EVENT, getNullableSchema, getNullableTable, } from '@finos/legend-graph';
 import { LEGEND_STUDIO_APP_EVENT } from '../../../__lib__/LegendStudioEvent.js';
 import { CODE_EDITOR_LANGUAGE, moveCursorToPosition, } from '@finos/legend-lego/code-editor';
 import { STO_RELATIONAL_LEGEND_STUDIO_COMMAND_KEY } from '../../../__lib__/STO_Relational_LegendStudioCommand.js';
 import { PANEL_MODE } from '../EditorConfig.js';
+import { GraphEditFormModeState } from '../GraphEditFormModeState.js';
 export class DatabaseSchemaExplorerTreeNodeData {
     isOpen;
     id;
     label;
     parentId;
     childrenIds;
+    isChecked = false;
     constructor(id, label, parentId) {
+        makeObservable(this, {
+            isChecked: observable,
+            setChecked: action,
+        });
         this.id = id;
         this.label = label;
         this.parentId = parentId;
     }
+    setChecked(val) {
+        this.isChecked = val;
+    }
 }
 export class DatabaseSchemaExplorerTreeSchemaNodeData extends DatabaseSchemaExplorerTreeNodeData {
     schema;
@@ -70,23 +79,29 @@ export class SQLPlaygroundPanelState {
     isFetchingSchema = false;
     isExecutingRawSQL = false;
     connection;
+    database;
     treeData;
     sqlEditorTextModel;
     sqlEditor;
     sqlEditorViewState;
     sqlText = DEFAULT_SQL_TEXT;
     sqlExecutionResult;
+    isBuildingDatabase = false;
+    isUpdatingDatabase = false;
     constructor(editorStore) {
         makeObservable(this, {
             isFetchingSchema: observable,
             isExecutingRawSQL: observable,
             connection: observable,
+            database: observable,
             treeData: observable,
             sqlText: observable,
             resetSQL: action,
             sqlExecutionResult: observable,
             sqlEditor: observable.ref,
             sqlEditorViewState: observable.ref,
+            isBuildingDatabase: observable,
+            isUpdatingDatabase: observable,
             setTreeData: action,
             setConnection: action,
             setSQLEditor: action,
@@ -97,6 +112,8 @@ export class SQLPlaygroundPanelState {
             fetchSchemaMetadata: flow,
             fetchTableMetadata: flow,
             executeRawSQL: flow,
+            generateDatabase: flow,
+            updateDatabase: flow,
         });
         this.editorStore = editorStore;
         this.sqlEditorTextModel = monacoEditorAPI.createModel(this.sqlText, CODE_EDITOR_LANGUAGE.SQL);
@@ -106,6 +123,9 @@ export class SQLPlaygroundPanelState {
     }
     setConnection(val) {
         this.connection = val;
+        if (val) {
+            this.database = guaranteeRelationalDatabaseConnection(val).store.value;
+        }
         this.sqlEditorTextModel.setValue(DEFAULT_SQL_TEXT);
     }
     setSQLText(val) {
@@ -158,9 +178,28 @@ export class SQLPlaygroundPanelState {
     }
     getChildNodes(node, treeData) {
         return node.childrenIds
-            ?.map((n) => treeData.nodes.get(n))
+            ?.map((childNode) => treeData.nodes.get(childNode))
             .filter(isNonNullable);
     }
+    toggleCheckedNode(node, treeData) {
+        node.setChecked(!node.isChecked);
+        if (node instanceof DatabaseSchemaExplorerTreeSchemaNodeData) {
+            this.getChildNodes(node, treeData)?.forEach((childNode) => {
+                childNode.setChecked(node.isChecked);
+            });
+        }
+        else if (node instanceof DatabaseSchemaExplorerTreeTableNodeData) {
+            if (node.parentId) {
+                const parent = treeData.nodes.get(node.parentId);
+                if (parent &&
+                    this.getChildNodes(parent, treeData)?.every((e) => e.isChecked === node.isChecked)) {
+                    parent.setChecked(node.isChecked);
+                }
+            }
+        }
+        // TODO: support toggling check for columns
+        this.setTreeData({ ...treeData });
+    }
     *fetchDatabaseMetadata() {
         if (!this.connection) {
             return;
@@ -185,6 +224,7 @@ export class SQLPlaygroundPanelState {
                 rootIds.push(schemaId);
                 const schemaNode = new DatabaseSchemaExplorerTreeSchemaNodeData(schemaId, dbSchema);
                 nodes.set(schemaId, schemaNode);
+                schemaNode.setChecked(Boolean(this.database?.schemas.find((schema) => schema.name === dbSchema.name)));
             });
             const treeData = { rootIds, nodes, database };
             this.setTreeData(treeData);
@@ -225,6 +265,15 @@ export class SQLPlaygroundPanelState {
                 const tableNode = new DatabaseSchemaExplorerTreeTableNodeData(tableId, schemaNode.id, schema, table);
                 treeData.nodes.set(tableId, tableNode);
                 addUniqueEntry(childrenIds, tableId);
+                if (this.database) {
+                    const matchingSchema = getNullableSchema(this.database, schema.name);
+                    tableNode.setChecked(Boolean(matchingSchema
+                        ? getNullableTable(matchingSchema, table.name)
+                        : undefined));
+                }
+                else {
+                    tableNode.setChecked(false);
+                }
             });
             schemaNode.childrenIds = childrenIds;
             this.setTreeData({ ...treeData });
@@ -324,5 +373,102 @@ export class SQLPlaygroundPanelState {
             this.isExecutingRawSQL = false;
         }
     }
+    *generateDatabase() {
+        if (!this.database || !this.connection || !this.treeData) {
+            throw new IllegalStateError(`Can't build database: builder is not properly set up`);
+        }
+        try {
+            this.isBuildingDatabase = true;
+            const treeData = this.treeData;
+            const databaseBuilderInput = new DatabaseBuilderInput(guaranteeRelationalDatabaseConnection(this.connection));
+            const packagePath = guaranteeNonNullable(this.database.package).path;
+            const databaseName = this.database.name;
+            databaseBuilderInput.targetDatabase = new TargetDatabase(packagePath, databaseName);
+            const config = databaseBuilderInput.config;
+            config.maxTables = undefined;
+            config.enrichTables = true;
+            config.enrichColumns = true;
+            config.enrichPrimaryKeys = true;
+            treeData.rootIds
+                .map((e) => treeData.nodes.get(e))
+                .filter(isNonNullable)
+                .forEach((schemaNode) => {
+                if (schemaNode instanceof DatabaseSchemaExplorerTreeSchemaNodeData) {
+                    const tableNodes = this.getChildNodes(schemaNode, treeData);
+                    const allChecked = tableNodes?.every((t) => t.isChecked === true);
+                    if (allChecked ||
+                        (schemaNode.isChecked && !schemaNode.childrenIds)) {
+                        config.patterns.push(new DatabasePattern(schemaNode.schema.name, undefined));
+                    }
+                    else {
+                        tableNodes?.forEach((t) => {
+                            if (t instanceof DatabaseSchemaExplorerTreeTableNodeData &&
+                                t.isChecked) {
+                                config.patterns.push(new DatabasePattern(schemaNode.schema.name, t.table.name));
+                            }
+                        });
+                    }
+                }
+            });
+            const entities = (yield this.editorStore.graphManagerState.graphManager.buildDatabase(databaseBuilderInput));
+            return getNonNullableEntry(entities, 0, 'Expected a database to be generated');
+        }
+        finally {
+            this.isBuildingDatabase = false;
+        }
+    }
+    *updateDatabase() {
+        if (!this.treeData || !this.database || !this.connection) {
+            return;
+        }
+        try {
+            this.isUpdatingDatabase = true;
+            const graph = this.editorStore.graphManagerState.createNewGraph();
+            (yield this.editorStore.graphManagerState.graphManager.buildGraph(graph, [(yield flowResult(this.generateDatabase()))], ActionState.create()));
+            const generatedDatabase = getNonNullableEntry(graph.ownDatabases, 0, 'Expected one database to be generated from input');
+            const currentDatabase = this.database;
+            // remove undefined schemas
+            const schemas = Array.from(this.treeData.nodes.values())
+                .map((schemaNode) => {
+                if (schemaNode instanceof DatabaseSchemaExplorerTreeSchemaNodeData) {
+                    return schemaNode.schema;
+                }
+                return undefined;
+            })
+                .filter(isNonNullable);
+            currentDatabase.schemas = currentDatabase.schemas.filter((schema) => {
+                if (schemas.find((item) => item.name === schema.name) &&
+                    !generatedDatabase.schemas.find((s) => s.name === schema.name)) {
+                    return false;
+                }
+                return true;
+            });
+            // update existing schemas
+            generatedDatabase.schemas.forEach((schema) => {
+                schema._OWNER = currentDatabase;
+                const currentSchemaIndex = currentDatabase.schemas.findIndex((item) => item.name === schema.name);
+                if (currentSchemaIndex !== -1) {
+                    currentDatabase.schemas[currentSchemaIndex] = schema;
+                }
+                else {
+                    currentDatabase.schemas.push(schema);
+                }
+            });
+            this.editorStore.applicationStore.notificationService.notifySuccess(`Database successfully updated`);
+            yield flowResult(this.editorStore
+                .getGraphEditorMode(GraphEditFormModeState)
+                .globalCompile({
+                message: `Can't compile graph after editing database. Redirecting you to text mode`,
+            }));
+        }
+        catch (error) {
+            assertErrorThrown(error);
+            this.editorStore.applicationStore.logService.error(LogEvent.create(LEGEND_STUDIO_APP_EVENT.DATABASE_BUILDER_FAILURE), error);
+            this.editorStore.applicationStore.notificationService.notifyError(error);
+        }
+        finally {
+            this.isUpdatingDatabase = false;
+        }
+    }
 }
 //# sourceMappingURL=SQLPlaygroundPanelState.js.map