import { CogniteWellsClient } from "src";
import { exampleRigOperationIngestion } from "../../fixtures/rigoperationIngestionFixtures";
import { exampleWellboreSourceItems } from "../../fixtures/wellboreSourceFixtures";
import { exampleWellSourceItems } from "../../fixtures/wellSourceFixtures";
import { setupLoggedInIngestionClient } from "../../testUtils";

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

test("ingest rig operation", async () => {
  expect(client).not.toBe(undefined);
  await client.wells.ingest(exampleWellSourceItems);
  await client.wellbores.ingest(exampleWellboreSourceItems);
  const sources = await client.rigOperations.ingest([
    exampleRigOperationIngestion,
  ]);

  const actual = sources.items[0];
  const expected = exampleRigOperationIngestion;
  expect(actual.source.externalId).toBe(expected.source.externalId);
  expect(actual.source.type).toBe(expected.source.type);
  expect(actual.rigName).toBe(expected.rigName);
  expect(actual.startTime).toBe(expected.startTime);
  expect(actual.endTime).toBe(expected.endTime);
});