gitpod/components/gitpod-db/src/user-message-views-db.spec.db.ts

/**
 * Copyright (c) 2020 Gitpod GmbH. All rights reserved.
 * Licensed under the GNU Affero General Public License (AGPL).
 * See License-AGPL.txt in the project root for license information.
 */


import * as chai from 'chai';
const expect = chai.expect;
import { suite, test, timeout } from 'mocha-typescript';

import { testContainer } from './test-container';
import { TypeORM } from './typeorm/typeorm';
import { DBUserMessageViewEntry } from './typeorm/entity/db-user-message-view-entry';
import { Repository } from 'typeorm';
import { UserMessageViewsDB } from './user-message-views-db';


@suite
class UserMessageViewsDBSpec {

    typeORM = testContainer.get<TypeORM>(TypeORM);
    viewsdb = testContainer.get<UserMessageViewsDB>(UserMessageViewsDB);

    protected async getUserMessageViewsRepo(): Promise<Repository<DBUserMessageViewEntry>> {
        return (await (await this.typeORM.getConnection()).manager).getRepository(DBUserMessageViewEntry);
    }

    async before() {
        await this.wipeRepo();
    }

    async after() {
        await this.wipeRepo();
    }

    async wipeRepo() {
        const repo = await this.getUserMessageViewsRepo();
        await repo.createQueryBuilder("view").delete().execute();
    }

    @test(timeout(10000))
    public async testSimple11() {
        const viewed = await this.viewsdb.didViewMessage('user1', 'message1');
        expect(viewed).to.be.false;
    }

    @test(timeout(10000))
    public async testSimple2() {
        await this.viewsdb.markAsViewed('user1', ['message1']);
        const viewed = await this.viewsdb.didViewMessage('user1', 'message1');
        expect(viewed).to.be.true;
    }
}

module.exports = new UserMessageViewsDBSpec()