server/classes/inbox/processor.test.ts

import { beforeEach, describe, expect, jest, mock, test } from "bun:test";
import { SignatureValidator } from "@versia/federation";
import type { Entity, Note as VersiaNote } from "@versia/federation/types";
import {
    type Instance,
    Note,
    Notification,
    Relationship,
    User,
} from "@versia/kit/db";
import type { Context } from "hono";
import { ValidationError } from "zod-validation-error";
import { config } from "~/packages/config-manager/index.ts";
import { InboxProcessor } from "./processor.ts";

// Mock dependencies
mock.module("@versia/kit/db", () => ({
    db: {
        insert: jest.fn(
            // Return something with a `.values()` method
            () => ({ values: jest.fn() }),
        ),
    },
    User: {
        resolve: jest.fn(),
        saveFromRemote: jest.fn(),
        sendFollowAccept: jest.fn(),
    },
    Instance: {
        fromUser: jest.fn(),
    },
    Note: {
        resolve: jest.fn(),
        fromVersia: jest.fn(),
        fromSql: jest.fn(),
    },
    Relationship: {
        fromOwnerAndSubject: jest.fn(),
    },
    Like: {
        fromSql: jest.fn(),
    },
    Notification: {
        fromSql: jest.fn(),
        insert: jest.fn(),
    },
}));

mock.module("@versia/federation", () => ({
    SignatureValidator: {
        fromStringKey: jest.fn(() => ({
            validate: jest.fn(),
        })),
    },
    EntityValidator: jest.fn(() => ({
        validate: jest.fn(),
    })),
    RequestParserHandler: jest.fn(),
}));

mock.module("~/packages/config-manager/index.ts", () => ({
    config: {
        debug: {
            federation: false,
        },
        federation: {
            blocked: [],
            bridge: {
                enabled: false,
                token: "test-token",
                allowed_ips: [],
            },
        },
    },
}));

