138 lines
4.3 KiB
TypeScript
138 lines
4.3 KiB
TypeScript
import { describe, it, expect, vi, beforeEach } from "vitest"
|
|
import { db } from "@/lib/db/index"
|
|
import { checkAndAutoPin } from "@/lib/api/auto-pin"
|
|
|
|
// Mock drizzle-orm to avoid ESM named-export resolution errors between test files
|
|
vi.mock("drizzle-orm", () => ({
|
|
eq: vi.fn((col, val) => ({ col, val })),
|
|
}))
|
|
vi.mock("drizzle-orm/pg-core", () => ({
|
|
pgTable: vi.fn((name, columns, indexes) => ({ name, columns, indexes })),
|
|
pgEnum: vi.fn((name, values) => ({ name, values })),
|
|
text: vi.fn((name) => name),
|
|
integer: vi.fn((name) => name),
|
|
real: vi.fn((name) => name),
|
|
boolean: vi.fn((name) => name),
|
|
timestamp: vi.fn((name) => name),
|
|
jsonb: vi.fn((name) => name),
|
|
index: vi.fn((name) => ({ on: vi.fn() })),
|
|
}))
|
|
|
|
// Mock db schema — only the symbols the module under test references
|
|
vi.mock("@/lib/db/schema", () => ({
|
|
performanceEntries: {
|
|
id: "id",
|
|
isPinned: "is_pinned",
|
|
isRemoved: "is_removed",
|
|
upvotes: "upvotes",
|
|
downvotes: "downvotes",
|
|
},
|
|
}))
|
|
|
|
// Mock the database
|
|
vi.mock("@/lib/db/index", () => ({
|
|
db: {
|
|
select: vi.fn(),
|
|
update: vi.fn(),
|
|
},
|
|
}))
|
|
|
|
function setupMockEntry(overrides: {
|
|
id?: string
|
|
isPinned?: boolean
|
|
isRemoved?: boolean
|
|
upvotes?: number
|
|
downvotes?: number
|
|
}) {
|
|
const entry = {
|
|
id: overrides.id ?? "test-entry-1",
|
|
isPinned: overrides.isPinned ?? false,
|
|
isRemoved: overrides.isRemoved ?? false,
|
|
upvotes: overrides.upvotes ?? 0,
|
|
downvotes: overrides.downvotes ?? 0,
|
|
}
|
|
|
|
// Mock select chain: .select(...).from(...).where(...).limit(...)
|
|
const limitMock = vi.fn().mockResolvedValue([entry])
|
|
const whereMock = vi.fn().mockReturnValue({ limit: limitMock })
|
|
const fromMock = vi.fn().mockReturnValue({ where: whereMock })
|
|
const selectMock = vi.fn().mockReturnValue({ from: fromMock })
|
|
;(db.select as any).mockReturnValue({ from: fromMock })
|
|
|
|
// Mock update chain: .update(...).set(...).where(...)
|
|
const updateWhereMock = vi.fn().mockResolvedValue(undefined)
|
|
const updateSetMock = vi.fn().mockReturnValue({ where: updateWhereMock })
|
|
const updateMock = vi.fn().mockReturnValue({ set: updateSetMock })
|
|
;(db.update as any).mockImplementation(updateMock)
|
|
|
|
return entry
|
|
}
|
|
|
|
describe("checkAndAutoPin", () => {
|
|
beforeEach(() => {
|
|
vi.clearAllMocks()
|
|
})
|
|
|
|
it("pins an entry with 8 upvotes and 2 downvotes (80% approval, 10 total)", async () => {
|
|
setupMockEntry({ upvotes: 8, downvotes: 2 })
|
|
|
|
const result = await checkAndAutoPin("test-entry-1")
|
|
expect(result).toBe(true)
|
|
})
|
|
|
|
it("does NOT pin an entry with 7 upvotes and 1 downvote (< 8 absolute upvotes)", async () => {
|
|
setupMockEntry({ upvotes: 7, downvotes: 1 })
|
|
|
|
const result = await checkAndAutoPin("test-entry-1")
|
|
expect(result).toBe(false)
|
|
})
|
|
|
|
it("does NOT pin an entry with 8 upvotes and 0 downvotes (< 10 total votes)", async () => {
|
|
setupMockEntry({ upvotes: 8, downvotes: 0 })
|
|
|
|
const result = await checkAndAutoPin("test-entry-1")
|
|
expect(result).toBe(false)
|
|
})
|
|
|
|
it("does NOT pin an entry with 6 upvotes and 4 downvotes (60% approval, below 80%)", async () => {
|
|
setupMockEntry({ upvotes: 6, downvotes: 4 })
|
|
|
|
const result = await checkAndAutoPin("test-entry-1")
|
|
expect(result).toBe(false)
|
|
})
|
|
|
|
it("does NOT pin an already-pinned entry", async () => {
|
|
setupMockEntry({ isPinned: true, upvotes: 20, downvotes: 0 })
|
|
|
|
const result = await checkAndAutoPin("test-entry-1")
|
|
expect(result).toBe(false)
|
|
})
|
|
|
|
it("does NOT pin a removed entry", async () => {
|
|
setupMockEntry({ isRemoved: true, upvotes: 20, downvotes: 0 })
|
|
|
|
const result = await checkAndAutoPin("test-entry-1")
|
|
expect(result).toBe(false)
|
|
})
|
|
|
|
it("pins an entry with 20 upvotes and 5 downvotes (80% approval exactly)", async () => {
|
|
setupMockEntry({ upvotes: 20, downvotes: 5 })
|
|
|
|
const result = await checkAndAutoPin("test-entry-1")
|
|
expect(result).toBe(true)
|
|
})
|
|
|
|
it("does NOT pin when entry is not found", async () => {
|
|
// Mock select returning empty array
|
|
const limitMock = vi.fn().mockResolvedValue([])
|
|
const whereMock = vi.fn().mockReturnValue({ limit: limitMock })
|
|
const fromMock = vi.fn().mockReturnValue({ where: whereMock })
|
|
const selectMock = vi.fn().mockReturnValue({ from: fromMock })
|
|
;(db.select as any).mockReturnValue({ from: fromMock })
|
|
;(db.update as any) = vi.fn()
|
|
|
|
const result = await checkAndAutoPin("nonexistent")
|
|
expect(result).toBe(false)
|
|
})
|
|
})
|