fe-middleware/test/ImageService/ImageService.test.ts

import { describe, expect, it, mock, spyOn } from "bun:test";
import Image from "../../src/types/Image";
import ImageService from "../../src/services/ImageService";
import GelbooruApiResponse from "../../src/types/GelbooruServiceResponse";
import { BotApiResponse } from "../../src/types/BotApiResponse";
import fs from "node:fs";

describe("endpoint gets a non repeated image", () => {
    it("should return an image that is not in the response of the /images endpoint of the bot API", async () => {
        const REPEATED_URL = "https://fastly.picsum.photos/id/1/10/20.jpg?hmac=gY6PvUXFacKfYpBpTTVcNLxumpyMmoCamM-J5DOPwNc";
        const UNIQUE_URL = "https://fastly.picsum.photos/id/2/10/20.jpg?hmac=zy6lz21CuRIstr9ETx9h5AuoH50s_L2uIEct3dROpY8";

        mock.module("../../src/services/GelbooruApiService", () => {
            let alreadyCalled = false;
            return {
                default: {
                    get: (): GelbooruApiResponse => {
                        if (alreadyCalled) {
                            return { posts: [{ url: UNIQUE_URL, tags: [] }] };
                        } else {
                            alreadyCalled = true;
                            return { posts: [{ url: REPEATED_URL, tags: [] }] };
                        }
                    }
                }
            }
        });

        mock.module("../../src/services/BotApiService", () => ({
            default: {
                getAll: (): BotApiResponse => ({
                    images: [
                        { _id: "0", url: REPEATED_URL, status: "consumed", tags: ["pokemon", "computer"], "__v": 0 }
                    ]
                })
            }
        }));

        const image: Image = await ImageService.get();
        expect(image.url).not.toBe(REPEATED_URL);
    });
});

describe("imageToBase64 works as expected", () => {
    it("should encode image to base64 properly", () => {
        const imageData = fs.readFileSync('./test/ImageService/testImage.jpg');
        const imageToBase64Spy = spyOn(ImageService as any, 'imageToBase64');
        
        const actualEncoding = imageToBase64Spy.call(ImageService, imageData);
        const expectedEncoding = fs.readFileSync('./test/ImageService/testImage.b64', "base64").toString();

        expect(actualEncoding).toBe(expectedEncoding);
    })
})