git-jira-branch/src/test/util.ts at 95e8b5e4b1c63972a37725a6779ec9f47ba22ba2 · ccntrq/git-jira-branch · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
import {Effect} from 'effect';
import type {Effect as EffectNs} from 'effect/Effect';
import {type Mock, vi} from 'vitest';

// biome-ignore lint/suspicious/noExplicitAny: any okay in tests
export interface EffectMock<T extends Array<any>, R = any, E = any>
  extends Mock<(...args: T) => Effect.Effect<R, E>> {
  mockSuccessValue: (obj: R) => this;
  mockSuccessValueOnce: (obj: R) => this;
  mockFailValue: (e: E) => this;
  mockFailValueOnce: (e: E) => this;
}

type ToEffectMock = {
  // biome-ignore lint/suspicious/noExplicitAny: Okay to use any in tests
  <T extends Array<any>, R = any, E = any>(
    fn: Mock<(...args: T) => Effect.Effect<R, E>>,
  ): EffectMock<T, R, E>;
  <
    FN extends (
      // biome-ignore lint/suspicious/noExplicitAny: Okay to use any in tests
      ...args: Array<any>
    ) => Effect.Effect<unknown, unknown, unknown>,
  >(
    fn: Mock<FN>,
  ): EffectMock<
    Parameters<FN>,
    EffectNs.Success<ReturnType<FN>>,
    EffectNs.Error<ReturnType<FN>>
  >;
};

export const toEffectMock: ToEffectMock = (
  // biome-ignore lint/suspicious/noExplicitAny: any okay in tests
  fn: Mock<any>,
  // biome-ignore lint/suspicious/noExplicitAny: any okay in tests
): EffectMock<any, any, any> => {
  const mock = Object.assign(fn, {
    // biome-ignore lint/suspicious/noExplicitAny: any okay in tests
    mockSuccessValue: (obj: any) => fn.mockReturnValue(Effect.succeed(obj)),
    // biome-ignore lint/suspicious/noExplicitAny: any okay in tests
    mockSuccessValueOnce: (obj: any) =>
      fn.mockReturnValueOnce(Effect.succeed(obj)),
    // biome-ignore lint/suspicious/noExplicitAny: any okay in tests
    mockFailValue: (e: any) => fn.mockReturnValue(Effect.fail(e)),
    // biome-ignore lint/suspicious/noExplicitAny: any okay in tests
    mockFailValueOnce: (e: any) => fn.mockReturnValueOnce(Effect.fail(e)),
    // biome-ignore lint/suspicious/noExplicitAny: any okay in tests
  }) as EffectMock<any, any, any>;
  if (!mock.getMockImplementation()) {
    mock.mockImplementation(
      // biome-ignore lint/suspicious/noExplicitAny: any okay in tests
      () => Effect.succeed(undefined) as unknown as Effect.Effect<any, any>,
    );
  }
  return mock;
};

type MKEffectMock = {
  // biome-ignore lint/suspicious/noExplicitAny: Okay to use any in tests
  <T extends Array<any>, R = any, E = any>(
    implementation?: (...args: T) => Effect.Effect<R, E>,
  ): EffectMock<T, R, E>;
  <
    FN extends (
      // biome-ignore lint/suspicious/noExplicitAny: Okay to use any in tests
      ...args: Array<any>
    ) => Effect.Effect<unknown, unknown, unknown>,
  >(
    implementation?: (
      ...args: Parameters<FN>
    ) => Effect.Effect<
      EffectNs.Success<ReturnType<FN>>,
      EffectNs.Error<ReturnType<FN>>
    >,
  ): EffectMock<
    Parameters<FN>,
    EffectNs.Success<ReturnType<FN>>,
    EffectNs.Error<ReturnType<FN>>
  >;
};

// biome-ignore lint/suspicious/noExplicitAny: Okay to use any in tests
export const effectMock: MKEffectMock = (implementation?: any) =>
  toEffectMock(
    implementation
      ? vi.fn(implementation)
      : // biome-ignore lint/suspicious/noExplicitAny: Okay to use any in tests
        vi.fn(() => Effect.succeed(undefined) as any),
  );

export const curriedEffectMock2 = <
  // biome-ignore lint/suspicious/noExplicitAny: any okay in tests
  T1 extends Array<any>,
  // biome-ignore lint/suspicious/noExplicitAny: any okay in tests
  T2 extends Array<any>,
  // biome-ignore lint/suspicious/noExplicitAny: any okay in tests
  R = any,
  // biome-ignore lint/suspicious/noExplicitAny: any okay in tests
  E = any,
>(
  implementation?: (...args: T2) => Effect.Effect<R, E>,
): Mock<(...args: T1) => EffectMock<T2, R, E>> & {
  innerMock: EffectMock<T2, R, E>;
} => {
  const mock = effectMock(implementation);
  return Object.assign(
    vi.fn<(...args: T1) => EffectMock<T2, R, E>>(() => mock),

    {innerMock: mock},
  );
};