gitpod/components/gitpod-db/src/user-storage-resources-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 { Repository } from 'typeorm';
import { UserStorageResourcesDB } from './user-storage-resources-db';
import { DBUserStorageResource } from './typeorm/entity/db-user-storage-resource';


@suite
class UserStorageResourcesDBSpec {

    typeORM = testContainer.get<TypeORM>(TypeORM);
    resourcesDb = testContainer.get<UserStorageResourcesDB>(UserStorageResourcesDB);

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

    async after() {
        const repo = await this.getRepo();
        await repo.createQueryBuilder("resource").delete().execute();
    }

    @test(timeout(10000))
    public async testGetEmpty() {
        const content = await this.resourcesDb.get('user1', 'some://uri');
        expect(content).to.be.eq('');
    }

    @test(timeout(10000))
    public async testUpdate() {
        await this.resourcesDb.update('user1', 'some://uri', 'content');
        const content = await this.resourcesDb.get('user1', 'some://uri');
        expect(content).to.be.eq('content');
    }
}

module.exports = new UserStorageResourcesDBSpec()