"use strict";
// SPDX-License-Identifier: Apache-2.0
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const ts_jest_1 = require("@golevelup/ts-jest");
const DataFixture_1 = require("../../../../../../../../__tests__/fixtures/shared/DataFixture");
const EvmAddress_1 = __importDefault(require("../../../../../../../domain/context/contract/EvmAddress"));
const Account_1 = __importDefault(require("../../../../../../../domain/context/account/Account"));
const SecurityRole_1 = require("../../../../../../../domain/context/security/SecurityRole");
const BaseError_1 = require("../../../../../../../core/error/BaseError");
const FreezePartialTokensCommandError_1 = require("./error/FreezePartialTokensCommandError");
const FreezePartialTokensCommand_1 = require("./FreezePartialTokensCommand");
const FreezePartialTokensCommandHandler_1 = require("./FreezePartialTokensCommandHandler");
const BigDecimal_1 = __importDefault(require("../../../../../../../domain/context/shared/BigDecimal"));
const SecurityFixture_1 = require("../../../../../../../../__tests__/fixtures/shared/SecurityFixture");
const Security_1 = require("../../../../../../../domain/context/security/Security");
const FreezeFixture_1 = require("../../../../../../../../__tests__/fixtures/freeze/FreezeFixture");
describe("FreezePartialTokensCommandHandler", () => {
    let handler;
    let command;
    const transactionServiceMock = (0, ts_jest_1.createMock)();
    const validationServiceMock = (0, ts_jest_1.createMock)();
    const accountServiceMock = (0, ts_jest_1.createMock)();
    const contractServiceMock = (0, ts_jest_1.createMock)();
    const securityServiceMock = (0, ts_jest_1.createMock)();
    const evmAddress = new EvmAddress_1.default(DataFixture_1.EvmAddressPropsFixture.create().value);
    const account = new Account_1.default({
        id: DataFixture_1.HederaIdPropsFixture.create().value,
        evmAddress: DataFixture_1.EvmAddressPropsFixture.create().value,
    });
    const transactionId = DataFixture_1.TransactionIdFixture.create().id;
    const errorMsg = DataFixture_1.ErrorMsgFixture.create().msg;
    const security = new Security_1.Security(SecurityFixture_1.SecurityPropsFixture.create());
    beforeEach(() => {
        handler = new FreezePartialTokensCommandHandler_1.FreezePartialTokensCommandHandler(securityServiceMock, accountServiceMock, transactionServiceMock, validationServiceMock, contractServiceMock);
        command = FreezeFixture_1.FreezePartialTokensCommandFixture.create();
    });
    afterEach(() => {
        jest.resetAllMocks();
    });
    describe("execute", () => {
        it("throws FreezePartialTokensCommandError when command fails with uncaught error", async () => {
            const fakeError = new Error(errorMsg);
            contractServiceMock.getContractEvmAddress.mockRejectedValue(fakeError);
            const resultPromise = handler.execute(command);
            await expect(resultPromise).rejects.toBeInstanceOf(FreezePartialTokensCommandError_1.FreezePartialTokensCommandError);
            await expect(resultPromise).rejects.toMatchObject({
                message: expect.stringContaining(`An error occurred while freeze partial tokens: ${errorMsg}`),
                errorCode: BaseError_1.ErrorCode.UncaughtCommandError,
            });
        });
        it("should successfully freeze amount of tokens", async () => {
            contractServiceMock.getContractEvmAddress.mockResolvedValueOnce(evmAddress);
            accountServiceMock.getAccountEvmAddress.mockResolvedValueOnce(evmAddress);
            accountServiceMock.getCurrentAccount.mockReturnValue(account);
            validationServiceMock.checkPause.mockResolvedValue(undefined);
            validationServiceMock.checkRole.mockResolvedValue(undefined);
            validationServiceMock.checkDecimals.mockResolvedValue(undefined);
            securityServiceMock.get.mockResolvedValue(security);
            transactionServiceMock.getHandler().freezePartialTokens.mockResolvedValue({
                id: transactionId,
            });
            const result = await handler.execute(command);
            expect(result).toBeInstanceOf(FreezePartialTokensCommand_1.FreezePartialTokensResponse);
            expect(result.payload).toBe(true);
            expect(result.transactionId).toBe(transactionId);
            expect(contractServiceMock.getContractEvmAddress).toHaveBeenCalledTimes(1);
            expect(validationServiceMock.checkPause).toHaveBeenCalledTimes(1);
            expect(validationServiceMock.checkAnyRole).toHaveBeenCalledTimes(1);
            expect(accountServiceMock.getCurrentAccount).toHaveBeenCalledTimes(1);
            expect(transactionServiceMock.getHandler().freezePartialTokens).toHaveBeenCalledTimes(1);
            expect(validationServiceMock.checkPause).toHaveBeenCalledWith(command.securityId);
            expect(validationServiceMock.checkAnyRole).toHaveBeenCalledWith([SecurityRole_1.SecurityRole._FREEZE_MANAGER_ROLE, SecurityRole_1.SecurityRole._AGENT_ROLE], account.id.toString(), command.securityId);
            expect(contractServiceMock.getContractEvmAddress).toHaveBeenCalledWith(command.securityId);
            expect(transactionServiceMock.getHandler().freezePartialTokens).toHaveBeenCalledWith(evmAddress, BigDecimal_1.default.fromString(command.amount, security.decimals), evmAddress, command.securityId);
        });
    });
});
