zkldi_Tachi/server/src/api/import/import.test.ts

import t from "tap";
import mockApi from "../../test-utils/mock-api";
import {
    GetKTDataBuffer,
    GetKTDataJSON,
    LoadKTBlackIIDXData,
    TestingIIDXEamusementCSV26,
    TestingIIDXEamusementCSV27,
} from "../../test-utils/test-data";
import { CloseAllConnections } from "../../test-utils/close-connections";
import { RequireNeutralAuthentication } from "../../test-utils/api-common";
import { CreateFakeAuthCookie } from "../../test-utils/fake-session";
import ResetDBState from "../../test-utils/reset-db-state";
import db from "../../db/db";

// reset DB handles the post-stuff

t.test("POST /api/import/file", async (t) => {
    const cookie = await CreateFakeAuthCookie(mockApi);

    t.beforeEach(ResetDBState);

    RequireNeutralAuthentication("/api/import/file", "POST");

    t.test("file/eamusement-iidx-csv", (t) => {
        t.beforeEach(LoadKTBlackIIDXData);

        t.test("Mini HV import", async (t) => {
            const res = await mockApi
                .post("/api/import/file")
                .set("Cookie", cookie)
                .attach(
                    "scoreData",
                    GetKTDataBuffer("./eamusement-iidx-csv/small-hv-file.csv"),
                    "my_csv.csv"
                )
                .field("importType", "file/eamusement-iidx-csv")
                .field("playtype", "SP");

            t.equal(res.body.success, true, "Should be successful.");

            t.equal(res.body.body.errors.length, 0, "Mini HV Import Should have 0 failed scores.");

            t.equal(res.body.body.scoreIDs.length, 2, "Should have 2 successful scores.");

            const scoreCount = await db.scores.find({
                scoreID: { $in: res.body.body.scoreIDs },
            });

            t.equal(
                scoreCount.length,
                res.body.body.scoreIDs.length,
                "All returned scoreIDs should be inserted to the DB."
            );

            t.end();
        });

        t.test("Valid Rootage CSV import", async (t) => {
            const res = await mockApi
                .post("/api/import/file")
                .set("Cookie", cookie)
                .attach("scoreData", TestingIIDXEamusementCSV26, "my_csv.csv")
                .field("importType", "file/eamusement-iidx-csv")
                .field("playtype", "SP");

            t.equal(res.body.success, true, "Should be successful.");

            t.equal(res.body.body.errors.length, 0, "Should have 0 failed scores.");

            const scoreCount = await db.scores.find({
                scoreID: { $in: res.body.body.scoreIDs },
            });

            t.equal(
                scoreCount.length,
                res.body.body.scoreIDs.length,
                "All returned scoreIDs should be inserted to the DB."
            );

            t.end();
        });

        t.test("Valid Heroic Verse CSV import", async (t) => {
            const res = await mockApi
                .post("/api/import/file")
                .set("Cookie", cookie)
                .attach("scoreData", TestingIIDXEamusementCSV27, "my_csv.csv")
                .field("importType", "file/eamusement-iidx-csv")
                .field("playtype", "SP");

            t.equal(res.body.success, true, "Should be successful.");

            t.equal(res.body.body.errors.length, 0, "Should have 0 failed scores.");

            const scoreCount = await db.scores.find({
                scoreID: { $in: res.body.body.scoreIDs },
            });

            t.equal(
                scoreCount.length,
                res.body.body.scoreIDs.length,
                "All returned scoreIDs should be inserted to the DB."
            );

            t.end();
        });

        t.end();
    });

    t.test("file/batch-manual", (t) => {
        t.test("Empty import", async (t) => {
            const res = await mockApi
                .post("/api/import/file")
                .set("Cookie", cookie)
                .attach(
                    "scoreData",
                    GetKTDataBuffer("./batch-manual/empty-file.json"),
                    "empty-file.json"
                )
                .field("importType", "file/batch-manual");

            t.equal(res.body.success, true, "Should be successful.");

            t.equal(res.body.body.errors.length, 0, "Import Should have 0 failed scores.");

            t.equal(res.body.body.scoreIDs.length, 0, "Should have 0 successful scores.");

            const scoreCount = await db.scores.find({
                scoreID: { $in: res.body.body.scoreIDs },
            });

            t.equal(
                scoreCount.length,
                res.body.body.scoreIDs.length,
                "All returned scoreIDs should be inserted to the DB."
            );

            t.end();
        });

        t.test("Single import", async (t) => {
            const res = await mockApi
                .post("/api/import/file")
                .set("Cookie", cookie)
                .attach(
                    "scoreData",
                    GetKTDataBuffer("./batch-manual/small-file.json"),
                    "small-file.json"
                )
                .field("importType", "file/batch-manual");

            t.equal(res.body.success, true, "Should be successful.");

            t.equal(res.body.body.errors.length, 0, "Import Should have 0 failed scores.");

            t.equal(res.body.body.scoreIDs.length, 1, "Should have 1 successful score.");

            const scoreCount = await db.scores.find({
                scoreID: { $in: res.body.body.scoreIDs },
            });

            t.equal(
                scoreCount.length,
                res.body.body.scoreIDs.length,
                "All returned scoreIDs should be inserted to the DB."
            );

            t.end();
        });

        t.end();
    });

    t.test("file/mer-iidx", (t) => {
        t.beforeEach(LoadKTBlackIIDXData);

        t.test("Example Import", async (t) => {
            const res = await mockApi
                .post("/api/import/file")
                .set("Cookie", cookie)
                .attach("scoreData", GetKTDataBuffer("./mer/base.json"), "base.json")
                .field("importType", "file/mer-iidx");

            t.equal(res.body.success, true, "Should be successful");

            t.equal(res.body.body.errors.length, 0, "Import Should have 0 failed scores.");

            t.equal(res.body.body.scoreIDs.length, 3, "Should have 3 successful scores.");

            t.end();
        });

        t.test("Example Import", async (t) => {
            const res = await mockApi
                .post("/api/import/file")
                .set("Cookie", cookie)
                .attach("scoreData", GetKTDataBuffer("./mer/large.json"), "base.json")
                .field("importType", "file/mer-iidx");

            t.equal(res.body.success, true, "Should be successful");

            t.equal(res.body.body.errors.length, 0, "Import Should have 0 failed scores.");

            t.equal(res.body.body.scoreIDs.length, 627, "Should have 627 successful scores.");

            t.end();
        });

        t.end();
    });

    t.skip("file/solid-state-squad", (t) => {
        t.beforeEach(LoadKTBlackIIDXData);

        t.test("Large Import", async (t) => {
            const res = await mockApi
                .post("/api/import/file")
                .set("Cookie", cookie)
                .attach("scoreData", GetKTDataBuffer("./s3/large-example.xml"), "large.xml")
                .field("importType", "file/solid-state-squad");

            t.equal(res.body.success, true, "Should be successful");
            t.equal(res.body.body.scoreIDs.length, null, "Should parse N scores.");

            t.end();
        });

        t.end();
    });

    t.end();
});

t.teardown(CloseAllConnections);