185 lines
8.2 KiB
TypeScript
185 lines
8.2 KiB
TypeScript
// 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);
|
|
expect(state.startDate.toDateString()).toBe(now.toDateString());
|
|
expect(state.endDate.toDateString()).toBe(now.toDateString());
|
|
});
|
|
|
|
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);
|
|
expect(state.startDate.toDateString()).toBe(now.toDateString());
|
|
expect(state.endDate.toDateString()).toBe(now.toDateString());
|
|
});
|
|
});
|
|
|
|
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),
|
|
);
|
|
|
|
const firstDate = new Date(Date.UTC(now.getUTCFullYear(), now.getUTCMonth(), 1));
|
|
|
|
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);
|
|
expect(state.startDate.toDateString()).toBe(firstDate.toDateString());
|
|
|
|
expect(state.endDate.getUTCFullYear()).toBe(now.getUTCFullYear());
|
|
expect(state.endDate.getUTCMonth()).toBe(now.getUTCMonth());
|
|
expect(state.endDate.getUTCDate()).toBe(now.getUTCDate());
|
|
expect(state.endDate.getUTCHours()).toBe(now.getUTCHours());
|
|
expect(state.endDate.getUTCMinutes()).toBe(now.getUTCMinutes());
|
|
});
|
|
|
|
it('success fetch previous project usage', async () => {
|
|
jest.spyOn(projectUsageApi, 'get').mockReturnValue(
|
|
Promise.resolve(testUsage),
|
|
);
|
|
|
|
const firstDate = new Date(Date.UTC(now.getUTCFullYear(), now.getUTCMonth() - 1, 1));
|
|
const secondDate = new Date(Date.UTC(now.getUTCFullYear(), now.getUTCMonth(), 0, 23, 59, 59));
|
|
|
|
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);
|
|
expect(state.startDate.toDateString()).toBe(firstDate.toDateString());
|
|
expect(state.endDate.toDateString()).toBe(secondDate.toDateString());
|
|
});
|
|
|
|
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);
|
|
expect(state.startDate.toDateString()).toBe(now.toDateString());
|
|
expect(state.endDate.toDateString()).toBe(now.toDateString());
|
|
});
|
|
|
|
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);
|
|
expect(state.startDate.toDateString()).toBe(now.toDateString());
|
|
expect(state.endDate.toDateString()).toBe(now.toDateString());
|
|
}
|
|
});
|
|
});
|