datajunction-ui 0.0.1-rc.9 → 0.0.2-0.dev1

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

@@ -0,0 +1,190 @@
+import React from 'react';
+import { render, waitFor, screen } from '@testing-library/react';
+import NodeMaterializationTab from '../NodeMaterializationTab';
+
+describe('<NodeMaterializationTab />', () => {
+  const mockDjClient = {
+    node: jest.fn(),
+    materializations: jest.fn(),
+    availabilityStates: jest.fn(),
+    materializationInfo: jest.fn(),
+    refreshLatestMaterialization: jest.fn(),
+  };
+
+  const mockMaterializations = [
+    {
+      name: 'mat_one',
+      config: {
+        cube: {
+          version: 'v1.0',
+        },
+      },
+      schedule: '@daily',
+      job: 'SparkSqlMaterializationJob',
+      backfills: [
+        {
+          spec: [
+            {
+              column_name: 'date',
+              values: ['20200101'],
+              range: ['20201010'],
+            },
+          ],
+          urls: ['https://example.com/'],
+        },
+      ],
+      strategy: 'full',
+      output_tables: ['table1'],
+      urls: ['https://example.com/'],
+      deactivated_at: null,
+    },
+  ];
+
+  const mockAvailabilityStates = [
+    {
+      id: 1,
+      catalog: 'default',
+      schema_: 'foo',
+      table: 'bar',
+      valid_through_ts: 1729667463,
+      url: 'https://www.table.com',
+      links: { dashboard: 'https://www.foobar.com/dashboard' },
+      categorical_partitions: [],
+      temporal_partitions: [],
+      min_temporal_partition: ['2022', '01', '01'],
+      max_temporal_partition: ['2023', '01', '25'],
+      partitions: [],
+      updated_at: '2023-08-21T16:48:52.880498+00:00',
+      node_revision_id: 1,
+      node_version: 'v1.0',
+    },
+  ];
+
+  const mockNode = {
+    node_revision_id: 1,
+    node_id: 1,
+    type: 'source',
+    name: 'default.repair_orders',
+    display_name: 'Default: Repair Orders',
+    version: 'v1.0',
+    status: 'valid',
+    mode: 'published',
+    catalog: {
+      id: 1,
+      uuid: '0fc18295-e1a2-4c3c-b72a-894725c12488',
+      created_at: '2023-08-21T16:48:51.146121+00:00',
+      updated_at: '2023-08-21T16:48:51.146122+00:00',
+      extra_params: {},
+      name: 'warehouse',
+    },
+    schema_: 'roads',
+    table: 'repair_orders',
+    description: 'Repair orders',
+    query: null,
+    availability: {
+      catalog: 'default',
+      categorical_partitions: [],
+      max_temporal_partition: ['2023', '01', '25'],
+      min_temporal_partition: ['2022', '01', '01'],
+      partitions: [],
+      schema_: 'foo',
+      table: 'bar',
+      temporal_partitions: [],
+      valid_through_ts: 1729667463,
+      url: 'https://www.table.com',
+      links: { dashboard: 'https://www.foobar.com/dashboard' },
+    },
+    columns: [
+      {
+        name: 'repair_order_id',
+        type: 'int',
+        attributes: [],
+        dimension: null,
+        partition: {
+          type_: 'temporal',
+          format: 'YYYYMMDD',
+          granularity: 'day',
+        },
+      },
+      {
+        name: 'municipality_id',
+        type: 'string',
+        attributes: [],
+        dimension: null,
+        partition: null,
+      },
+      {
+        name: 'hard_hat_id',
+        type: 'int',
+        attributes: [],
+        dimension: null,
+        partition: null,
+      },
+    ],
+    updated_at: '2023-08-21T16:48:52.880498+00:00',
+    materializations: [
+      {
+        name: 'mat1',
+        config: {},
+        schedule: 'string',
+        job: 'string',
+        backfills: [
+          {
+            spec: [
+              {
+                column_name: 'string',
+                values: ['string'],
+                range: ['string'],
+              },
+            ],
+            urls: ['string'],
+          },
+        ],
+        strategy: 'string',
+        output_tables: ['string'],
+        urls: ['https://example.com/'],
+      },
+    ],
+    parents: [],
+    dimension_links: [
+      {
+        dimension: {
+          name: 'default.contractor',
+        },
+        join_type: 'left',
+        join_sql:
+          'default.contractor.contractor_id = default.repair_orders.contractor_id',
+        join_cardinality: 'one_to_one',
+        role: 'contractor',
+      },
+    ],
+  };
+
+  beforeEach(() => {
+    mockDjClient.materializations.mockReset();
+    mockDjClient.availabilityStates.mockReset();
+    mockDjClient.materializationInfo.mockReset();
+  });
+
+  it('renders NodeMaterializationTab tab correctly', async () => {
+    mockDjClient.materializations.mockReturnValue(mockMaterializations);
+    mockDjClient.availabilityStates.mockReturnValue(mockAvailabilityStates);
+    mockDjClient.materializationInfo.mockReturnValue({
+      job_types: [],
+      strategies: [],
+    });
+
+    render(<NodeMaterializationTab node={mockNode} djClient={mockDjClient} />);
+    await waitFor(() => {
+      // Check that the version tab is rendered
+      expect(screen.getByText('v1.0 (latest)')).toBeInTheDocument();
+
+      // Check that the materialization is rendered
+      expect(screen.getByText('Spark Sql')).toBeInTheDocument();
+
+      // Check that the dashboard link is rendered in the availability section
+      const link = screen.getByText('dashboard').closest('a');
+      expect(link).toHaveAttribute('href', `https://www.foobar.com/dashboard`);
+    });
+  });
+});