add tests

Author: paula-stacho

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

@@ -6,7 +6,7 @@ import {
   render,
   userEvent,
 } from '@mongodb-js/testing-library-compass';
-import { getFieldsFromSchema } from '../utils/nodes-and-edges';
+import { getFieldsFromSchema } from './nodes-and-edges';
 
 describe('getFieldsFromSchema', function () {
   const validateMixedType = async (

packages/compass-data-modeling/src/utils/schema-traversal.tsx

@@ -21,6 +21,7 @@ export const traverseSchema = ({
   }: {
     fieldPath: FieldPath;
     fieldTypes: string[];
+    fieldSchema: MongoDBJSONSchema;
   }) => void;
   parentFieldPath?: FieldPath;
 }): void => {
@@ -61,6 +62,7 @@ export const traverseSchema = ({
     visitor({
       fieldPath: newFieldPath,
       fieldTypes: types.flat(),
+      fieldSchema: field,
     });
 
     children.flat().forEach((child) =>