describe("InboxProcessor", () => {
    let mockContext: Context;
    let mockBody: Entity;
    let mockSenderInstance: Instance;
    let mockHeaders: {
        signature: string;
        nonce: string;
        authorization?: string;
    };
    let processor: InboxProcessor;

    beforeEach(() => {
        // Reset all mocks
        mock.restore();

        // Setup basic mock context
        mockContext = {
            json: jest.fn(),
            text: jest.fn(),
            req: {
                url: "https://test.com",
                method: "POST",
                text: jest.fn().mockResolvedValue("test-body"),
            },
            env: {
                ip: { address: "127.0.0.1" },
            },
        } as unknown as Context;

        // Setup basic mock sender
        mockSenderInstance = {
            id: "test-id",
            data: {
                publicKey: {
                    key: "test-key",
                },
            },
        } as unknown as Instance;

        // Setup basic mock headers
        mockHeaders = {
            signature: "test-signature",
            nonce: "test-nonce",
        };

        // Setup basic mock body
        mockBody = {} as Entity;

        // Create processor instance
        processor = new InboxProcessor(
            mockContext,
            mockBody,
            mockSenderInstance,
            mockHeaders,
        );
    });

    describe("isSignatureValid", () => {
        test("returns true for valid signature", async () => {
            const mockValidator = {
                validate: jest.fn().mockResolvedValue(true),
            };
            SignatureValidator.fromStringKey = jest
                .fn()
                .mockResolvedValue(mockValidator);

            // biome-ignore lint/complexity/useLiteralKeys: Private method
            const result = await processor["isSignatureValid"]();
            expect(result).toBe(true);
            expect(mockValidator.validate).toHaveBeenCalled();
        });

        test("returns false for invalid signature", async () => {
            const mockValidator = {
                validate: jest.fn().mockResolvedValue(false),
            };
            SignatureValidator.fromStringKey = jest
                .fn()
                .mockResolvedValue(mockValidator);

            // biome-ignore lint/complexity/useLiteralKeys: Private method
            const result = await processor["isSignatureValid"]();
            expect(result).toBe(false);
        });
    });

    describe("shouldCheckSignature", () => {
        test("returns true when bridge is disabled", () => {
            // biome-ignore lint/complexity/useLiteralKeys: Private method
            const result = processor["shouldCheckSignature"]();
            expect(result).toBe(true);
        });

        test("returns false for valid bridge request", () => {
            config.federation.bridge.enabled = true;
            config.federation.bridge.token = "valid-token";
            config.federation.bridge.allowed_ips = ["127.0.0.1"];
            mockHeaders.authorization = "Bearer valid-token";

            // biome-ignore lint/complexity/useLiteralKeys: Private method
            const result = processor["shouldCheckSignature"]();
            expect(result).toBe(false);
        });

        test("returns error response for invalid token", () => {
            config.federation.bridge.enabled = true;
            mockHeaders.authorization = "Bearer invalid-token";

            // biome-ignore lint/complexity/useLiteralKeys: Private method
            const result = processor["shouldCheckSignature"]() as {
                code: number;
            };
            expect(result.code).toBe(401);
        });
    });

    describe("processNote", () => {
        test("successfully processes valid note", async () => {
            const mockNote = { author: "test-author" };
            const mockAuthor = { id: "test-id" };

            User.resolve = jest.fn().mockResolvedValue(mockAuthor);
            Note.fromVersia = jest.fn().mockResolvedValue(true);
            mockContext.text = jest.fn().mockReturnValue({ status: 201 });

            // biome-ignore lint/complexity/useLiteralKeys: Private variable
            processor["body"] = mockNote as VersiaNote;
            // biome-ignore lint/complexity/useLiteralKeys: Private method
            await processor["processNote"]();

            expect(User.resolve).toHaveBeenCalledWith("test-author");
            expect(Note.fromVersia).toHaveBeenCalledWith(mockNote, mockAuthor);
            expect(mockContext.text).toHaveBeenCalledWith("Note created", 201);
        });

        test("returns 404 when author not found", async () => {
            User.resolve = jest.fn().mockResolvedValue(null);
            mockContext.json = jest.fn().mockReturnValue({ status: 404 });

            // biome-ignore lint/complexity/useLiteralKeys: Private method
            await processor["processNote"]();

            expect(mockContext.json).toHaveBeenCalledWith(
                { error: "Author not found" },
                404,
            );
        });
    });

    describe("processFollowRequest", () => {
        test("successfully processes follow request for unlocked account", async () => {
            const mockFollow = {
                author: "test-author",
                followee: "test-followee",
            };
            const mockAuthor = { id: "author-id" };
            const mockFollowee = {
                id: "followee-id",
                data: { isLocked: false },
                sendFollowAccept: jest.fn(),
            };
            const mockRelationship = {
                data: { following: false },
                update: jest.fn(),
            };

            User.resolve = jest
                .fn()
                .mockResolvedValueOnce(mockAuthor)
                .mockResolvedValueOnce(mockFollowee);
            Relationship.fromOwnerAndSubject = jest
                .fn()
                .mockResolvedValue(mockRelationship);
            Notification.insert = jest.fn();
            mockContext.text = jest.fn().mockReturnValue({ status: 200 });

            // biome-ignore lint/complexity/useLiteralKeys: Private variable
            processor["body"] = mockFollow as unknown as Entity;
            // biome-ignore lint/complexity/useLiteralKeys: Private method
            await processor["processFollowRequest"]();

            expect(mockRelationship.update).toHaveBeenCalledWith({
                following: true,
                requested: false,
                showingReblogs: true,
                notifying: true,
                languages: [],
            });
        });

        test("returns 404 when author not found", async () => {
            User.resolve = jest.fn().mockResolvedValue(null);
            mockContext.json = jest.fn().mockReturnValue({ status: 404 });

            // biome-ignore lint/complexity/useLiteralKeys: Private method
            await processor["processFollowRequest"]();

            expect(mockContext.json).toHaveBeenCalledWith(
                { error: "Author not found" },
                404,
            );
        });
    });

    describe("processDelete", () => {
        test("successfully deletes a note", async () => {
            const mockDelete = {
                deleted_type: "Note",
                deleted: "test-uri",
            };
            const mockNote = {
                delete: jest.fn(),
            };

            Note.fromSql = jest.fn().mockResolvedValue(mockNote);
            mockContext.text = jest.fn().mockReturnValue({ status: 200 });

            // biome-ignore lint/complexity/useLiteralKeys: Private variable
            processor["body"] = mockDelete as unknown as Entity;
            // biome-ignore lint/complexity/useLiteralKeys: Private method
            await processor["processDelete"]();

            expect(mockNote.delete).toHaveBeenCalled();
            expect(mockContext.text).toHaveBeenCalledWith("Note deleted", 200);
        });

        test("returns 404 when note not found", async () => {
            const mockDelete = {
                deleted_type: "Note",
                deleted: "test-uri",
            };

            Note.fromSql = jest.fn().mockResolvedValue(null);
            mockContext.json = jest.fn().mockReturnValue({ status: 404 });

            // biome-ignore lint/complexity/useLiteralKeys: Private variable
            processor["body"] = mockDelete as unknown as Entity;
            // biome-ignore lint/complexity/useLiteralKeys: Private method
            await processor["processDelete"]();

            expect(mockContext.json).toHaveBeenCalledWith(
                { error: "Note to delete not found or not owned by sender" },
                404,
            );
        });
    });

    describe("processLikeRequest", () => {
        test("successfully processes like request", async () => {
            const mockLike = {
                author: "test-author",
                liked: "test-note",
                uri: "test-uri",
            };
            const mockAuthor = {
                like: jest.fn(),
            };
            const mockNote = { id: "note-id" };

            User.resolve = jest.fn().mockResolvedValue(mockAuthor);
            Note.resolve = jest.fn().mockResolvedValue(mockNote);
            mockContext.text = jest.fn().mockReturnValue({ status: 200 });

            // biome-ignore lint/complexity/useLiteralKeys: Private variable
            processor["body"] = mockLike as unknown as Entity;
            // biome-ignore lint/complexity/useLiteralKeys: Private method
            await processor["processLikeRequest"]();

            expect(mockAuthor.like).toHaveBeenCalledWith(mockNote, "test-uri");
            expect(mockContext.text).toHaveBeenCalledWith("Like created", 200);
        });

        test("returns 404 when author not found", async () => {
            User.resolve = jest.fn().mockResolvedValue(null);
            mockContext.json = jest.fn().mockReturnValue({ status: 404 });

            // biome-ignore lint/complexity/useLiteralKeys: Private method
            await processor["processLikeRequest"]();

            expect(mockContext.json).toHaveBeenCalledWith(
                { error: "Author not found" },
                404,
            );
        });
    });

    describe("processUserRequest", () => {
        test("successfully processes user update", async () => {
            const mockUser = {
                uri: "test-uri",
            };
            const mockUpdatedUser = { id: "user-id" };

            User.saveFromRemote = jest.fn().mockResolvedValue(mockUpdatedUser);
            mockContext.text = jest.fn().mockReturnValue({ status: 200 });

            // biome-ignore lint/complexity/useLiteralKeys: Private variable
            processor["body"] = mockUser as unknown as Entity;
            // biome-ignore lint/complexity/useLiteralKeys: Private method
            await processor["processUserRequest"]();

            expect(User.saveFromRemote).toHaveBeenCalledWith("test-uri");
            expect(mockContext.text).toHaveBeenCalledWith("User updated", 200);
        });

        test("returns 500 when update fails", async () => {
            User.saveFromRemote = jest.fn().mockResolvedValue(null);
            mockContext.json = jest.fn().mockReturnValue({ status: 500 });

            // biome-ignore lint/complexity/useLiteralKeys: Private method
            await processor["processUserRequest"]();

            expect(mockContext.json).toHaveBeenCalledWith(
                { error: "Failed to update user" },
                500,
            );
        });
    });

    describe("handleError", () => {
        test("handles validation errors", () => {
            const validationError = new ValidationError("Invalid data");
            mockContext.json = jest.fn().mockReturnValue({ status: 400 });

            // biome-ignore lint/complexity/useLiteralKeys: Private method
            processor["handleError"](validationError);

            expect(mockContext.json).toHaveBeenCalledWith(
                {
                    error: "Failed to process request",
                    error_description: "Invalid data",
                },
                400,
            );
        });

        test("handles general errors", () => {
            const error = new Error("Something went wrong");
            mockContext.json = jest.fn().mockReturnValue({ status: 500 });

            // biome-ignore lint/complexity/useLiteralKeys: Private method
            processor["handleError"](error);

            expect(mockContext.json).toHaveBeenCalledWith(
                {
                    error: "Failed to process request",
                    message: "Something went wrong",
                },
                500,
            );
        });
    });
});