datajunction-ui 0.0.1-a1

package/src/app/pages/NodePage/__tests__/EditColumnPopover.test.jsx

@@ -0,0 +1,148 @@
+import React from 'react';
+import { render, fireEvent, waitFor, screen } from '@testing-library/react';
+import EditColumnPopover from '../EditColumnPopover';
+import DJClientContext from '../../../providers/djclient';
+
+const mockDjClient = {
+  DataJunctionAPI: {
+    setAttributes: jest.fn(),
+  },
+};
+
+describe('<EditColumnPopover />', () => {
+  it('renders correctly and handles form submission', async () => {
+    // Mock necessary data
+    const column = {
+      name: 'column1',
+      dimension: { name: 'dimension1' },
+      attributes: [
+        { attribute_type: { name: 'primary_key', namespace: 'system' } },
+      ],
+    };
+    const node = { name: 'default.node1' };
+    const options = [
+      { value: 'dimension', label: 'Dimension' },
+      { value: 'primary_key', label: 'Primary Key' },
+    ];
+
+    // Mock onSubmit function
+    const onSubmitMock = jest.fn();
+
+    mockDjClient.DataJunctionAPI.setAttributes.mockReturnValue({
+      status: 201,
+      json: [
+        {
+          name: 'id',
+          type: 'int',
+          attributes: [
+            { attribute_type: { name: 'primary_key', namespace: 'system' } },
+          ],
+          dimension: null,
+        },
+      ],
+    });
+
+    // Render the component
+    const { getByLabelText, getByText, getByTestId } = render(
+      <DJClientContext.Provider value={mockDjClient}>
+        <EditColumnPopover
+          column={column}
+          node={node}
+          options={options}
+          onSubmit={onSubmitMock}
+        />
+      </DJClientContext.Provider>,
+    );
+
+    // Open the popover
+    fireEvent.click(getByLabelText('EditColumn'));
+
+    // Click on one attribute in the select
+    const editAttributes = getByTestId('edit-attributes');
+    fireEvent.keyDown(editAttributes.firstChild, { key: 'ArrowDown' });
+    fireEvent.click(screen.getByText('Dimension'));
+    fireEvent.click(getByText('Save'));
+    getByText('Save').click();
+
+    // Expect setAttributes to be called
+    await waitFor(() => {
+      expect(mockDjClient.DataJunctionAPI.setAttributes).toHaveBeenCalled();
+      expect(getByText('Saved!')).toBeInTheDocument();
+    });
+
+    // Click on two attributes in the select
+    fireEvent.keyDown(editAttributes.firstChild, { key: 'ArrowDown' });
+    fireEvent.click(screen.getByText('Dimension'));
+    fireEvent.click(screen.getByText('Primary Key'));
+    fireEvent.click(getByText('Save'));
+    getByText('Save').click();
+
+    // Expect setAttributes to be called
+    await waitFor(() => {
+      expect(mockDjClient.DataJunctionAPI.setAttributes).toHaveBeenCalledWith(
+        'default.node1',
+        'column1',
+        ['primary_key', 'dimension'],
+      );
+      expect(getByText('Saved!')).toBeInTheDocument();
+    });
+
+    // Close the popover
+    fireEvent.click(getByLabelText('EditColumn'));
+  });
+
+  it('handles failed form submission', async () => {
+    // Mock necessary data
+    const column = {
+      name: 'column1',
+      dimension: { name: 'dimension1' },
+      attributes: [
+        { attribute_type: { name: 'primary_key', namespace: 'system' } },
+      ],
+    };
+    const node = { name: 'default.node1' };
+    const options = [
+      { value: 'dimension', label: 'Dimension' },
+      { value: 'primary_key', label: 'Primary Key' },
+    ];
+
+    // Mock onSubmit function
+    const onSubmitMock = jest.fn();
+
+    mockDjClient.DataJunctionAPI.setAttributes.mockReturnValue({
+      status: 500,
+      json: { message: 'bad request' },
+    });
+
+    // Render the component
+    const { getByLabelText, getByText, getByTestId } = render(
+      <DJClientContext.Provider value={mockDjClient}>
+        <EditColumnPopover
+          column={column}
+          node={node}
+          options={options}
+          onSubmit={onSubmitMock}
+        />
+      </DJClientContext.Provider>,
+    );
+
+    // Open the popover
+    fireEvent.click(getByLabelText('EditColumn'));
+
+    // Click on one attribute in the select
+    const editAttributes = getByTestId('edit-attributes');
+    fireEvent.keyDown(editAttributes.firstChild, { key: 'ArrowDown' });
+    fireEvent.click(screen.getByText('Dimension'));
+    fireEvent.click(getByText('Save'));
+    getByText('Save').click();
+
+    // Expect setAttributes to be called and the failure message to show up
+    await waitFor(() => {
+      expect(mockDjClient.DataJunctionAPI.setAttributes).toHaveBeenCalled();
+      expect(getByText('bad request')).toBeInTheDocument();
+    });
+
+    // Close the popover
+    fireEvent.click(getByLabelText('EditColumn'));
+  });
+});

package/src/app/pages/NodePage/__tests__/LinkDimensionPopover.test.jsx

