storj/web/satellite/tests/unit/store/usage.spec.ts

// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.

import Vuex from 'vuex';

import { ProjectsApiGql } from '@/api/projects';
import { ProjectUsageApiGql } from '@/api/usage';
import { makeProjectsModule } from '@/store/modules/projects';
import { makeUsageModule, PROJECT_USAGE_ACTIONS, PROJECT_USAGE_MUTATIONS } from '@/store/modules/usage';
import { Project } from '@/types/projects';
import { DateRange, ProjectUsage } from '@/types/usage';
import { createLocalVue } from '@vue/test-utils';

const Vue = createLocalVue();
const projectUsageApi = new ProjectUsageApiGql();
const usageModule = makeUsageModule(projectUsageApi);

const projectsApi = new ProjectsApiGql();
const projectsModule = makeProjectsModule(projectsApi);
const selectedProject = new Project('', '', '', '');
selectedProject.id = '1';
projectsModule.state.selectedProject = selectedProject;

const now = new Date();
const testDate1 = new Date(Date.UTC(now.getUTCFullYear(), now.getUTCMonth(), now.getUTCDate()));
const testDate2 = new Date(Date.UTC(now.getUTCFullYear(), now.getUTCMonth(), now.getUTCDate(), 23,  59));
const testUsage = new ProjectUsage(2, 3, 4, testDate1, testDate2);

Vue.use(Vuex);

const store = new Vuex.Store({ modules: { usageModule, projectsModule } });

const state = (store.state as any).usageModule;

describe('mutations', () => {
    beforeEach(() => {
        createLocalVue().use(Vuex);
    });

    it('fetch project usage', () => {
        store.commit(PROJECT_USAGE_MUTATIONS.SET_PROJECT_USAGE, testUsage);

        expect(state.projectUsage.storage.bytes).toBe(2);
        expect(state.projectUsage.egress.bytes).toBe(3);
        expect(state.projectUsage.storage.formattedBytes).toBe('0.0020');
        expect(state.projectUsage.egress.formattedBytes).toBe('0.0030');
        expect(state.projectUsage.storage.label).toBe('KB');
        expect(state.projectUsage.egress.label).toBe('KB');
        expect(state.projectUsage.objectCount).toBe(4);
    });

    it('set dates', () => {
        const dateRange: DateRange = new DateRange(testDate1, testDate2);
        store.commit(PROJECT_USAGE_MUTATIONS.SET_DATE, dateRange);

        expect(state.startDate.toDateString()).toBe(testDate1.toDateString());
        expect(state.endDate.toDateString()).toBe(testDate2.toDateString());
    });

    it('clear usage', () => {
        store.commit(PROJECT_USAGE_MUTATIONS.CLEAR);

        expect(state.projectUsage.storage.bytes).toBe(0);
        expect(state.projectUsage.egress.bytes).toBe(0);
        expect(state.projectUsage.storage.formattedBytes).toBe('0.0000');
        expect(state.projectUsage.egress.formattedBytes).toBe('0.0000');
        expect(state.projectUsage.storage.label).toBe('Bytes');
        expect(state.projectUsage.egress.label).toBe('Bytes');
        expect(state.projectUsage.objectCount).toBe(0);
    });
});

describe('actions', () => {
    beforeEach(() => {
        jest.resetAllMocks();
        createLocalVue().use(Vuex);
    });

    it('success fetch project usage', async () => {
        jest.spyOn(projectUsageApi, 'get').mockReturnValue(
            Promise.resolve(testUsage),
        );
        const startUTC = new Date(Date.UTC(1999, 1, 1, 20, 15));
        const dateRange: DateRange = new DateRange(startUTC, testDate1);

        await store.dispatch(PROJECT_USAGE_ACTIONS.FETCH, dateRange);

        expect(state.projectUsage.storage.bytes).toBe(2);
        expect(state.projectUsage.egress.bytes).toBe(3);
        expect(state.projectUsage.storage.formattedBytes).toBe('0.0020');
        expect(state.projectUsage.egress.formattedBytes).toBe('0.0030');
        expect(state.projectUsage.storage.label).toBe('KB');
        expect(state.projectUsage.egress.label).toBe('KB');
        expect(state.projectUsage.objectCount).toBe(4);
        expect(state.startDate.toDateString()).toBe(startUTC.toDateString());
        expect(state.endDate.getUTCFullYear()).toBe(testDate1.getUTCFullYear());
        expect(state.endDate.getUTCMonth()).toBe(testDate1.getUTCMonth());
        expect(state.endDate.getUTCDate()).toBe(testDate1.getUTCDate());
    });

    it('success fetch current project usage', async () => {
        jest.spyOn(projectUsageApi, 'get').mockReturnValue(
            Promise.resolve(testUsage),
        );

        await store.dispatch(PROJECT_USAGE_ACTIONS.FETCH_CURRENT_ROLLUP);

        expect(state.projectUsage.storage.bytes).toBe(2);
        expect(state.projectUsage.egress.bytes).toBe(3);
        expect(state.projectUsage.storage.formattedBytes).toBe('0.0020');
        expect(state.projectUsage.egress.formattedBytes).toBe('0.0030');
        expect(state.projectUsage.storage.label).toBe('KB');
        expect(state.projectUsage.egress.label).toBe('KB');
        expect(state.projectUsage.objectCount).toBe(4);
    });

    it('success fetch previous project usage', async () => {
        jest.spyOn(projectUsageApi, 'get').mockReturnValue(
            Promise.resolve(testUsage),
        );

        await store.dispatch(PROJECT_USAGE_ACTIONS.FETCH_PREVIOUS_ROLLUP);

        expect(state.projectUsage.storage.bytes).toBe(2);
        expect(state.projectUsage.egress.bytes).toBe(3);
        expect(state.projectUsage.storage.formattedBytes).toBe('0.0020');
        expect(state.projectUsage.egress.formattedBytes).toBe('0.0030');
        expect(state.projectUsage.storage.label).toBe('KB');
        expect(state.projectUsage.egress.label).toBe('KB');
        expect(state.projectUsage.objectCount).toBe(4);
    });

    it('success clear usage', async () => {
        await store.dispatch(PROJECT_USAGE_ACTIONS.CLEAR);

        expect(state.projectUsage.storage.bytes).toBe(0);
        expect(state.projectUsage.egress.bytes).toBe(0);
        expect(state.projectUsage.storage.formattedBytes).toBe('0.0000');
        expect(state.projectUsage.egress.formattedBytes).toBe('0.0000');
        expect(state.projectUsage.storage.label).toBe('Bytes');
        expect(state.projectUsage.egress.label).toBe('Bytes');
        expect(state.projectUsage.objectCount).toBe(0);
    });

    it('create throws an error when create api call fails', async () => {
        state.projects = [];
        jest.spyOn(projectUsageApi, 'get').mockImplementation(() => { throw new Error(); });

        try {
            await store.dispatch(PROJECT_USAGE_ACTIONS.FETCH_PREVIOUS_ROLLUP);
            expect(true).toBe(false);
        } catch (error) {
            expect(state.projectUsage.storage.bytes).toBe(0);
            expect(state.projectUsage.egress.bytes).toBe(0);
            expect(state.projectUsage.storage.formattedBytes).toBe('0.0000');
            expect(state.projectUsage.egress.formattedBytes).toBe('0.0000');
            expect(state.projectUsage.storage.label).toBe('Bytes');
            expect(state.projectUsage.egress.label).toBe('Bytes');
            expect(state.projectUsage.objectCount).toBe(0);
        }
    });
});