datajunction-ui 0.0.1-rc.9 → 0.0.2

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

@@ -0,0 +1,87 @@
+import React from 'react';
+import { render, fireEvent, waitFor, screen } from '@testing-library/react';
+import DJClientContext from '../../../providers/djclient';
+import AddMaterializationPopover from '../AddMaterializationPopover';
+import { mocks } from '../../../../mocks/mockNodes';
+
+const mockDjClient = {
+  DataJunctionAPI: {
+    materialize: jest.fn(),
+    materializationInfo: jest.fn(),
+  },
+};
+
+describe('<AddMaterializationPopover />', () => {
+  it('renders correctly and handles form submission', async () => {
+    // Mock onSubmit function
+    const onSubmitMock = jest.fn();
+    mockDjClient.DataJunctionAPI.materialize.mockReturnValue({
+      status: 201,
+      json: {
+        message: 'Saved!',
+      },
+    });
+    mockDjClient.DataJunctionAPI.materializationInfo.mockReturnValue({
+      status: 200,
+      json: {
+        job_types: [
+          {
+            name: 'spark_sql',
+            label: 'Spark SQL',
+            description: 'Spark SQL materialization job',
+            allowed_node_types: ['transform', 'dimension', 'cube'],
+            job_class: 'SparkSqlMaterializationJob',
+          },
+          {
+            name: 'druid_metrics_cube',
+            label: 'Druid Cube',
+            description:
+              'Used to materialize a cube to Druid for low-latency access to a set of metrics and dimensions. While the logical cube definition is at the level of metrics and dimensions, a materialized Druid cube will reference measures and dimensions, with rollup configured on the measures where appropriate.',
+            allowed_node_types: ['cube'],
+            job_class: 'DruidCubeMaterializationJob',
+          },
+        ],
+        strategies: [
+          {
+            name: 'full',
+            label: 'Full',
+          },
+          {
+            name: 'snapshot',
+            label: 'Snapshot',
+          },
+          {
+            name: 'incremental_time',
+            label: 'Incremental Time',
+          },
+          {
+            name: 'view',
+            label: 'View',
+          },
+        ],
+      },
+    });
+
+    // Render the component
+    const { getByText } = render(
+      <DJClientContext.Provider value={mockDjClient}>
+        <AddMaterializationPopover
+          node={mocks.mockMetricNode}
+          onSubmit={onSubmitMock}
+        />
+      </DJClientContext.Provider>,
+    );
+
+    // Open the popover
+    fireEvent.click(getByText('+ Add Materialization'));
+
+    // Save the materialization
+    fireEvent.click(getByText('Save'));
+
+    // Expect setAttributes to be called
+    await waitFor(() => {
+      expect(mockDjClient.DataJunctionAPI.materialize).toHaveBeenCalled();
+      expect(getByText('Saved!')).toBeInTheDocument();
+    });
+  });
+});

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

@@ -0,0 +1,74 @@
+import React from 'react';
+import { render, screen, waitFor } from '@testing-library/react';
+import userEvent from '@testing-library/user-event';
+import DJClientContext from '../../../providers/djclient';
+import DimensionFilter from '../DimensionFilter';
+
+// Mock DJClientContext
+const mockDJClient = {
+  DataJunctionAPI: {
+    node: jest.fn(),
+    nodeData: jest.fn(),
+  },
+};
+
+const mockDimension = {
+  label: 'Dimension Label [Test]',
+  value: 'dimension_value',
+  metadata: {
+    node_name: 'test_node',
+    node_display_name: 'Test Node',
+  },
+};
+
+const getByTextStartsWith = (container, text) => {
+  return Array.from(container.querySelectorAll('*')).find(element => {
+    return element.textContent.trim().startsWith(text);
+  });
+};
+
+describe('DimensionFilter', () => {
+  it('fetches dimension data and renders correctly', async () => {
+    // Mock node response
+    const mockNodeResponse = {
+      type: 'dimension',
+      name: 'test_node',
+      columns: [
+        {
+          name: 'id',
+          attributes: [{ attribute_type: { name: 'primary_key' } }],
+        },
+        { name: 'name', attributes: [{ attribute_type: { name: 'label' } }] },
+      ],
+    };
+    mockDJClient.DataJunctionAPI.node.mockResolvedValue(mockNodeResponse);
+
+    // Mock node data response
+    const mockNodeDataResponse = {
+      results: [
+        {
+          columns: [{ semantic_entity: 'id' }, { semantic_entity: 'name' }],
+          rows: [
+            [1, 'Value 1'],
+            [2, 'Value 2'],
+          ],
+        },
+      ],
+    };
+    mockDJClient.DataJunctionAPI.nodeData.mockResolvedValue(
+      mockNodeDataResponse,
+    );
+
+    const { container } = render(
+      <DJClientContext.Provider value={mockDJClient}>
+        <DimensionFilter dimension={mockDimension} onChange={jest.fn()} />
+      </DJClientContext.Provider>,
+    );
+
+    // Check if the dimension label and node display name are rendered
+    expect(
+      getByTextStartsWith(container, 'Dimension Label'),
+    ).toBeInTheDocument();
+    expect(screen.getByText('Test Node')).toBeInTheDocument();
+  });
+});

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

