@reltio/components 1.4.1906 → 1.4.1907

package/cjs/ConnectionEditor/ConnectionEditor.test.js

@@ -46,300 +46,343 @@ var __generator = (this && this.__generator) || function (thisArg, body) {
         if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
     }
 };
-var __spreadArray = (this && this.__spreadArray) || function (to, from, pack) {
-    if (pack || arguments.length === 2) for (var i = 0, l = from.length, ar; i < l; i++) {
-        if (ar || !(i in from)) {
-            if (!ar) ar = Array.prototype.slice.call(from, 0, i);
-            ar[i] = from[i];
-        }
-    }
-    return to.concat(ar || Array.prototype.slice.call(from));
-};
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
-var _a, _b;
 Object.defineProperty(exports, "__esModule", { value: true });
 var react_1 = __importDefault(require("react"));
-var enzyme_1 = require("enzyme");
-var test_utils_1 = require("react-dom/test-utils");
 var mdm_sdk_1 = require("@reltio/mdm-sdk");
+var user_event_1 = __importDefault(require("@testing-library/user-event"));
+var react_2 = require("@testing-library/react");
+var LocalizationProvider_1 = require("@mui/x-date-pickers/LocalizationProvider");
+var AdapterMoment_1 = require("@mui/x-date-pickers/AdapterMoment");
+var test_utils_1 = require("../test-utils");
 var MdmModuleContext_1 = require("../contexts/MdmModuleContext");
-var EditModeAttributesList_1 = require("../EditModeAttributesList");
-var EntitySelector_1 = require("../EntitySelector");
-var ConnectionRelationTypeSelector_1 = require("../ConnectionRelationTypeSelector");
-var ErrorWrapper_1 = require("../ErrorWrapper");
 var ConnectionEditor_1 = require("./ConnectionEditor");
