fleet/frontend/test/handlers/device-handler.ts

import { http, HttpResponse } from "msw";

import createMockDeviceUser, {
  createMockDeviceSoftwareResponse,
  createMockSetupSoftwareStatusesResponse,
} from "__mocks__/deviceUserMock";
import createMockHost from "__mocks__/hostMock";
import createMockLicense from "__mocks__/licenseMock";
import createMockMacAdmins from "__mocks__/macAdminsMock";
import { createMockHostCertificate } from "__mocks__/certificatesMock";
import { baseUrl } from "test/test-utils";
import { IDeviceUserResponse } from "interfaces/host";
import {
  IGetDeviceSoftwareResponse,
  IGetSetupSoftwareStatusesResponse,
} from "services/entities/device_user";
import { IGetHostCertificatesResponse } from "services/entities/hosts";

export const defaultDeviceHandler = http.get(baseUrl("/device/:token"), () => {
  return HttpResponse.json({
    host: createMockHost(),
    license: createMockLicense(),
    org_logo_url: "",
    global_config: {
      mdm: { enabled_and_configured: false },
    },
  });
});

export const customDeviceHandler = (overrides?: Partial<IDeviceUserResponse>) =>
  http.get(baseUrl("/device/:token"), () => {
    return HttpResponse.json(
      Object.assign(
        {
          host: createMockHost(),
          license: createMockLicense(),
          org_logo_url: "",
          global_config: {
            mdm: { enabled_and_configured: false },
          },
        },
        overrides
      )
    );
  });

export const defaultDeviceMappingHandler = http.get(
  baseUrl("/device/:token/device_mapping"),
  () => {
    return HttpResponse.json({
      device_mapping: [createMockDeviceUser()],
      host_id: 1,
    });
  }
);

export const defaultMacAdminsHandler = http.get(
  baseUrl("/device/:token/macadmins"),
  () => {
    return HttpResponse.json({
      macadmins: createMockMacAdmins(),
    });
  }
);

export const customDeviceSoftwareHandler = (
  overrides?: Partial<IGetDeviceSoftwareResponse>
) =>
  http.get(baseUrl("/device/:token/software"), () => {
    return HttpResponse.json(createMockDeviceSoftwareResponse(overrides));
  });

export const defaultDeviceCertificatesHandler = http.get(
  baseUrl("/device/:token/certificates"),
  () => {
    return HttpResponse.json<IGetHostCertificatesResponse>({
      certificates: [createMockHostCertificate()],
      meta: {
        has_next_results: false,
        has_previous_results: false,
      },
      count: 1,
    });
  }
);

export const deviceSetupExperienceHandler = (
  overrides?: Partial<IGetSetupSoftwareStatusesResponse>
) =>
  http.post(baseUrl("/device/:token/setup_experience/status"), () => {
    return HttpResponse.json(
      createMockSetupSoftwareStatusesResponse(overrides)
    );
  });

export const emptySetupExperienceHandler = deviceSetupExperienceHandler({
  setup_experience_results: { software: [] },
});