@@ -0,0 +1,149 @@
+import React from 'react';
+import {
+  render,
+  fireEvent,
+  waitFor,
+  screen,
+  within,
+} from '@testing-library/react';
+import EditColumnDescriptionPopover from '../EditColumnDescriptionPopover';
+import DJClientContext from '../../../providers/djclient';
+
+const mockDjClient = {
+  DataJunctionAPI: {
+    setColumnDescription: jest.fn(),
+  },
+};
+
+describe('<EditColumnDescriptionPopover />', () => {
+  it('renders correctly and handles successful form submission', async () => {
+    // Mock necessary data
+    const column = {
+      name: 'column1',
+      description: 'Initial description',
+    };
+    const node = { name: 'default.node1' };
+
+    // Mock onSubmit function
+    const onSubmitMock = jest.fn();
+
+    mockDjClient.DataJunctionAPI.setColumnDescription.mockReturnValue({
+      status: 200,
+      json: { message: 'Description updated successfully' },
+    });
+
+    // Render the component
+    render(
+      <DJClientContext.Provider value={mockDjClient}>
+        <EditColumnDescriptionPopover
+          column={column}
+          node={node}
+          onSubmit={onSubmitMock}
+        />
+      </DJClientContext.Provider>,
+    );
+
+    // Open the popover
+    fireEvent.click(screen.getByLabelText('EditColumnDescription'));
+
+    // Update the description
+    const dialog = screen.getByRole('dialog', { name: /edit-description/i });
+    const descriptionTextarea = within(dialog).getByRole('textbox');
+    fireEvent.change(descriptionTextarea, {
+      target: { value: 'Updated description' },
+    });
+
+    // Submit the form
+    fireEvent.click(screen.getByText('Save'));
+
+    // Expect setColumnDescription to be called
+    await waitFor(() => {
+      expect(
+        mockDjClient.DataJunctionAPI.setColumnDescription,
+      ).toHaveBeenCalledWith('default.node1', 'column1', 'Updated description');
+      expect(screen.getByText('Saved!')).toBeInTheDocument();
+      expect(onSubmitMock).toHaveBeenCalled();
+    });
+  });
+
+  it('handles failed form submission', async () => {
+    // Mock necessary data
+    const column = {
+      name: 'column1',
+      description: 'Initial description',
+    };
+    const node = { name: 'default.node1' };
+
+    // Mock onSubmit function
+    const onSubmitMock = jest.fn();
+
+    mockDjClient.DataJunctionAPI.setColumnDescription.mockReturnValue({
+      status: 500,
+      json: { message: 'Server error' },
+    });
+
+    // Render the component
+    render(
+      <DJClientContext.Provider value={mockDjClient}>
+        <EditColumnDescriptionPopover
+          column={column}
+          node={node}
+          onSubmit={onSubmitMock}
+        />
+      </DJClientContext.Provider>,
+    );
+
+    // Open the popover
+    fireEvent.click(screen.getByLabelText('EditColumnDescription'));
+
+    // Update the description
+    const dialog = screen.getByRole('dialog', { name: /edit-description/i });
+    const descriptionTextarea = within(dialog).getByRole('textbox');
+    fireEvent.change(descriptionTextarea, {
+      target: { value: 'Updated description' },
+    });
+
+    // Submit the form
+    fireEvent.click(screen.getByText('Save'));
+
+    // Expect setColumnDescription to be called and the failure message to show up
+    await waitFor(() => {
+      expect(
+        mockDjClient.DataJunctionAPI.setColumnDescription,
+      ).toHaveBeenCalledWith('default.node1', 'column1', 'Updated description');
+      expect(screen.getByText('Server error')).toBeInTheDocument();
+      expect(onSubmitMock).toHaveBeenCalled();
+    });
+  });
+
+  it('renders with empty initial description', async () => {
+    // Mock necessary data with no description
+    const column = {
+      name: 'column1',
+      description: null,
+    };
+    const node = { name: 'default.node1' };
+
+    // Mock onSubmit function
+    const onSubmitMock = jest.fn();
+
+    // Render the component
+    render(
+      <DJClientContext.Provider value={mockDjClient}>
+        <EditColumnDescriptionPopover
+          column={column}
+          node={node}
+          onSubmit={onSubmitMock}
+        />
+      </DJClientContext.Provider>,
+    );
+
+    // Open the popover
+    fireEvent.click(screen.getByLabelText('EditColumnDescription'));
+
+    // Check that the textarea is empty
+    const dialog = screen.getByRole('dialog', { name: /edit-description/i });
+    const descriptionTextarea = within(dialog).getByRole('textbox');
+    expect(descriptionTextarea.value).toBe('');
+  });
+});

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,161 @@
+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: '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}
+          dimensionNodes={['default.dimension1']}
+          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 2'));
+    fireEvent.click(getByText('Save'));
+
+    // Expect linkDimension to be called
+    await waitFor(() => {
+      expect(mockDjClient.DataJunctionAPI.linkDimension).toHaveBeenCalledWith(
+        'default.node1',
+        'column1',
+        'default.dimension2',
+      );
+      expect(getByText('Saved!')).toBeInTheDocument();
+    });
+
+    // Click on the 'Remove' option and save
+    const removeButton = screen.getByLabelText('Remove default.dimension1');
+    fireEvent.click(removeButton);
+    fireEvent.click(getByText('Save'));
+
+    // 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: '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}
+          dimensionNodes={['default.dimension1']}
+          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 2'));
+    fireEvent.click(getByText('Save'));
+
+    // Expect linkDimension to be called
+    await waitFor(() => {
+      expect(mockDjClient.DataJunctionAPI.linkDimension).toHaveBeenCalledWith(
+        'default.node1',
+        'column1',
+        'default.dimension2',
+      );
+      expect(
+        getByText('Failed due to nonexistent dimension'),
+      ).toBeInTheDocument();
+    });
+
+    // Click on the 'Remove' option and save
+    const removeButton = screen.getByLabelText('Remove default.dimension1');
+    fireEvent.click(removeButton);
+    fireEvent.click(getByText('Save'));
+
+    // 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();
+    });
+  });
+});