-jest.mock('../EntitySelector', function () { return ({
-    EntitySelector: function () { return null; }
-}); });
-jest.mock('../EditModeAttributesList', function () { return ({
-    EditModeAttributesList: function () { return null; }
-}); });
-var newConnection = {
-    relation: {
-        uri: 'relations/uri$$123',
-        attributes: {},
-        type: undefined
-    },
-    entity: null
-};
-var relatedEntity = {
-    uri: 'entities/entity123',
-    type: 'configuration/entityTypes/HCP'
-};
-var metadata = {
-    uri: 'configuration',
-    entityTypes: [
-        {
-            uri: 'configuration/entityTypes/HCP'
-        },
-        {
-            uri: 'configuration/entityTypes/HCO'
-        },
-        {
-            uri: 'configuration/entityTypes/Committee'
+jest.mock('@reltio/mdm-sdk', function () { return (__assign(__assign({}, jest.requireActual('@reltio/mdm-sdk')), { getLookups: jest.fn(function () { return Promise.resolve(); }), getDependentLookups: jest.fn(function () { return Promise.resolve(); }), resolveLookupsList: jest.fn(function () { return Promise.resolve(); }), typeAheadSearch: jest.fn().mockReturnValue(Promise.resolve([])) })); });
+describe('ConnectionEditor test', function () {
+    var newConnection = {
+        relation: {
+            uri: 'relations/uri$$123',
+            attributes: {},
+            type: undefined
         },
-        {
-            uri: 'configuration/entityTypes/Location'
-        }
-    ],
-    relationTypes: [
-        {
-            uri: 'configuration/relationTypes/HeadNursing',
-            label: 'Head of Nursing',
-            startObject: {
-                objectTypeURI: 'configuration/entityTypes/HCP',
-                uri: 'configuration/relationTypes/HeadNursing/startObject'
+        entity: null
+    };
+    var relatedEntity = {
+        uri: 'entities/entity123',
+        type: 'configuration/entityTypes/HCP'
+    };
+    var metadata = {
+        uri: 'configuration',
+        entityTypes: [
+            {
+                uri: 'configuration/entityTypes/HCP',
+                label: 'HCP'
+            },
+            {
+                uri: 'configuration/entityTypes/HCO',
+                label: 'HCO'
             },
-            endObject: {
-                objectTypeURI: 'configuration/entityTypes/HCO',
-                uri: 'configuration/relationTypes/HeadNursing/endObject'
+            {
+                uri: 'configuration/entityTypes/Committee',
+                label: 'Committee'
             },
-            attributes: [
-                {
-                    label: 'Affiliation SubType',
-                    name: 'AffiliationSubType',
-                    type: 'String',
-                    lookupCode: 'AFFILIATION_SUB_TYPE_CD',
-                    uri: 'configuration/relationTypes/HCPtoHCA/attributes/AffiliationSubType'
+            {
+                uri: 'configuration/entityTypes/Location',
+                label: 'Location'
+            }
+        ],
+        relationTypes: [
+            {
+                uri: 'configuration/relationTypes/HeadNursing',
+                label: 'Head of Nursing',
+                startObject: {
+                    objectTypeURI: 'configuration/entityTypes/HCP',
+                    uri: 'configuration/relationTypes/HeadNursing/startObject'
                 },
-                {
-                    label: 'Primary Affiliation Indicator',
-                    name: 'PrimaryAffiliationIndicator',
-                    type: 'Boolean',
-                    uri: 'configuration/relationTypes/HCPtoHCA/attributes/PrimaryAffiliationIndicator'
+                endObject: {
+                    objectTypeURI: 'configuration/entityTypes/HCO',
+                    uri: 'configuration/relationTypes/HeadNursing/endObject'
+                },
+                direction: 'directed',
+                attributes: [
+                    {
+                        label: 'Affiliation SubType',
+                        name: 'AffiliationSubType',
+                        type: 'String',
+                        lookupCode: 'AFFILIATION_SUB_TYPE_CD',
+                        uri: 'configuration/relationTypes/HCPtoHCA/attributes/AffiliationSubType'
+                    },
+                    {
+                        label: 'First name',
+                        name: 'FirstName',
+                        type: 'String',
+                        uri: 'configuration/relationTypes/HCPtoHCA/attributes/FirstName'
+                    },
+                    {
+                        label: 'Primary Affiliation Indicator',
+                        name: 'PrimaryAffiliationIndicator',
+                        type: 'Boolean',
+                        uri: 'configuration/relationTypes/HCPtoHCA/attributes/PrimaryAffiliationIndicator'
+                    },
+                    {
+                        label: 'RelFirstLevelNested',
+                        name: 'RelFirstLevelNested',
+                        type: 'Nested',
+                        uri: 'configuration/relationTypes/HCPtoHCA/attributes/RefFirstLevelNested',
+                        attributes: [
+                            {
+                                label: 'FirstLevelNestedSub1',
+                                name: 'FirstLevelNestedSub1',
+                                type: 'String',
+                                lookupCode: 'LANG_CD',
+                                uri: 'configuration/relationTypes/HCPtoHCA/attributes/RefFirstLevelNested/attributes/FirstLevelNestedSub1'
+                            }
+                        ]
+                    }
+                ]
+            },
+            {
+                uri: 'configuration/relationTypes/HasAddress',
+                label: 'has address',
+                startObject: {
+                    objectTypeURI: 'configuration/entityTypes/HCP',
+                    uri: 'configuration/relationTypes/HasAddress/startObject'
                 },
-                {
-                    label: 'RelFirstLevelNested',
-                    name: 'RelFirstLevelNested',
-                    type: 'Nested',
-                    uri: 'configuration/relationTypes/HCPtoHCA/attributes/RefFirstLevelNested',
-                    attributes: [
-                        {
-                            label: 'FirstLevelNestedSub1',
-                            name: 'FirstLevelNestedSub1',
-                            type: 'String',
-                            lookupCode: 'LANG_CD',
-                            uri: 'configuration/relationTypes/HCPtoHCA/attributes/RefFirstLevelNested/attributes/FirstLevelNestedSub1'
-                        }
-                    ]
+                endObject: {
+                    objectTypeURI: 'configuration/entityTypes/Location',
+                    uri: 'configuration/relationTypes/HasAddress/endObject'
                 }
-            ]
-        },
-        {
-            uri: 'configuration/relationTypes/HasAddress',
-            label: 'has address',
-            startObject: {
-                objectTypeURI: 'configuration/entityTypes/HCP',
-                uri: 'configuration/relationTypes/HasAddress/startObject'
             },
-            endObject: {
-                objectTypeURI: 'configuration/entityTypes/Location',
-                uri: 'configuration/relationTypes/HasAddress/endObject'
+            {
+                uri: 'configuration/relationTypes/IsInCommittee',
+                label: 'is in committee',
+                startObject: {
+                    objectTypeURI: 'configuration/entityTypes/Committee',
+                    uri: 'configuration/relationTypes/IsInCommittee/startObject'
+                },
+                endObject: {
+                    objectTypeURI: 'configuration/entityTypes/HCP',
+                    uri: 'configuration/relationTypes/IsInCommittee/endObject'
+                }
             }
-        },
-        {
-            uri: 'configuration/relationTypes/IsInCommittee',
-            label: 'is in committee',
-            startObject: {
-                objectTypeURI: 'configuration/entityTypes/Committee',
-                uri: 'configuration/relationTypes/IsInCommittee/startObject'
+        ]
+    };
+    var inRelationTypes = [
+        __assign(__assign({}, metadata.relationTypes[2]), { startObject: __assign(__assign({}, metadata.relationTypes[2].startObject), { objectType: {
+                    uri: 'configuration/entityTypes/HCP'
+                } }), endObject: __assign(__assign({}, metadata.relationTypes[2].endObject), { objectType: {
+                    uri: 'configuration/entityTypes/HCO'
+                } }) })
+    ];
+    var outRelationTypes = [
+        __assign(__assign({}, metadata.relationTypes[0]), { startObject: __assign(__assign({}, metadata.relationTypes[0].startObject), { objectType: {
+                    uri: 'configuration/entityTypes/HCP'
+                } }), endObject: __assign(__assign({}, metadata.relationTypes[0].endObject), { objectType: {
+                    uri: 'configuration/entityTypes/HCO'
+                } }) }),
+        __assign(__assign({}, metadata.relationTypes[1]), { startObject: __assign(__assign({}, metadata.relationTypes[1].startObject), { objectType: {
+                    uri: 'configuration/entityTypes/HCP'
+                } }), endObject: __assign(__assign({}, metadata.relationTypes[1].endObject), { objectType: {
+                    uri: 'configuration/entityTypes/Location'
+                } }) })
+    ];
+    var initConnection = {
+        relation: {
+            attributes: {
+                FirstName: [
+                    {
+                        type: 'configuration/relationTypes/HCPtoHCA/attributes/FirstName',
+                        ov: true,
+                        value: 'HCP_5',
+                        uri: 'relations/19oYeyx1/attributes/FirstName/1dg2njmxf'
+                    }
+                ]
             },
-            endObject: {
-                objectTypeURI: 'configuration/entityTypes/HCP',
-                uri: 'configuration/relationTypes/IsInCommittee/endObject'
-            }
-        }
-    ]
-};
-var inRelationTypes = (_a = metadata.relationTypes) === null || _a === void 0 ? void 0 : _a.slice(2, 3);
-var outRelationTypes = (_b = metadata.relationTypes) === null || _b === void 0 ? void 0 : _b.slice(0, 2);
-var initConnection = {
-    relation: {
-        attributes: {
-            FirstName: [
-                {
-                    type: 'configuration/entityTypes/HCP/attributes/FirstName',
-                    ov: true,
-                    value: 'HCP_5',
-                    uri: 'entities/19oYeyx1/attributes/FirstName/1dg2njmxf'
-                }
-            ]
+            uri: 'relation/uri',
+            type: 'configuration/relationTypes/HeadNursing',
+            direction: 'out',
+            startDate: 123
         },
-        uri: 'relation/uri',
-        type: 'configuration/relationTypes/HeadNursing',
-        direction: 'out',
-        startDate: 123
-    },
-    entity: {
-        type: 'configuration/entityTypes/HCO',
-        uri: 'entities/uriHCO'
-    }
-};
-var props = {
-    onChangeAttribute: jest.fn(),
-    onAddAttributes: jest.fn(),
-    onRemoveAttribute: jest.fn(),
-    showEmptyEditors: true
-};
-var mdmValues = {
-    metadata: metadata,
-    globalSearchRequestOptions: {}
-};
-var setUpMount = function (extraProps) {
-    var Component = function (props) { return (react_1.default.createElement(MdmModuleContext_1.MdmModuleProvider, { values: mdmValues },
-        react_1.default.createElement(ConnectionEditor_1.ConnectionEditor, __assign({}, props)))); };
-    return (0, enzyme_1.mount)(react_1.default.createElement(Component, __assign({}, props, extraProps)));
-};
-describe('ConnectionEditor test', function () {
+        entity: {
+            type: 'configuration/entityTypes/HCO',
+            uri: 'entities/uriHCO',
+            label: 'initRelationEntity'
+        }
+    };
+    var mdmValues = {
+        metadata: metadata,
+        globalSearchRequestOptions: {},
+        dependentLookups: { editors: {}, structure: {}, initedTypeUris: [] },
+        lookups: {}
+    };
+    var onChangeAttribute = jest.fn();
+    var onAddAttributes = jest.fn();
+    var onRemoveAttribute = jest.fn();
+    var onChangeRelationType = jest.fn();
+    var onChangeEntity = jest.fn();
+    var defaultProps = {
+        connection: initConnection,
+        onChangeAttribute: onChangeAttribute,
+        onAddAttributes: onAddAttributes,
+        onRemoveAttribute: onRemoveAttribute,
+        showEmptyEditors: true
+    };
+    var propsWithRelations = __assign(__assign({}, defaultProps), { connection: newConnection, relatedEntity: relatedEntity, inRelationTypes: inRelationTypes, outRelationTypes: outRelationTypes, onChangeRelationType: onChangeRelationType, onChangeEntity: onChangeEntity });
+    var setUp = function (props) {
+        if (props === void 0) { props = defaultProps; }
+        var Providers = function (_a) {
+            var children = _a.children;
+            return (react_1.default.createElement(LocalizationProvider_1.LocalizationProvider, { dateAdapter: AdapterMoment_1.AdapterMoment },
+                react_1.default.createElement(MdmModuleContext_1.MdmModuleProvider, { values: mdmValues }, children)));
+        };
+        var user = user_event_1.default.setup();
+        return __assign({ user: user }, (0, react_2.render)(react_1.default.createElement(ConnectionEditor_1.ConnectionEditor, __assign({}, props)), { wrapper: Providers }));
+    };
+    afterEach(function () {
+        jest.clearAllMocks();
+    });
     it('should render EditModeAttributesList', function () {
-        var _a;
-        var component = setUpMount({ connection: initConnection });
-        expect(component.find(EditModeAttributesList_1.EditModeAttributesList)).toHaveLength(1);
-        expect(component.find(EditModeAttributesList_1.EditModeAttributesList).prop('attrTypes')).toEqual(__spreadArray(__spreadArray([], mdm_sdk_1.ACTIVENESS_ATTR_TYPES, true), (((_a = metadata.relationTypes) === null || _a === void 0 ? void 0 : _a[0].attributes) || []), true));
-        expect(component.find(EditModeAttributesList_1.EditModeAttributesList).prop('entity')).toEqual({
-            attributes: initConnection.relation.attributes,
-            startDate: 123
-        });
-        expect(component.find(EditModeAttributesList_1.EditModeAttributesList).prop('showEmptyEditors')).toBe(true);
-        expect(component.find(EditModeAttributesList_1.EditModeAttributesList).prop('mode')).toBe('Editing');
-        expect(component.find(EditModeAttributesList_1.EditModeAttributesList).prop('parentUri')).toBe(initConnection.relation.uri);
-        expect(component.find(EditModeAttributesList_1.EditModeAttributesList).prop('onAddAttributes')).toBe(props.onAddAttributes);
-        expect(component.find(EditModeAttributesList_1.EditModeAttributesList).prop('onChangeAttribute')).toBe(props.onChangeAttribute);
-        expect(component.find(EditModeAttributesList_1.EditModeAttributesList).prop('onDeleteAttribute')).toBe(props.onRemoveAttribute);
+        setUp();
+        expect(react_2.screen.queryByText('First name')).toBeInTheDocument();
+        expect(react_2.screen.queryAllByTestId('reltio-edit-attribute')).toHaveLength(6);
     });
+    it('should call onAddAttributes', function () { return __awaiter(void 0, void 0, void 0, function () {
+        var user;
+        return __generator(this, function (_a) {
+            switch (_a.label) {
+                case 0:
+                    user = setUp().user;
+                    return [4 /*yield*/, user.click(react_2.screen.getByRole('button', { name: 'More attributes' }))];
+                case 1:
+                    _a.sent();
+                    return [4 /*yield*/, user.click(react_2.screen.getByRole('menuitem', { name: 'First name' }))];
+                case 2:
+                    _a.sent();
+                    return [4 /*yield*/, user.click(react_2.screen.getByRole('presentation').firstChild)];
+                case 3:
+                    _a.sent();
+                    expect(onAddAttributes).toHaveBeenCalledWith([
+                        { attributeType: metadata.relationTypes[0].attributes[1], parentUri: initConnection.relation.uri }
+                    ]);
+                    return [2 /*return*/];
+            }
+        });
+    }); });
+    it('should call onChangeAttribute', function () { return __awaiter(void 0, void 0, void 0, function () {
+        var user, firstNameValueInput;
+        return __generator(this, function (_a) {
+            switch (_a.label) {
+                case 0:
+                    user = setUp().user;
+                    firstNameValueInput = react_2.screen.getAllByRole('textbox')[2];
+                    return [4 /*yield*/, user.type(firstNameValueInput, 'newValue{Enter}')];
+                case 1:
+                    _a.sent();
+                    expect(onChangeAttribute).toHaveBeenCalledWith({
+                        attributeType: metadata.relationTypes[0].attributes[1],
+                        uri: initConnection.relation.attributes.FirstName[0].uri,
+                        value: 'HCP_5newValue'
+                    });
+                    return [2 /*return*/];
+            }
+        });
+    }); });
+    it('should call onRemoveAttribute', function () { return __awaiter(void 0, void 0, void 0, function () {
+        var user, firstNameDeleteButton;
+        return __generator(this, function (_a) {
+            switch (_a.label) {
+                case 0:
+                    user = setUp().user;
+                    firstNameDeleteButton = (0, test_utils_1.getMuiIconsByName)('Delete')[3];
+                    return [4 /*yield*/, user.click(firstNameDeleteButton)];
+                case 1:
+                    _a.sent();
+                    expect(onRemoveAttribute).toHaveBeenCalledWith({
+                        attributeType: metadata.relationTypes[0].attributes[1],
+                        uri: initConnection.relation.attributes.FirstName[0].uri
+                    });
+                    return [2 /*return*/];
+            }
+        });
+    }); });
     it('should render ConnectionRelationTypeSelector and EntitySelector', function () {
-        var props = {
-            connection: initConnection,
-            relatedEntity: relatedEntity,
-            inRelationTypes: inRelationTypes,
-            outRelationTypes: outRelationTypes,
-            onChangeRelationType: jest.fn(),
-            onChangeEntity: jest.fn()
-        };
-        var component = setUpMount(props);
-        expect(component.find(ConnectionRelationTypeSelector_1.ConnectionRelationTypeSelector)).toHaveLength(1);
-        expect(component.find(ErrorWrapper_1.ErrorWrapper)).toHaveLength(1);
-        expect(component.find(EntitySelector_1.EntitySelector)).toHaveLength(1);
-        var compatibleRelation = {
-            relationType: initConnection.relation.type,
-            direction: initConnection.relation.direction
-        };
-        expect(component.find(ConnectionRelationTypeSelector_1.ConnectionRelationTypeSelector).prop('relation')).toEqual(compatibleRelation);
-        expect(component.find(EntitySelector_1.EntitySelector).prop('onChange')).toBe(props.onChangeEntity);
-        var compatibleEntity = {
-            entityUri: initConnection.entity.uri,
-            entityType: initConnection.entity.type,
-            entityLabel: '<No label>'
-        };
-        expect(component.find(EntitySelector_1.EntitySelector).prop('entity')).toEqual(compatibleEntity);
-        expect(component.find(EntitySelector_1.EntitySelector).prop('entityTypesUris')).toEqual(['configuration/entityTypes/HCO']);
-    });
-    it('should set entityTypesUris prop for EntitySelector', function () {
-        var props = {
-            connection: newConnection,
-            relatedEntity: relatedEntity,
-            inRelationTypes: inRelationTypes,
-            outRelationTypes: outRelationTypes,
-            onChangeRelationType: jest.fn(),
-            onChangeEntity: jest.fn()
-        };
-        var component = setUpMount(props);
-        expect(component.find(EntitySelector_1.EntitySelector).prop('entityTypesUris')).toEqual([
-            'configuration/entityTypes/HCO',
-            'configuration/entityTypes/Location',
-            'configuration/entityTypes/Committee'
-        ]);
+        setUp(__assign(__assign({}, propsWithRelations), { connection: initConnection }));
+        expect(react_2.screen.queryByText('Head of Nursing')).toBeInTheDocument();
+        expect(react_2.screen.queryByText('Select HCO profile')).toBeInTheDocument();
     });
+    it('should call onChangeEntity for EntitySelector', function () { return __awaiter(void 0, void 0, void 0, function () {
+        var newEntity, wrongEntity, user, profileSelector;
+        return __generator(this, function (_a) {
+            switch (_a.label) {
+                case 0:
+                    newEntity = {
+                        uri: 'entities/entityCommittee',
+                        type: 'configuration/entityTypes/Committee',
+                        label: 'Committee'
+                    };
+                    wrongEntity = {
+                        uri: 'entities/wrongEntity',
+                        type: 'configuration/entityTypes/HCP',
+                        label: 'WrongEntity'
+                    };
+                    mdm_sdk_1.typeAheadSearch.mockResolvedValueOnce([newEntity]);
+                    user = setUp(propsWithRelations).user;
+                    profileSelector = react_2.screen.getAllByRole('textbox')[1];
+                    return [4 /*yield*/, user.click(profileSelector)];
+                case 1:
+                    _a.sent();
+                    expect(react_2.screen.queryByText(wrongEntity.label)).not.toBeInTheDocument();
+                    return [4 /*yield*/, user.click(react_2.screen.getByRole('menuitem', { name: 'C Committee' }))];
+                case 2:
+                    _a.sent();
+                    expect(onChangeEntity).toHaveBeenCalledWith(__assign(__assign({}, newEntity), { entityType: metadata.entityTypes[2] }));
+                    return [2 /*return*/];
+            }
+        });
+    }); });
     it('should call onChangeRelationType for ConnectionRelationTypeSelector, if relation type was changed', function () { return __awaiter(void 0, void 0, void 0, function () {
-        var onChangeRelationType, component, newRelationType;
+        var user, relationshipTypeSelector;
         return __generator(this, function (_a) {
-            onChangeRelationType = jest.fn();
-            component = setUpMount({
-                connection: newConnection,
-                relatedEntity: relatedEntity,
-                inRelationTypes: inRelationTypes,
-                outRelationTypes: outRelationTypes,
-                onChangeRelationType: onChangeRelationType,
-                onChangeEntity: jest.fn()
-            });
-            expect(onChangeRelationType).not.toHaveBeenCalledWith();
-            newRelationType = {
-                type: { uri: 'configuration/relationTypes/IsInCommittee' },
-                direction: mdm_sdk_1.Directions.OUT
-            };
-            (0, test_utils_1.act)(function () {
-                component.find(ConnectionRelationTypeSelector_1.ConnectionRelationTypeSelector).prop('onChange')(newRelationType);
-            });
-            component.update();
-            expect(onChangeRelationType).toHaveBeenCalledWith({
-                type: newRelationType.type.uri,
-                direction: newRelationType.direction
-            });
-            return [2 /*return*/];
+            switch (_a.label) {
+                case 0:
+                    user = setUp(propsWithRelations).user;
+                    expect(onChangeRelationType).not.toHaveBeenCalled();
+                    relationshipTypeSelector = react_2.screen.getAllByRole('textbox')[0];
+                    return [4 /*yield*/, user.click(relationshipTypeSelector)];
+                case 1:
+                    _a.sent();
+                    return [4 /*yield*/, user.click(react_2.screen.getByRole('menuitem', { name: 'H L has address undefined ↔ undefined' }))];
+                case 2:
+                    _a.sent();
+                    expect(onChangeRelationType).toHaveBeenCalledWith({
+                        direction: mdm_sdk_1.Directions.OUT,
+                        type: 'configuration/relationTypes/HasAddress'
+                    });
+                    return [2 /*return*/];
+            }
         });
     }); });
     it('should call onChangeRelationType on mount for ConnectionRelationTypeSelector, if available relation type is only one', function () { return __awaiter(void 0, void 0, void 0, function () {
-        var onChangeRelationType, component;
         return __generator(this, function (_a) {
-            onChangeRelationType = jest.fn();
-            component = setUpMount({
-                connection: newConnection,
-                relatedEntity: relatedEntity,
-                inRelationTypes: inRelationTypes === null || inRelationTypes === void 0 ? void 0 : inRelationTypes.slice(0, 1),
-                outRelationTypes: [],
-                onChangeRelationType: onChangeRelationType,
-                onChangeEntity: jest.fn()
-            });
-            expect(component.find(ConnectionRelationTypeSelector_1.ConnectionRelationTypeSelector)).toHaveLength(1);
+            setUp(__assign(__assign({}, propsWithRelations), { inRelationTypes: inRelationTypes === null || inRelationTypes === void 0 ? void 0 : inRelationTypes.slice(0, 1), outRelationTypes: [] }));
             expect(onChangeRelationType).toHaveBeenCalledWith({
                 type: 'configuration/relationTypes/IsInCommittee',
-                direction: 'in'
+                direction: mdm_sdk_1.Directions.IN
             });
             return [2 /*return*/];
         });
     }); });
     it('should call onChangeRelationType for ConnectionRelationTypeSelector, after entity selected and compatible relation type is only one', function () { return __awaiter(void 0, void 0, void 0, function () {
-        var onChangeRelationType, component;
+        var rerender, newProps;
         return __generator(this, function (_a) {
-            onChangeRelationType = jest.fn();
-            component = setUpMount({
-                connection: newConnection,
-                relatedEntity: relatedEntity,
-                inRelationTypes: inRelationTypes,
-                outRelationTypes: outRelationTypes,
-                onChangeRelationType: onChangeRelationType,
-                onChangeEntity: jest.fn()
-            });
-            expect(component.find(ConnectionRelationTypeSelector_1.ConnectionRelationTypeSelector)).toHaveLength(1);
-            component.setProps({
-                connection: __assign(__assign({}, newConnection), { entity: {
+            rerender = setUp(propsWithRelations).rerender;
+            newProps = __assign(__assign({}, propsWithRelations), { connection: __assign(__assign({}, newConnection), { entity: {
                         uri: 'entities/entityCommittee',
                         type: 'configuration/entityTypes/Committee'
-                    } })
-            });
-            component.update();
+                    } }) });
+            rerender(react_1.default.createElement(ConnectionEditor_1.ConnectionEditor, __assign({}, newProps)));
             expect(onChangeRelationType).toHaveBeenCalledWith({
                 type: 'configuration/relationTypes/IsInCommittee',
                 direction: 'in'
@@ -347,30 +390,4 @@ describe('ConnectionEditor test', function () {
             return [2 /*return*/];
         });
     }); });
-    it('should call onChangeEntity for EntitySelector', function () { return __awaiter(void 0, void 0, void 0, function () {
-        var onChangeEntity, component, newEntity;
-        return __generator(this, function (_a) {
-            onChangeEntity = jest.fn();
-            component = setUpMount({
-                connection: newConnection,
-                relatedEntity: relatedEntity,
-                inRelationTypes: inRelationTypes,
-                outRelationTypes: outRelationTypes,
-                onChangeRelationType: jest.fn(),
-                onChangeEntity: onChangeEntity
-            });
-            newEntity = {
-                uri: 'entities/entityCommittee',
-                type: 'configuration/entityTypes/Committee',
-                label: 'Committee',
-                entityType: { uri: 'configuration/entityTypes/Committee' }
-            };
-            (0, test_utils_1.act)(function () {
-                component.find(EntitySelector_1.EntitySelector).prop('onChange')(newEntity);
-            });
-            component.update();
-            expect(onChangeEntity).toHaveBeenCalledWith(newEntity);
-            return [2 /*return*/];
-        });
-    }); });
 });