@@ -0,0 +1,165 @@
+import React from 'react';
+import { render, fireEvent, waitFor, screen } from '@testing-library/react';
+import LinkDimensionPopover from '../LinkDimensionPopover';
+import DJClientContext from '../../../providers/djclient';
+
+const mockDjClient = {
+  DataJunctionAPI: {
+    linkDimension: jest.fn(),
+    unlinkDimension: jest.fn(),
+  },
+};
+
+describe('<LinkDimensionPopover />', () => {
+  it('renders correctly and handles form submission', async () => {
+    // Mock necessary data
+    const column = {
+      name: 'column1',
+      dimension: { name: 'default.dimension1' },
+    };
+    const node = { name: 'default.node1' };
+    const options = [
+      { value: 'Remove', label: '[Remove dimension link]' },
+      { value: 'default.dimension1', label: 'Dimension 1' },
+      { value: 'default.dimension2', label: 'Dimension 2' },
+    ];
+
+    // Mock onSubmit function
+    const onSubmitMock = jest.fn();
+
+    mockDjClient.DataJunctionAPI.linkDimension.mockReturnValue({
+      status: 200,
+      json: { message: 'Success' },
+    });
+
+    mockDjClient.DataJunctionAPI.unlinkDimension.mockReturnValue({
+      status: 200,
+      json: { message: 'Success' },
+    });
+
+    // Render the component
+    const { getByLabelText, getByText, getByTestId } = render(
+      <DJClientContext.Provider value={mockDjClient}>
+        <LinkDimensionPopover
+          column={column}
+          node={node}
+          options={options}
+          onSubmit={onSubmitMock}
+        />
+      </DJClientContext.Provider>,
+    );
+
+    // Open the popover
+    fireEvent.click(getByLabelText('LinkDimension'));
+
+    // Click on a dimension and save
+    const linkDimension = getByTestId('link-dimension');
+    fireEvent.keyDown(linkDimension.firstChild, { key: 'ArrowDown' });
+    fireEvent.click(screen.getByText('Dimension 1'));
+    fireEvent.click(getByText('Save'));
+    getByText('Save').click();
+
+    // Expect linkDimension to be called
+    await waitFor(() => {
+      expect(mockDjClient.DataJunctionAPI.linkDimension).toHaveBeenCalledWith(
+        'default.node1',
+        'column1',
+        'default.dimension1',
+      );
+      expect(getByText('Saved!')).toBeInTheDocument();
+    });
+
+    // Click on the 'Remove' option and save
+    fireEvent.keyDown(linkDimension.firstChild, { key: 'ArrowDown' });
+    fireEvent.click(screen.getByText('[Remove dimension link]'));
+    fireEvent.click(getByText('Save'));
+    getByText('Save').click();
+
+    // Expect unlinkDimension to be called
+    await waitFor(() => {
+      expect(mockDjClient.DataJunctionAPI.unlinkDimension).toHaveBeenCalledWith(
+        'default.node1',
+        'column1',
+        'default.dimension1',
+      );
+      expect(getByText('Removed dimension link!')).toBeInTheDocument();
+    });
+  });
+
+  it('handles failed form submission', async () => {
+    // Mock necessary data
+    const column = {
+      name: 'column1',
+      dimension: { name: 'default.dimension1' },
+    };
+    const node = { name: 'default.node1' };
+    const options = [
+      { value: 'Remove', label: '[Remove dimension link]' },
+      { value: 'default.dimension1', label: 'Dimension 1' },
+      { value: 'default.dimension2', label: 'Dimension 2' },
+    ];
+
+    // Mock onSubmit function
+    const onSubmitMock = jest.fn();
+
+    mockDjClient.DataJunctionAPI.linkDimension.mockReturnValue({
+      status: 500,
+      json: { message: 'Failed due to nonexistent dimension' },
+    });
+
+    mockDjClient.DataJunctionAPI.unlinkDimension.mockReturnValue({
+      status: 500,
+      json: { message: 'Failed due to no dimension link' },
+    });
+
+    // Render the component
+    const { getByLabelText, getByText, getByTestId } = render(
+      <DJClientContext.Provider value={mockDjClient}>
+        <LinkDimensionPopover
+          column={column}
+          node={node}
+          options={options}
+          onSubmit={onSubmitMock}
+        />
+      </DJClientContext.Provider>,
+    );
+
+    // Open the popover
+    fireEvent.click(getByLabelText('LinkDimension'));
+
+    // Click on a dimension and save
+    const linkDimension = getByTestId('link-dimension');
+    fireEvent.keyDown(linkDimension.firstChild, { key: 'ArrowDown' });
+    fireEvent.click(screen.getByText('Dimension 1'));
+    fireEvent.click(getByText('Save'));
+    getByText('Save').click();
+
+    // Expect linkDimension to be called
+    await waitFor(() => {
+      expect(mockDjClient.DataJunctionAPI.linkDimension).toHaveBeenCalledWith(
+        'default.node1',
+        'column1',
+        'default.dimension1',
+      );
+      expect(
+        getByText('Failed due to nonexistent dimension'),
+      ).toBeInTheDocument();
+    });
+
+    // Click on the 'Remove' option and save
+    fireEvent.keyDown(linkDimension.firstChild, { key: 'ArrowDown' });
+    fireEvent.click(screen.getByText('[Remove dimension link]'));
+    fireEvent.click(getByText('Save'));
+    getByText('Save').click();
+
+    // Expect unlinkDimension to be called
+    await waitFor(() => {
+      expect(mockDjClient.DataJunctionAPI.unlinkDimension).toHaveBeenCalledWith(
+        'default.node1',
+        'column1',
+        'default.dimension1',
+      );
+      expect(getByText('Failed due to no dimension link')).toBeInTheDocument();
+    });
+  });
+});