/ opencodehub / tests / integration / stack-auto-update-route.test.ts
swadhinbiswas Complete stack rebase and auto-update workflow depth · 2/19/2026
fcfb528
118 lines · 3.6 KB
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
113
114
115
116
117
118
import { beforeEach, describe, expect, it, vi } from "vitest";

const {
  canWriteRepoMock,
  stackNeedsRebaseMock,
  autoUpdateStackMock,
  fakeSchema,
} = vi.hoisted(() => ({
  canWriteRepoMock: vi.fn(async () => true),
  stackNeedsRebaseMock: vi.fn(async () => ({ needsRebase: true, behindBy: 2 })),
  autoUpdateStackMock: vi.fn(async () => ({ success: true, rebased: [], failed: [], conflicts: [] })),
  fakeSchema: {
    users: { username: {} },
    repositories: { ownerId: {}, name: {}, id: {} },
    prStacks: { id: {}, repositoryId: {} },
  } as any,
}));

let mockDb: any;

vi.mock("@/db", () => ({
  getDatabase: () => mockDb,
  schema: fakeSchema,
}));

vi.mock("@/lib/permissions", () => ({
  canWriteRepo: canWriteRepoMock,
}));

vi.mock("@/lib/stack-rebase", () => ({
  stackNeedsRebase: stackNeedsRebaseMock,
  autoUpdateStack: autoUpdateStackMock,
}));

import { GET as autoUpdateGet, POST as autoUpdatePost } from "@/pages/api/repos/[owner]/[repo]/stacks/[stackId]/auto-update";

function makeDb() {
  return {
    query: {
      users: {
        findFirst: vi.fn(async () => ({ id: "owner-1" })),
      },
      repositories: {
        findFirst: vi.fn(async () => ({ id: "repo-1", ownerId: "owner-1", name: "demo" })),
      },
      prStacks: {
        findFirst: vi.fn(async () => ({ id: "stack-1", repositoryId: "repo-1" })),
      },
    },
  };
}

async function readJson(response: Response): Promise<any> {
  return response.json();
}

describe("stack auto-update route", () => {
  beforeEach(() => {
    vi.clearAllMocks();
    mockDb = makeDb();
    canWriteRepoMock.mockResolvedValue(true);
    stackNeedsRebaseMock.mockResolvedValue({ needsRebase: true, behindBy: 2 });
    autoUpdateStackMock.mockResolvedValue({ success: true, rebased: [], failed: [], conflicts: [] });
  });

  it("returns 403 when user lacks write access", async () => {
    canWriteRepoMock.mockResolvedValue(false);

    const response = await autoUpdatePost({
      params: { owner: "owner-1", repo: "demo", stackId: "stack-1" },
      locals: { user: { id: "user-1" } },
    } as any);

    expect(response.status).toBe(403);
  });

  it("returns rebase status via GET", async () => {
    const response = await autoUpdateGet({
      params: { owner: "owner-1", repo: "demo", stackId: "stack-1" },
      locals: { user: { id: "user-1" } },
    } as any);

    const body = await readJson(response);
    expect(response.status).toBe(200);
    expect(body?.data?.needsRebase).toBe(true);
    expect(body?.data?.upToDate).toBe(false);
  });

  it("returns auto-update result with pre-check status", async () => {
    const response = await autoUpdatePost({
      params: { owner: "owner-1", repo: "demo", stackId: "stack-1" },
      locals: { user: { id: "user-1" } },
    } as any);

    const body = await readJson(response);
    expect(response.status).toBe(200);
    expect(body?.data?.needsRebase).toBe(true);
    expect(body?.data?.behindBy).toBe(2);
    expect(body?.data?.performed).toBe(true);
    expect(autoUpdateStackMock).toHaveBeenCalledWith("stack-1");
  });

  it("returns a no-op result when stack is already up to date", async () => {
    stackNeedsRebaseMock.mockResolvedValueOnce({ needsRebase: false, behindBy: 0 });

    const response = await autoUpdatePost({
      params: { owner: "owner-1", repo: "demo", stackId: "stack-1" },
      locals: { user: { id: "user-1" } },
    } as any);

    const body = await readJson(response);
    expect(response.status).toBe(200);
    expect(body?.data?.performed).toBe(false);
    expect(body?.data?.message).toMatch(/already up to date/i);
    expect(autoUpdateStackMock).not.toHaveBeenCalled();
  });
});