import { CogniteWellsClient } from "src";
import { setupLoggedInClient } from "../testUtils";

let client: CogniteWellsClient;
beforeAll(async () => {
  client = await setupLoggedInClient();
});

test("get list of rig operations", async () => {
  expect(client).not.toBe(undefined);
  const rigOperations = await client.rigOperations.list({});

  const actual = rigOperations.items.map((rigOperation) => [
    rigOperation.source.externalId,
    rigOperation.source.type,
    rigOperation.wellboreMatchingId,
  ]);
  expect(actual).toEqual([
    ["in_well1-A", "asset", "15/9-F-15 A"],
    ["in_well1-A_2", "sequence", "15/9-F-15 A"],
    ["in_well1-D", "asset", "15/9-F-15 D"],
    ["in_well2", "sequence", "15/9-F-4"],
    ["in_well3", "asset", "13/10-F-11 T2"],
    ["in_well3", "sequence", "13/10-F-11 T2"],
    ["in_well3", "relationship", "13/10-F-11 T2"],
    ["in_well_3_2", "asset", "13/10-F-11 T2"],
    ["in_well_3_3", "relationship", "13/10-F-11 T2"],
  ]);
});

test("filter rig operations by startTime", async () => {
  expect(client).not.toBe(undefined);
  const rigOperations = await client.rigOperations.list({
    filter: { startTime: { min: 1609603200000, max: 1609761600000 } },
  });

  const actual = rigOperations.items.map((rigOperation) => [
    rigOperation.source.externalId,
    rigOperation.source.type,
    rigOperation.wellboreMatchingId,
  ]);
  expect(actual).toEqual([
    ["in_well1-A_2", "sequence", "15/9-F-15 A"],
    ["in_well1-D", "asset", "15/9-F-15 D"],
    ["in_well2", "sequence", "15/9-F-4"],
    ["in_well3", "asset", "13/10-F-11 T2"],
    ["in_well3", "sequence", "13/10-F-11 T2"],
    ["in_well_3_2", "asset", "13/10-F-11 T2"],
  ]);
});

test("filter rig operations by rigNames", async () => {
  expect(client).not.toBe(undefined);
  const rigOperations = await client.rigOperations.list({
    filter: { rigNames: ["rig_4"] },
  });

  const actual = rigOperations.items.map((rigOperation) => [
    rigOperation.source.externalId,
    rigOperation.source.type,
    rigOperation.wellboreMatchingId,
  ]);
  expect(actual).toEqual([
    ["in_well_3_2", "asset", "13/10-F-11 T2"],
    ["in_well_3_3", "relationship", "13/10-F-11 T2"],
  ]);
});

test("list rig operations with pagination", async () => {
  expect(client).not.toBe(undefined);
  const firstGet = await client.rigOperations.list({ limit: 2 });
  expect(firstGet.items.length).toBe(2);
  expect(firstGet.nextCursor).not.toBeUndefined();

  const secondGet = await client.rigOperations.list({
    limit: 3,
    cursor: firstGet.nextCursor,
  });
  expect(secondGet.items.length).toBe(3);

  const actual = firstGet.items
    .concat(secondGet.items)
    .map((rigOperation) => [
      rigOperation.source.externalId,
      rigOperation.source.type,
      rigOperation.wellboreMatchingId,
    ]);

  expect(actual).toEqual([
    ["in_well1-A", "asset", "15/9-F-15 A"],
    ["in_well1-A_2", "sequence", "15/9-F-15 A"],
    ["in_well1-D", "asset", "15/9-F-15 D"],
    ["in_well2", "sequence", "15/9-F-4"],
    ["in_well3", "asset", "13/10-F-11 T2"],
  ]);
});