Reorganize tests in prep for adding more

Author: DTCurrie

src/__fixtures__/credentials.ts

@@ -0,0 +1,24 @@
+import type { Credential, AccessToken } from '../main';
+
+export const testCredential: Credential = {
+  authEntity: 'test-entity',
+  type: 'api-key',
+  payload: 'test-payload',
+};
+
+export const differentCredential: Credential = {
+  authEntity: 'different-entity',
+  type: 'api-key',
+  payload: 'different-payload',
+};
+
+export const testAccessToken: AccessToken = {
+  type: 'access-token',
+  payload: 'test-access-token',
+};
+
+export const differentAccessToken: AccessToken = {
+  type: 'access-token',
+  payload: 'different-access-token',
+};
+

src/__fixtures__/test-constants.ts

@@ -0,0 +1,5 @@
+export const TEST_URL = 'http://base.test';
+export const TEST_HOST = 'test-host';
+export const TEST_AUTH_ENTITY = 'test-entity';
+export const TEST_SIGNALING_ADDRESS = 'https://signaling.test';
+

src/__mocks__/transports.ts

@@ -0,0 +1,9 @@
+import type { Transport } from '@connectrpc/connect';
+import { vi } from 'vitest';
+
+export const createMockTransport = (): Transport => {
+  return {
+    unary: vi.fn(),
+    stream: vi.fn(),
+  } satisfies Transport;
+};

src/__mocks__/webrtc.ts

@@ -0,0 +1,29 @@
+import { vi } from 'vitest';
+
+export const createMockPeerConnection = (
+  closeFn = vi.fn(),
+  addEventListenerFn = vi.fn(),
+  removeEventListenerFn = vi.fn(),
+  iceConnectionState: RTCIceConnectionState = 'connected'
+): RTCPeerConnection => {
+  return {
+    close: closeFn,
+    addEventListener: addEventListenerFn,
+    removeEventListener: removeEventListenerFn,
+    iceConnectionState,
+  } as unknown as RTCPeerConnection;
+};
+
+export const createMockDataChannel = (
+  closeFn = vi.fn(),
+  addEventListenerFn = vi.fn(),
+  removeEventListenerFn = vi.fn(),
+  readyState: RTCDataChannelState = 'open'
+): RTCDataChannel => {
+  return {
+    close: closeFn,
+    addEventListener: addEventListenerFn,
+    removeEventListener: removeEventListenerFn,
+    readyState,
+  } as unknown as RTCDataChannel;
+};

src/robot/__fixtures__/dial-configs.ts

@@ -0,0 +1,20 @@
+import {
+  TEST_HOST,
+  TEST_SIGNALING_ADDRESS,
+} from '../../__fixtures__/test-constants';
+import type { DialConf } from '../../main';
+
+export const baseDialConfig: DialConf = {
+  host: TEST_HOST,
+  signalingAddress: TEST_SIGNALING_ADDRESS,
+} as const;
+
+export const withDisableSessions: DialConf = {
+  ...baseDialConfig,
+  disableSessions: true,
+} as const;
+
+export const withNoReconnect: DialConf = {
+  ...baseDialConfig,
+  noReconnect: true,
+} as const;

src/robot/__mocks__/robot-service.ts

@@ -0,0 +1,17 @@
+import { createRouterTransport, type Transport } from '@connectrpc/connect';
+import { RobotService } from '../../gen/robot/v1/robot_connect';
+import type { PartialMessage } from '@bufbuild/protobuf';
+import type { Operation } from '../../gen/robot/v1/robot_pb';
+import type { ResourceName } from '../../gen/common/v1/common_pb';
+
+export const createMockRobotServiceTransport = (
+  resources: PartialMessage<ResourceName>[] = [],
+  operations: PartialMessage<Operation>[] = []
+): Transport => {
+  return createRouterTransport(({ service }) => {
+    service(RobotService, {
+      resourceNames: () => ({ resources }),
+      getOperations: () => ({ operations }),
+    });
+  });
+};