@reltio/components 1.4.1905 → 1.4.1907

package/cjs/CollaborationItem/CollaborationItem.test.js

@@ -15,12 +15,10 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 var react_1 = __importDefault(require("react"));
-var enzyme_1 = require("enzyme");
 var mdm_sdk_1 = require("@reltio/mdm-sdk");
-var Comment_1 = require("./components/Comment");
-var RepliedComment_1 = require("./components/RepliedComment");
-var SendMessageArea_1 = require("./components/SendMessageArea");
 var CollaborationItem_1 = __importDefault(require("./CollaborationItem"));
+var react_2 = require("@testing-library/react");
+var MdmModuleContext_1 = require("../contexts/MdmModuleContext");
 var createComment = function (hasReplies) {
     if (hasReplies === void 0) { hasReplies = false; }
     var comment = {
@@ -44,12 +42,12 @@ var createComment = function (hasReplies) {
         comment.replies.push({
             selfLink: 'https://cservice-dev.reltio.com/tst-01/sokol/comments/00001lo/replies/000007s',
             replyId: '000007s',
-            createdBy: 'vyacheslav.kovyazin',
+            createdBy: 'other.user',
             createdTime: 1620280061505,
             modifiedTime: 1620280061505,
             deleted: false,
             action: 'none',
-            content: 'abc',
+            content: 'other comment',
             namedUsers: []
         });
     }
@@ -63,37 +61,76 @@ var createProps = function (comment) {
         objectType: mdm_sdk_1.CollaborationObjectTypes.ENTITY_ATTRIBUTE
     };
 };
-var setUp = function (props) { return (0, enzyme_1.shallow)(react_1.default.createElement(CollaborationItem_1.default, __assign({}, props))); };
 describe('CollaborationItem tests', function () {
+    var entity = {
+        type: 'configuration/entityTypes/HCP',
+        uri: 'entities/02ZRRE9',
+        attributes: {
+            FirstName: [
+                {
+                    type: 'configuration/entityTypes/HCP/attributes/FirstName',
+                    ov: true,
+                    value: 'nameValue',
+                    uri: 'entities/02ZRRE9/attributes/FirstName/15p60Tel'
+                }
+            ]
+        }
+    };
+    var metadata = {
+        entityTypes: [
+            {
+                uri: 'configuration/entityTypes/HCP',
+                attributes: [
+                    {
+                        uri: 'configuration/entityTypes/HCP/attributes/FirstName',
+                        type: 'String',
+                        name: 'FirstName',
+                        label: 'First Name',
+                        access: ['READ', 'CREATE', 'UPDATE', 'INITIATE_CHANGE_REQUEST']
+                    }
+                ]
+            }
+        ]
+    };
+    var actions = {
+        removeAttribute: function () { }
+    };
+    var mdmDefaultValues = {
+        user: {
+            username: 'test.testmail'
+        },
+        entity: entity,
+        metadata: metadata,
+        modifiedEntities: { 'entities/02ZRRE9': entity },
+        dependentLookups: { editors: {}, structure: {}, initedTypeUris: [] }
+    };
+    var setUp = function (props) {
+        var Providers = function (_a) {
+            var children = _a.children;
+            return (react_1.default.createElement(MdmModuleContext_1.MdmModuleProvider, { values: mdmDefaultValues, actions: actions }, children));
+        };
+        return (0, react_2.render)(react_1.default.createElement(CollaborationItem_1.default, __assign({}, props)), { wrapper: Providers });
+    };
     it('should render Comment correctly if comment exist', function () {
-        var component = setUp(createProps(createComment()));
-        expect(component.find(Comment_1.Comment)).toHaveLength(1);
-        expect(component.find(Comment_1.Comment).prop('uri')).toEqual('entities/13JKIKAz/attributes/Name/2mlWuqDq5');
+        setUp(createProps(createComment()));
+        expect(react_2.screen.queryByText('VK')).toBeInTheDocument();
+        expect(react_2.screen.queryByText('vyacheslav.kovyazin')).toBeInTheDocument();
+        expect(react_2.screen.queryByText('comment')).toBeInTheDocument();
     });
     it('should not render Comment and RepliedComment if comment do not exist', function () {
-        var component = setUp(createProps());
-        expect(component.find(Comment_1.Comment)).toHaveLength(0);
-        expect(component.find(RepliedComment_1.RepliedComment)).toHaveLength(0);
+        var container = setUp(createProps()).container;
+        expect(container.querySelector('.comment')).not.toBeInTheDocument();
+        expect(container.querySelector('.replies')).not.toBeInTheDocument();
     });
     it('should render SendMessageArea correctly', function () {
-        var component = setUp(createProps(createComment()));
-        expect(component.find(SendMessageArea_1.SendMessageArea).prop('uri')).toBe('entities/13JKIKAz/attributes/Name/2mlWuqDq5');
-        expect(component.find(SendMessageArea_1.SendMessageArea).prop('objectType')).toBe('entityAttribute');
-        expect(component.find(SendMessageArea_1.SendMessageArea).prop('relatedObjectUris')).toEqual(['entities/13JKIKAz']);
-    });
-    it('should not render replied comments correctly if they do not exist', function () {
-        var component = setUp(createProps(createComment(false)));
-        expect(component.find(RepliedComment_1.RepliedComment)).toHaveLength(0);
+        var container = setUp(createProps(createComment())).container;
+        expect(react_2.screen.queryByText('TT')).toBeInTheDocument();
+        expect(container.querySelector('.sendMessageArea')).toBeInTheDocument();
     });
     it('should render replied comments correctly if they do exist', function () {
-        var component = setUp(createProps(createComment(true)));
-        expect(component.find(RepliedComment_1.RepliedComment)).toHaveLength(1);
-        expect(component.find(RepliedComment_1.RepliedComment).prop('reply')).toMatchObject({
-            createdTime: 1620280061505,
-            content: 'abc',
-            createdBy: 'vyacheslav.kovyazin',
-            action: 'none',
-            namedUsers: []
-        });
+        setUp(createProps(createComment(true)));
+        expect(react_2.screen.queryByText('OU')).toBeInTheDocument();
+        expect(react_2.screen.queryByText('other.user')).toBeInTheDocument();
+        expect(react_2.screen.queryByText('other comment')).toBeInTheDocument();
     });
 });

package/cjs/CollaborationItem/components/Comment/Comment.test.js

@@ -51,234 +51,268 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 var react_1 = __importDefault(require("react"));
-var enzyme_1 = require("enzyme");
 var react_2 = require("@testing-library/react");
 var user_event_1 = __importDefault(require("@testing-library/user-event"));
-var test_utils_1 = require("react-dom/test-utils");
 var mdm_sdk_1 = require("@reltio/mdm-sdk");
-var Link_1 = __importDefault(require("@mui/material/Link"));
-var DropDownMenuButton_1 = require("../../../DropDownMenuButton");
 var CollaborationContext_1 = require("../../../contexts/CollaborationContext");
 var MdmModuleContext_1 = require("../../../contexts/MdmModuleContext");
 var helpers_1 = require("../../helpers");
-var Avatar_1 = require("../Avatar");
-var SendMessageArea_1 = require("../SendMessageArea");
-var CommentContent_1 = require("../CommentContent");
 var Comment_1 = require("./Comment");
-jest.mock('../CommentTarget', function () { return ({
-    CommentTarget: function () { return null; }
-}); });
-var TIME = 1620280049022;
-var createComment = function (status, createdBy) {
-    if (status === void 0) { status = 'open'; }
-    if (createdBy === void 0) { createdBy = 'vyacheslav.kovyazin'; }
-    return {
-        createdBy: createdBy,
-        createdTime: TIME,
-        status: status,
-        objectId: 'entities/13JKIKAz/attributes/Name/2mlWuqDq5',
-        objectType: mdm_sdk_1.CollaborationObjectTypes.ENTITY_ATTRIBUTE,
-        content: 'comment',
-        relatedObjectUris: ['entities/13JKIKAz'],
-        commentId: 'commentId',
-        namedUsers: [],
-        replies: [],
-        modifiedTime: 0,
-        deleted: false,
-        visibility: '',
-        selfLink: '',
-        permanentLink: ''
+describe('Comment tests', function () {
+    var TIME = 1620280049022;
+    var createComment = function (status, createdBy) {
+        if (status === void 0) { status = 'open'; }
+        if (createdBy === void 0) { createdBy = 'vyacheslav.kovyazin'; }
+        return {
+            createdBy: createdBy,
+            createdTime: TIME,
+            status: status,
+            objectId: 'entities/13JKIKAz/attributes/Name/2mlWuqDq5',
+            objectType: mdm_sdk_1.CollaborationObjectTypes.ENTITY_ATTRIBUTE,
+            content: 'comment',
+            relatedObjectUris: ['entities/13JKIKAz'],
+            commentId: 'commentId',
+            namedUsers: [],
+            replies: [],
+            modifiedTime: 0,
+            deleted: false,
+            visibility: '',
+            selfLink: '',
+            permanentLink: ''
+        };
     };
-};
-var defaultCurrentComment = createComment();
-var defaultCommentsMap = {
-    uri1: [
-        {
-            commentId: defaultCurrentComment.commentId,
-            status: 'open',
-            replies: 0
+    var defaultCurrentComment = createComment();
+    var defaultCommentsMap = {
+        uri1: [
+            {
+                commentId: defaultCurrentComment.commentId,
+                status: 'open',
+                replies: 0
+            }
+        ]
+    };
+    var defaultUsername = 'vyacheslav.kovyazin';
+    var entity = {
+        type: 'configuration/entityTypes/HCP',
+        uri: 'entities/02ZRRE9',
+        attributes: {
+            FirstName: [
+                {
+                    type: 'configuration/entityTypes/HCP/attributes/FirstName',
+                    ov: true,
+                    value: 'nameValue',
+                    uri: 'entities/02ZRRE9/attributes/FirstName/15p60Tel'
+                }
+            ]
         }
-    ]
-};
-var defaultUsername = 'vyacheslav.kovyazin';
-var defaultMdmValues = {
-    user: {
-        username: defaultUsername
-    }
-};
-var defaultCollaborationContextValue = {
-    commentsMap: defaultCommentsMap,
-    currentComment: defaultCurrentComment,
-    createComment: function () { return Promise.resolve(); },
-    editComment: function () { return Promise.resolve(); },
-    editReply: function () { return Promise.resolve(); },
-    deleteComment: jest.fn(),
-    deleteReply: function () { },
-    createReply: function () { return Promise.resolve(); },
-    reOpenThread: jest.fn(),
-    resolveThread: jest.fn(),
-    getComment: jest.fn(),
-    clearCurrentComment: jest.fn(),
-    sending: false,
-    loading: false,
-    pageToken: '',
-    getComments: function () { return Promise.resolve(); },
-    comments: [],
-    getCommentsCount: function () { return Promise.resolve(); },
-    getCommentState: function () { return ({}); },
-    updateCommentState: jest.fn(),
-    clearCommentState: function () { }
-};
-var defaultProps = {
-    comment: defaultCurrentComment,
-    onReply: jest.fn(),
-    uri: 'entities/13JKIKAz/attributes/Name/2mlWuqDq5',
-    relatedObjectUris: ['entities/13JKIKAz'],
-    objectType: mdm_sdk_1.CollaborationObjectTypes.ENTITY_ATTRIBUTE
-};
-var setUp = function (_a) {
-    var _b = _a === void 0 ? {} : _a, _c = _b.props, props = _c === void 0 ? defaultProps : _c, _d = _b.mdmValues, mdmValues = _d === void 0 ? defaultMdmValues : _d, _e = _b.collaborationContextValue, collaborationContextValue = _e === void 0 ? defaultCollaborationContextValue : _e;
-    return (0, enzyme_1.mount)(react_1.default.createElement(MdmModuleContext_1.MdmModuleProvider, { values: mdmValues },
-        react_1.default.createElement(CollaborationContext_1.CollaborationContext.Provider, { value: collaborationContextValue },
-            react_1.default.createElement(Comment_1.Comment, __assign({}, props)))));
-};
-var findLastElementByClassName = function (component, className) {
-    var elements = component.find(className);
-    return elements.at(elements.length - 1);
-};
-describe('Comment tests', function () {
-    afterEach(function () {
-        jest.resetAllMocks();
-    });
+    };
+    var metadata = {
+        entityTypes: [
+            {
+                uri: 'configuration/entityTypes/HCP',
+                attributes: [
+                    {
+                        uri: 'configuration/entityTypes/HCP/attributes/FirstName',
+                        type: 'String',
+                        name: 'FirstName',
+                        label: 'First Name',
+                        access: ['READ', 'CREATE', 'UPDATE', 'INITIATE_CHANGE_REQUEST']
+                    }
+                ]
+            }
+        ]
+    };
+    var defaultMdmValues = {
+        user: {
+            username: defaultUsername
+        },
+        entity: entity,
+        metadata: metadata
+    };
+    var defaultCollaborationContextValue = {
+        commentsMap: defaultCommentsMap,
+        currentComment: defaultCurrentComment,
+        createComment: function () { return Promise.resolve(); },
+        editComment: function () { return Promise.resolve(); },
+        editReply: function () { return Promise.resolve(); },
+        deleteComment: jest.fn(),
+        deleteReply: function () { },
+        createReply: function () { return Promise.resolve(); },
+        reOpenThread: jest.fn(),
+        resolveThread: jest.fn(),
+        getComment: jest.fn(),
+        clearCurrentComment: jest.fn(),
+        sending: false,
+        loading: false,
+        pageToken: '',
+        getComments: function () { return Promise.resolve(); },
+        comments: [],
+        getCommentsCount: function () { return Promise.resolve(); },
+        getCommentState: function () { return ({}); },
+        updateCommentState: jest.fn(),
+        clearCommentState: function () { }
+    };
+    var defaultProps = {
+        comment: defaultCurrentComment,
+        onReply: jest.fn(),
+        uri: 'entities/13JKIKAz/attributes/Name/2mlWuqDq5',
+        relatedObjectUris: ['entities/13JKIKAz'],
+        objectType: mdm_sdk_1.CollaborationObjectTypes.ENTITY_ATTRIBUTE
+    };
+    var setUp = function (_a) {
+        var _b = _a === void 0 ? {} : _a, _c = _b.props, props = _c === void 0 ? defaultProps : _c, _d = _b.mdmValues, mdmValues = _d === void 0 ? defaultMdmValues : _d, _e = _b.collaborationContextValue, collaborationContextValue = _e === void 0 ? defaultCollaborationContextValue : _e;
+        var Providers = function (_a) {
+            var children = _a.children;
+            return (react_1.default.createElement(MdmModuleContext_1.MdmModuleProvider, { values: mdmValues },
+                react_1.default.createElement(CollaborationContext_1.CollaborationContext.Provider, { value: collaborationContextValue }, children)));
+        };
+        var user = user_event_1.default.setup();
+        return __assign({ user: user }, (0, react_2.render)(react_1.default.createElement(Comment_1.Comment, __assign({}, props)), { wrapper: Providers }));
+    };
     it('should render avatar correctly', function () {
-        var component = setUp();
-        expect(component.find(Avatar_1.Avatar)).toHaveLength(1);
-        expect(component.find(Avatar_1.Avatar).prop('children')).toBe('vyacheslav.kovyazin');
+        setUp();
+        expect(react_2.screen.queryByText('VK')).toBeInTheDocument();
     });
     it('should render username correctly', function () {
-        var component = setUp();
-        expect(component.find('.username').length).toBeTruthy();
-        expect(component.find('.username').at(0).text()).toBe('vyacheslav.kovyazin');
+        setUp();
+        expect(react_2.screen.queryByText('vyacheslav.kovyazin')).toBeInTheDocument();
     });
     it('should render time correctly', function () {
-        var component = setUp();
-        expect(component.find('.createdAt').length).toBeTruthy();
-        expect(component.find('.createdAt').at(0).text()).toBe((0, helpers_1.formatDate)(TIME));
+        setUp();
+        expect(react_2.screen.queryByText((0, helpers_1.formatDate)(TIME))).toBeInTheDocument();
     });
     it('should render CommentContent correctly', function () {
-        var component = setUp();
-        expect(component.find('.message').length).toBeTruthy();
-        expect(component.find('.message').at(0).props()).toMatchObject({
-            comment: {
-                content: 'comment',
-                namedUsers: []
-            }
-        });
+        setUp();
+        expect(react_2.screen.queryByText('comment')).toBeInTheDocument();
     });
     it('should render buttons correctly if status is "open"', function () {
-        var component = setUp();
-        var buttons = findLastElementByClassName(component, '.buttons');
-        expect(buttons.children()).toHaveLength(2);
-        expect(buttons.childAt(0).text()).toBe('Reply');
-        expect(buttons.childAt(1).text()).toBe('Resolve');
+        setUp();
+        expect(react_2.screen.queryByText('Reply')).toBeInTheDocument();
+        expect(react_2.screen.queryByText('Resolve')).toBeInTheDocument();
+        expect(react_2.screen.queryByText('Re-open')).not.toBeInTheDocument();
     });
     it('should render buttons correctly if status is "resolved"', function () {
         var props = __assign(__assign({}, defaultProps), { comment: createComment('resolved') });
-        var component = setUp({ props: props });
-        var buttons = findLastElementByClassName(component, '.buttons');
-        expect(buttons.children()).toHaveLength(2);
-        expect(buttons.childAt(0).text()).toBe('Reply');
-        expect(buttons.childAt(1).text()).toBe('Re-open');
+        setUp({ props: props });
+        expect(react_2.screen.queryByText('Reply')).toBeInTheDocument();
+        expect(react_2.screen.queryByText('Re-open')).toBeInTheDocument();
+        expect(react_2.screen.queryByText('Resolve')).not.toBeInTheDocument();
     });
-    it('onReply handler should work correctly', function () {
-        var onReply = defaultProps.onReply;
-        var component = setUp();
-        var buttons = findLastElementByClassName(component, '.buttons');
-        expect(onReply).toHaveBeenCalledTimes(0);
-        (0, test_utils_1.act)(function () {
-            var onClick = buttons.childAt(0).prop('onClick');
-            onClick();
+    it('onReply handler should work correctly', function () { return __awaiter(void 0, void 0, void 0, function () {
+        var onReply, user;
+        return __generator(this, function (_a) {
+            switch (_a.label) {
+                case 0:
+                    onReply = defaultProps.onReply;
+                    user = setUp().user;
+                    expect(onReply).toHaveBeenCalledTimes(0);
+                    return [4 /*yield*/, user.click(react_2.screen.getByText('Reply'))];
+                case 1:
+                    _a.sent();
+                    expect(onReply).toHaveBeenCalledTimes(1);
+                    return [2 /*return*/];
+            }
         });
-        expect(onReply).toHaveBeenCalledTimes(1);
-    });
+    }); });
     it('should disable "Resolve" button if sending is "true"', function () {
         var collaborationContextValue = __assign(__assign({}, defaultCollaborationContextValue), { sending: true });
-        var component = setUp({ collaborationContextValue: collaborationContextValue });
-        var resolveBtn = component.find(Link_1.default).filterWhere(function (button) { return button.text() === 'Resolve'; });
-        expect(resolveBtn.prop('disabled')).toBe(true);
+        setUp({ collaborationContextValue: collaborationContextValue });
+        expect(react_2.screen.getByText('Resolve')).toBeDisabled();
     });
     it('should disable "Re-Open" button if sending is "true"', function () {
         var collaborationContextValue = __assign(__assign({}, defaultCollaborationContextValue), { sending: true });
         var props = __assign(__assign({}, defaultProps), { comment: createComment('resolved') });
-        var component = setUp({ props: props, collaborationContextValue: collaborationContextValue });
-        var resolveBtn = component.find(Link_1.default).filterWhere(function (button) { return button.text() === 'Re-open'; });
-        expect(resolveBtn.prop('disabled')).toBe(true);
-    });
-    it('handleResolve handler should work correctly', function () {
-        var resolveThread = defaultCollaborationContextValue.resolveThread;
-        var component = setUp();
-        var resolvedBtn = component.find(Link_1.default).filterWhere(function (button) { return button.text() === 'Resolve'; });
-        resolvedBtn.find('button').simulate('click');
-        expect(resolveThread).toHaveBeenCalledWith({
-            commentId: 'commentId',
-            uri: 'entities/13JKIKAz/attributes/Name/2mlWuqDq5'
-        });
+        setUp({ props: props, collaborationContextValue: collaborationContextValue });
+        expect(react_2.screen.getByText('Re-open')).toBeDisabled();
     });
-    it('handleReOpen handler should work correctly', function () {
-        var reOpenThread = defaultCollaborationContextValue.reOpenThread;
-        var props = __assign(__assign({}, defaultProps), { comment: createComment('resolved') });
-        var component = setUp({ props: props });
-        var reOpenBtn = component.find(Link_1.default).filterWhere(function (button) { return button.text() === 'Re-open'; });
-        reOpenBtn.find('button').simulate('click');
-        expect(reOpenThread).toHaveBeenCalledWith({
-            commentId: 'commentId',
-            uri: 'entities/13JKIKAz/attributes/Name/2mlWuqDq5'
+    it('handleResolve handler should work correctly', function () { return __awaiter(void 0, void 0, void 0, function () {
+        var resolveThread, user;
+        return __generator(this, function (_a) {
+            switch (_a.label) {
+                case 0:
+                    resolveThread = defaultCollaborationContextValue.resolveThread;
+                    user = setUp().user;
+                    return [4 /*yield*/, user.click(react_2.screen.getByText('Resolve'))];
+                case 1:
+                    _a.sent();
+                    expect(resolveThread).toHaveBeenCalledWith({
+                        commentId: 'commentId',
+                        uri: 'entities/13JKIKAz/attributes/Name/2mlWuqDq5'
+                    });
+                    return [2 /*return*/];
+            }
         });
-    });
-    it('should call updateCommentState with right params when onCancel called', function () {
-        var updateCommentState = defaultCollaborationContextValue.updateCommentState;
-        var getCommentState = function () { return ({ isEditing: true }); };
-        var collaborationContextValue = __assign(__assign({}, defaultCollaborationContextValue), { getCommentState: getCommentState });
-        var component = setUp({ collaborationContextValue: collaborationContextValue });
-        (0, test_utils_1.act)(function () {
-            var _a;
-            (_a = component.find(SendMessageArea_1.SendMessageArea).prop('onCancel')) === null || _a === void 0 ? void 0 : _a();
+    }); });
+    it('handleReOpen handler should work correctly', function () { return __awaiter(void 0, void 0, void 0, function () {
+        var reOpenThread, props, user;
+        return __generator(this, function (_a) {
+            switch (_a.label) {
+                case 0:
+                    reOpenThread = defaultCollaborationContextValue.reOpenThread;
+                    props = __assign(__assign({}, defaultProps), { comment: createComment('resolved') });
+                    user = setUp({ props: props }).user;
+                    return [4 /*yield*/, user.click(react_2.screen.getByText('Re-open'))];
+                case 1:
+                    _a.sent();
+                    expect(reOpenThread).toHaveBeenCalledWith({
+                        commentId: 'commentId',
+                        uri: 'entities/13JKIKAz/attributes/Name/2mlWuqDq5'
+                    });
+                    return [2 /*return*/];
+            }
         });
-        expect(updateCommentState).toHaveBeenCalledWith(defaultProps.uri, 'comment/commentId', { isEditing: false });
-    });
-    it('should call updateCommentState with right params when updateIsEditingComment called', function () {
-        var updateCommentState = defaultCollaborationContextValue.updateCommentState;
-        var getCommentState = function () { return ({ isEditing: false }); };
-        var collaborationContextValue = __assign(__assign({}, defaultCollaborationContextValue), { getCommentState: getCommentState });
-        var component = setUp({ collaborationContextValue: collaborationContextValue });
-        var menuItems = component.find(DropDownMenuButton_1.DropDownMenuButton).prop('menuItems');
-        var EditButton = menuItems.find(function (_a) {
-            var text = _a.text;
-            return text === 'Edit';
+    }); });
+    it('should call updateCommentState with right params when onCancel called', function () { return __awaiter(void 0, void 0, void 0, function () {
+        var updateCommentState, getCommentState, collaborationContextValue, user;
+        return __generator(this, function (_a) {
+            switch (_a.label) {
+                case 0:
+                    updateCommentState = defaultCollaborationContextValue.updateCommentState;
+                    getCommentState = function () { return ({ isEditing: true }); };
+                    collaborationContextValue = __assign(__assign({}, defaultCollaborationContextValue), { getCommentState: getCommentState });
+                    user = setUp({ collaborationContextValue: collaborationContextValue }).user;
+                    return [4 /*yield*/, user.click(react_2.screen.getByText('Cancel'))];
+                case 1:
+                    _a.sent();
+                    expect(updateCommentState).toHaveBeenCalledWith(defaultProps.uri, 'comment/commentId', { isEditing: false });
+                    return [2 /*return*/];
+            }
         });
-        (0, test_utils_1.act)(function () {
-            EditButton === null || EditButton === void 0 ? void 0 : EditButton.onClick();
+    }); });
+    it('should call updateCommentState with right params when updateIsEditingComment called', function () { return __awaiter(void 0, void 0, void 0, function () {
+        var updateCommentState, getCommentState, collaborationContextValue, user;
+        return __generator(this, function (_a) {
+            switch (_a.label) {
+                case 0:
+                    updateCommentState = defaultCollaborationContextValue.updateCommentState;
+                    getCommentState = function () { return ({ isEditing: false }); };
+                    collaborationContextValue = __assign(__assign({}, defaultCollaborationContextValue), { getCommentState: getCommentState });
+                    user = setUp({ collaborationContextValue: collaborationContextValue }).user;
+                    return [4 /*yield*/, user.click(react_2.screen.getByLabelText('Open menu'))];
+                case 1:
+                    _a.sent();
+                    return [4 /*yield*/, user.click(react_2.screen.getByText('Edit'))];
+                case 2:
+                    _a.sent();
+                    expect(updateCommentState).toHaveBeenCalledWith(defaultProps.uri, 'comment/commentId', { isEditing: true });
+                    return [2 /*return*/];
+            }
         });
-        expect(updateCommentState).toHaveBeenCalledWith(defaultProps.uri, 'comment/commentId', { isEditing: true });
-    });
+    }); });
     it('should render sendMessageArea insead of comment if isEditing equals true', function () {
         var getCommentState = function () { return ({ isEditing: true }); };
         var collaborationContextValue = __assign(__assign({}, defaultCollaborationContextValue), { getCommentState: getCommentState });
-        var component = setUp({ collaborationContextValue: collaborationContextValue });
-        expect(component.find(CommentContent_1.CommentContent)).toHaveLength(0);
-        var sendMessageArea = component.find(SendMessageArea_1.SendMessageArea);
-        expect(sendMessageArea).toHaveLength(1);
-        var inputRef = sendMessageArea.prop('inputRef');
-        expect(inputRef.current.value).toEqual('comment');
+        setUp({ collaborationContextValue: collaborationContextValue });
+        expect(react_2.screen.getByRole('textbox')).toHaveValue('comment');
     });
     it('should render DropDownMenuButton if comment created by current user', function () {
-        var component = setUp();
-        expect(component.find(DropDownMenuButton_1.DropDownMenuButton)).toHaveLength(1);
+        setUp();
+        expect(react_2.screen.queryByLabelText('Open menu')).toBeInTheDocument();
     });
     it('should not render DropDownMenuButton if comment created by another user', function () {
         var props = __assign(__assign({}, defaultProps), { comment: createComment('open', 'alexey.smirnov') });
-        var component = setUp({ props: props });
-        expect(component.find(DropDownMenuButton_1.DropDownMenuButton)).toHaveLength(0);
+        setUp({ props: props });
+        expect(react_2.screen.queryByLabelText('Open menu')).not.toBeInTheDocument();
     });
     it('should hide Edit button in DropDownMenuButton if comment is resolved', function () { return __awaiter(void 0, void 0, void 0, function () {
         var props, user;
@@ -286,9 +320,7 @@ describe('Comment tests', function () {
             switch (_a.label) {
                 case 0:
                     props = __assign(__assign({}, defaultProps), { comment: createComment('resolved') });
-                    user = user_event_1.default.setup();
-                    (0, react_2.render)(react_1.default.createElement(MdmModuleContext_1.MdmModuleProvider, { values: defaultMdmValues },
-                        react_1.default.createElement(Comment_1.Comment, __assign({}, props))));
+                    user = setUp({ props: props }).user;
                     return [4 /*yield*/, user.click(react_2.screen.getByLabelText('Open menu'))];
                 case 1:
                     _a.sent();