@hashgraph/solo/dist/test/e2e/integration/core/lease.test.js

An opinionated CLI tool to deploy and manage private Hedera Networks.

// SPDX-License-Identifier: Apache-2.0
import { it, describe, before, after } from 'mocha';
import { expect } from 'chai';
import { IntervalLock } from '../../../../src/core/lock/interval-lock.js';
import { LockHolder } from '../../../../src/core/lock/lock-holder.js';
import { sleep } from '../../../../src/core/helpers.js';
import { NoopLeaseRenewalService } from './noop-lease-renewal-service.test.js';
import { Duration } from '../../../../src/core/time/duration.js';
import { container } from 'tsyringe-neo';
import { NamespaceName } from '../../../../src/types/namespace/namespace-name.js';
import { InjectTokens } from '../../../../src/core/dependency-injection/inject-tokens.js';
import { LockRelinquishmentError } from '../../../../src/core/lock/lock-relinquishment-error.js';
const defaultTimeout = Duration.ofMinutes(2).toMillis();
const leaseDuration = 4;
describe('Lease', async () => {
    const k8Factory = container.resolve(InjectTokens.K8Factory);
    const testNamespace = NamespaceName.of('lease-e2e');
    const renewalService = new NoopLeaseRenewalService();
    before(async function () {
        this.timeout(defaultTimeout);
        if (await k8Factory.default().namespaces().has(testNamespace)) {
            await k8Factory.default().namespaces().delete(testNamespace);
            await sleep(Duration.ofSeconds(5));
        }
        await k8Factory.default().namespaces().create(testNamespace);
    });
    after(async function () {
        this.timeout(defaultTimeout);
        await k8Factory.default().namespaces().delete(testNamespace);
    });
    describe('acquire and release', async function () {
        this.timeout(defaultTimeout);
        it('non-expired lease', async () => {
            const lease = new IntervalLock(k8Factory, renewalService, LockHolder.default(), testNamespace, undefined, leaseDuration);
            await lease.acquire();
            expect(await lease.isAcquired()).to.be.true;
            await lease?.release();
            expect(await lease.isAcquired()).to.be.false;
        });
        it('non-expired lease held by another user should not be released', async () => {
            const lease = new IntervalLock(k8Factory, renewalService, LockHolder.default(), testNamespace, undefined, leaseDuration);
            const newLease = new IntervalLock(k8Factory, renewalService, LockHolder.of('other'), testNamespace, undefined, leaseDuration);
            await lease.acquire();
            expect(await lease.isAcquired()).to.be.true;
            expect(await lease.isExpired()).to.be.false;
            await expect(newLease.release()).to.be.rejectedWith(LockRelinquishmentError);
            expect(await lease.isAcquired()).to.be.true;
            expect(await lease.isExpired()).to.be.false;
            await lease?.release();
            expect(await lease.isAcquired()).to.be.false;
            expect(await lease.isExpired()).to.be.false;
        });
        it('expired lease held by another user should be released', async () => {
            const lease = new IntervalLock(k8Factory, renewalService, LockHolder.default(), testNamespace, undefined, leaseDuration);
            const newLease = new IntervalLock(k8Factory, renewalService, LockHolder.of('other'), testNamespace, undefined, leaseDuration);
            await lease.acquire();
            expect(await lease.isAcquired()).to.be.true;
            expect(await lease.isExpired()).to.be.false;
            await sleep(Duration.ofSeconds(lease.durationSeconds).plusSeconds(1));
            expect(await lease.isAcquired()).to.be.false;
            expect(await lease.isExpired()).to.be.true;
            await newLease.release();
            expect(await lease.isAcquired()).to.be.false;
            expect(await lease.isExpired()).to.be.false;
        });
        it('expired lease should be released', async () => {
            const lease = new IntervalLock(k8Factory, renewalService, LockHolder.default(), testNamespace, undefined, leaseDuration);
            await lease.acquire();
            expect(await lease.isAcquired()).to.be.true;
            await sleep(Duration.ofSeconds(lease.durationSeconds).plusSeconds(1));
            expect(await lease.isAcquired()).to.be.false;
            expect(await lease.isExpired()).to.be.true;
            await lease?.release();
            expect(await lease.isAcquired()).to.be.false;
            expect(await lease.isExpired()).to.be.false;
        });
    });
    describe('tryAcquire and tryRelease', async function () {
        this.timeout(defaultTimeout);
        it('non-expired lease', async () => {
            const lease = new IntervalLock(k8Factory, renewalService, LockHolder.default(), testNamespace, undefined, leaseDuration);
            expect(await lease.tryAcquire()).to.be.true;
            expect(await lease.isAcquired()).to.be.true;
            expect(await lease.isExpired()).to.be.false;
            expect(await lease.tryRelease()).to.be.true;
            expect(await lease.isAcquired()).to.be.false;
            expect(await lease.isExpired()).to.be.false;
        });
        it('non-expired lease held by another user should not be released', async () => {
            const lease = new IntervalLock(k8Factory, renewalService, LockHolder.default(), testNamespace, undefined, leaseDuration);
            const newLease = new IntervalLock(k8Factory, renewalService, LockHolder.of('other'), testNamespace, undefined, leaseDuration);
            expect(await lease.tryAcquire()).to.be.true;
            expect(await lease.isAcquired()).to.be.true;
            expect(await lease.isExpired()).to.be.false;
            expect(await newLease.tryRelease()).to.be.false;
            expect(await lease.isAcquired()).to.be.true;
            expect(await lease.isExpired()).to.be.false;
            expect(await lease.tryRelease()).to.be.true;
            expect(await lease.isAcquired()).to.be.false;
            expect(await lease.isExpired()).to.be.false;
        });
        it('expired lease held by another user should be released', async () => {
            const lease = new IntervalLock(k8Factory, renewalService, LockHolder.default(), testNamespace, undefined, leaseDuration);
            const newLease = new IntervalLock(k8Factory, renewalService, LockHolder.of('other'), testNamespace, undefined, leaseDuration);
            expect(await lease.tryAcquire()).to.be.true;
            expect(await lease.isAcquired()).to.be.true;
            expect(await lease.isExpired()).to.be.false;
            await sleep(Duration.ofSeconds(lease.durationSeconds).plusSeconds(1));
            expect(await lease.isAcquired()).to.be.false;
            expect(await lease.isExpired()).to.be.true;
            expect(await newLease.tryRelease()).to.be.true;
            expect(await lease.isAcquired()).to.be.false;
            expect(await lease.isExpired()).to.be.false;
        });
        it('expired lease should be released', async () => {
            const lease = new IntervalLock(k8Factory, renewalService, LockHolder.default(), testNamespace, undefined, leaseDuration);
            expect(await lease.tryAcquire()).to.be.true;
            expect(await lease.isAcquired()).to.be.true;
            expect(await lease.isExpired()).to.be.false;
            await sleep(Duration.ofSeconds(lease.durationSeconds).plusSeconds(1));
            expect(await lease.isAcquired()).to.be.false;
            expect(await lease.isExpired()).to.be.true;
            expect(await lease.tryRelease()).to.be.true;
            expect(await lease.isAcquired()).to.be.false;
            expect(await lease.isExpired()).to.be.false;
        });
    });
});
//# sourceMappingURL=lease.test.js.map