import { module, test } from 'qunit'; import { visit } from '@ember/test-helpers'; import { setupApplicationTest } from 'ember-qunit'; import Pretender from 'pretender'; import formatRFC3339 from 'date-fns/formatRFC3339'; import { addDays, subDays } from 'date-fns'; const generateHealthResponse = (state) => { let expiry; switch (state) { case 'expired': expiry = subDays(new Date(), 2); break; case 'expiring': expiry = addDays(new Date(), 10); break; default: expiry = addDays(new Date(), 33); break; } return { initialized: true, sealed: false, standby: false, license: { expiry_time: formatRFC3339(expiry), state: 'stored', }, performance_standby: false, replication_performance_mode: 'disabled', replication_dr_mode: 'disabled', server_time_utc: 1622562585, version: '1.9.0+ent', cluster_name: 'vault-cluster-e779cd7c', cluster_id: '5f20f5ab-acea-0481-787e-71ec2ff5a60b', last_wal: 121, }; }; module('Acceptance | Enterprise | License banner warnings', function (hooks) { setupApplicationTest(hooks); hooks.afterEach(function () { this.server.shutdown(); }); test('it shows no license banner if license expires in > 30 days', async function (assert) { const healthResp = generateHealthResponse(); this.server = new Pretender(function () { this.get('/v1/sys/health', (response) => { return [response, { 'Content-Type': 'application/json' }, JSON.stringify(healthResp)]; }); this.get('/v1/sys/internal/ui/feature-flags', this.passthrough); this.get('/v1/sys/internal/ui/mounts', this.passthrough); this.get('/v1/sys/seal-status', this.passthrough); this.get('/v1/sys/license/features', this.passthrough); }); await visit('/vault/auth'); assert.dom('[data-test-license-banner]').doesNotExist('license banner does not show'); this.server.shutdown(); }); test('it shows license banner warning if license expires within 30 days', async function (assert) { const healthResp = generateHealthResponse('expiring'); this.server = new Pretender(function () { this.get('/v1/sys/health', (response) => { return [response, { 'Content-Type': 'application/json' }, JSON.stringify(healthResp)]; }); this.get('/v1/sys/internal/ui/feature-flags', this.passthrough); this.get('/v1/sys/internal/ui/mounts', this.passthrough); this.get('/v1/sys/seal-status', this.passthrough); this.get('/v1/sys/license/features', this.passthrough); }); await visit('/vault/auth'); assert.dom('[data-test-license-banner-warning]').exists('license warning shows'); this.server.shutdown(); }); test('it shows license banner alert if license has already expired', async function (assert) { const healthResp = generateHealthResponse('expired'); this.server = new Pretender(function () { this.get('/v1/sys/health', (response) => { return [response, { 'Content-Type': 'application/json' }, JSON.stringify(healthResp)]; }); this.get('/v1/sys/internal/ui/feature-flags', this.passthrough); this.get('/v1/sys/internal/ui/mounts', this.passthrough); this.get('/v1/sys/seal-status', this.passthrough); this.get('/v1/sys/license/features', this.passthrough); }); await visit('/vault/auth'); assert.dom('[data-test-license-banner-expired]').exists('expired license message shows'); this.server.shutdown(); }); });