import {

loadFirestoreRules,

initializeTestApp,

clearFirestoreData,

initializeAdminApp,

} from '@firebase/rules-unit-testing';

import firebase from 'firebase/app';

import { readFileSync } from 'fs';

const serviceAccountKeyFile = readFileSync('../serviceAccountKey.json', 'utf8');

const serviceAccountKey = JSON.parse(serviceAccountKeyFile);

export async function setup(auth: any, data: any) {

const projectId = serviceAccountKey.project_id;

const app = initializeTestApp({

projectId,

auth,

});

const db = app.firestore();

// Write mock documents before rules

if (data) {

const admin = initializeAdminApp({

projectId,

});

for (const key in data) {

const ref = admin.firestore().doc(key);

await ref.set(data[key]);

}

}

// Apply rules

await loadFirestoreRules({

projectId,

rules: readFileSync('../firestore.rules', 'utf8'),

});

return db;

}

export async function teardown() {

Promise.all(firebase.apps.map((app) => app.delete()));

await clearFirestoreData({ projectId: serviceAccountKey.project_id });

}