update test and params

Author: paula-stacho

packages/compass-data-modeling/src/components/diagram-editor-toolbar.spec.tsx

@@ -69,7 +69,7 @@ describe('DiagramEditorToolbar', function () {
     it('renders it active if isInRelationshipDrawingMode is true', function () {
       renderDiagramEditorToolbar({ isInRelationshipDrawingMode: true });
       const addButton = screen.getByRole('button', {
-        name: 'Add Relationship',
+        name: 'Exit Relationship Drawing Mode',
       });
       expect(addButton).to.have.attribute('aria-pressed', 'true');
     });

packages/compass-data-modeling/src/components/diagram-editor.tsx

@@ -157,8 +157,7 @@ const DiagramContent: React.FunctionComponent<{
         !!selectedItems &&
         selectedItems.type === 'collection' &&
         selectedItems.id === coll.ns;
-      return collectionToDiagramNode({
-        coll,
+      return collectionToDiagramNode(coll, {
         selectedFields,
         selected,
         isInRelationshipDrawingMode,

packages/compass-data-modeling/src/utils/nodes-and-edges.tsx

@@ -142,17 +142,20 @@ export const getFieldsFromSchema = (
   return fields;
 };
 
-export function collectionToDiagramNode({
-  coll,
-  selectedFields = {},
-  selected = false,
-  isInRelationshipDrawingMode = false,
-}: {
-  coll: Pick<DataModelCollection, 'ns' | 'jsonSchema' | 'displayPosition'>;
-  selectedFields?: Record<string, string[][] | undefined>;
-  selected?: boolean;
-  isInRelationshipDrawingMode?: boolean;
-}): NodeProps {
+export function collectionToDiagramNode(
+  coll: Pick<DataModelCollection, 'ns' | 'jsonSchema' | 'displayPosition'>,
+  options: {
+    selectedFields?: Record<string, string[][] | undefined>;
+    selected?: boolean;
+    isInRelationshipDrawingMode?: boolean;
+  } = {}
+): NodeProps {
+  const {
+    selectedFields = {},
+    selected = false,
+    isInRelationshipDrawingMode = false,
+  } = options;
+
   return {
     id: coll.ns,
     type: 'collection',