src/mlpccg/database.ts

@@ -1,10 +1,10 @@
 import Dexie from "dexie";
-import { Card, CardFilter, StoredImage } from "./types";
+import { Card, CardFilter, StoredImages } from "./types";
 import { cardFullName } from "./card";
 
 class CardDatabase extends Dexie {
   public cards: Dexie.Table<Card, string>;
-  public images: Dexie.Table<StoredImage, string>;
+  public images: Dexie.Table<StoredImages, string>;
 
   public constructor() {
     super("CardDatabase");
@@ -20,9 +20,7 @@ class CardDatabase extends Dexie {
 export let Database: CardDatabase | null = null;
 
 export function initDB() {
-  if (Database == null) {
-    Database = new CardDatabase();
-  }
+  Database = new CardDatabase();
 }
 
 export async function getCards(filter: CardFilter) {
@@ -155,9 +153,6 @@ export async function getCards(filter: CardFilter) {
 }
 
 export async function cardFromIDs(cardIDs: string[]): Promise<Card[]> {
-  if (Database == null) {
-    throw new Error("Database was not initialized, init with 'initDB()'");
-  }
   let table = Database.cards;
   //TODO Replace with .bulkGet when upgrading to Dexie 3.x
   return await table

src/mlpccg/draft/booster.ts

@@ -192,7 +192,7 @@ export class PackBuilder {
 }
 
 // Yields random cards from a chosen pool
-export function* randomProvider(pool: Card[]) {
+function* randomProvider(pool: Card[]) {
   while (true) {
     const idx = Math.floor(Math.random() * pool.length);
     yield pool[idx];
@@ -201,12 +201,12 @@ export function* randomProvider(pool: Card[]) {
 
 // Divides a list of card to a map of rarities
 // ie. [ff14, ff16, ff17] => { "C" : ["ff14"], "U": ["ff17"], "R": ["ff16"] }
-export function spanByRarity(pool: Card[]): Record<string, Card[]> {
+function spanByRarity(pool: Card[]): Record<string, Card[]> {
   return pool.reduce((map, current) => {
     if (!(current.Rarity in map)) {
       map[current.Rarity] = [];
     }
-    map[current.Rarity].push(current);
+    map[current.Rarity] = current;
     return map;
   }, Object.create(null));
 }

src/mlpccg/draft/index.ts

@@ -1,3 +0,0 @@
-export * from "./cube";
-export * from "./booster";
-export * from "./types";

src/mlpccg/types.ts

@@ -2,7 +2,7 @@ export type Rarity = "C" | "U" | "R" | "SR" | "UR" | "RR";
 
 export type PowerRequirement = { [key: string]: number };
 
-export interface StoredImage {
+export interface StoredImages {
   id: string;
   image: Blob;
 }

src/testing/IDBShim.ts

@@ -1,11 +0,0 @@
-import Dexie from "dexie";
-
-let init = false;
-
-export function setupIDBShim() {
-  if (!init) {
-    const setGlobalVars = require("indexeddbshim");
-    setGlobalVars(Dexie.dependencies);
-    init = true;
-  }
-}

src/tests/unit/database.spec.ts

@@ -1,7 +1,7 @@
 import { loadSets, getCards, Database, initDB, cardFullName } from "@/mlpccg";
-import { setupIDBShim } from "@/testing/IDBShim";
-
-setupIDBShim();
+import Dexie from "dexie";
+const setGlobalVars = require("indexeddbshim");
+setGlobalVars(Dexie.dependencies);
 
 describe("mlpccg/Database", () => {
   beforeAll(async () => {

src/tests/unit/draft.spec.ts

@@ -1,25 +0,0 @@
-import { setupIDBShim } from "@/testing/IDBShim";
-import { initDB, loadSets, Database } from "@/mlpccg";
-import { PackBuilder, spanByRarity } from "@/mlpccg/draft";
-
-setupIDBShim();
-
-describe("mlpccg/draft", () => {
-  beforeAll(async () => {
-    jest.setTimeout(15000);
-    initDB();
-    await loadSets();
-  });
-
-  test("Set booster packs are generated correctly", async () => {
-    expect(Database).toBeTruthy();
-    const builder = await PackBuilder.fromSet("FF");
-    const pack = builder.buildPack();
-    // Check pack size
-    expect(pack).toHaveLength(12);
-    const rarities = spanByRarity(pack);
-    // Check pack distribution
-    expect(rarities["R"]).toHaveLength(1);
-    expect(rarities["U"]).toHaveLength(3);
-  });
-});