setup for report double vote

Author: HoOngEe

test/src/e2e.dynval/dv.double-vote.test.ts

@@ -0,0 +1,152 @@
+// Copyright 2019 Kodebox, Inc.
+// This file is part of CodeChain.
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as
+// published by the Free Software Foundation, either version 3 of the
+// License, or (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+import * as chai from "chai";
+import { expect } from "chai";
+import * as chaiAsPromised from "chai-as-promised";
+import { SDK } from "codechain-sdk";
+import * as stake from "codechain-stakeholder-sdk";
+import { Custom } from "codechain-sdk/lib/core/transaction/Custom";
+import {
+    U64,
+    H256,
+    signSchnorr,
+    blake256,
+    H512
+} from "codechain-primitives/lib";
+import "mocha";
+import * as RLP from "rlp";
+
+import { validators as originalValidators } from "../../tendermint.dynval/constants";
+import { faucetAddress, faucetSecret } from "../helper/constants";
+import { PromiseExpect } from "../helper/promise";
+import { Signer } from "../helper/spawn";
+
+chai.use(chaiAsPromised);
+
+const HANDLER_ID = 2;
+const REPORT_DOUBLE_VOTE_ACTION_ID = 5;
+
+type MessageData = {
+    height: number;
+    view: number;
+    step: "propose" | "prevote" | "precommit" | "commit";
+    blockHash: H256 | null;
+    privKey: string;
+    signerIdx: number;
+};
+
+function encodableMessage(data: MessageData): RLP.Input {
+    const { height, view, step, blockHash, privKey, signerIdx } = data;
+    const encodableStep = (stepString => {
+        switch (stepString) {
+            case "propose":
+                return 0;
+            case "prevote":
+                return 1;
+            case "precommit":
+                return 2;
+            case "commit":
+                return 3;
+        }
+    })(step);
+    const encodableVoteStep = [
+        U64.ensure(height).toEncodeObject(),
+        U64.ensure(view).toEncodeObject(),
+        encodableStep
+    ];
+    const encodableBlockHash =
+        blockHash === null ? [] : [blockHash.toEncodeObject()];
+    const encodableVoteOn = [encodableVoteStep, encodableBlockHash];
+
+    const rlpVoteOn = RLP.encode(encodableVoteOn);
+    const messageForSchnorr = blake256(rlpVoteOn);
+    const schnorrSignature = signSchnorr(messageForSchnorr, privKey);
+    // pad because signSchnorr function does not guarantee the length of r and s to be 64.
+    const encodableSchnorrSignature = new H512(
+        schnorrSignature.r.padStart(64, "0") +
+            schnorrSignature.s.padStart(64, "0")
+    ).toEncodeObject();
+
+    return [
+        encodableVoteOn,
+        encodableSchnorrSignature,
+        U64.ensure(signerIdx).toEncodeObject()
+    ];
+}
+
+function createDoubleVoteMessages(
+    data: Omit<MessageData, "blockHash">,
+    blockHash1: H256 | null,
+    blockHash2: H256 | null
+) {
+    return [
+        encodableMessage({ ...data, blockHash: blockHash1 }),
+        encodableMessage({ ...data, blockHash: blockHash2 })
+    ];
+}
+
+function createReportDoubleVoteTransaction(
+    sdk: SDK,
+    message1: RLP.Input,
+    message2: RLP.Input
+): Custom {
+    return sdk.core.createCustomTransaction({
+        handlerId: HANDLER_ID,
+        bytes: RLP.encode([REPORT_DOUBLE_VOTE_ACTION_ID, message1, message2])
+    });
+}
+
+const allDynValidators = originalValidators.slice(0, 4);
+const [alice, ...otherDynValidators] = allDynValidators;
+
+async function expectPossibleAuthors(
+    sdk: SDK,
+    expected: Signer[],
+    blockNumber?: number
+) {
+    const authors = (await stake.getPossibleAuthors(sdk, blockNumber))!.map(
+        author => author.toString()
+    );
+    expect(authors)
+        .to.have.lengthOf(expected.length)
+        .and.to.include.members(
+            expected.map(signer => signer.platformAddress.toString())
+        );
+}
+
+// FIXME: neeeds to use common refactored function when gets banned state accounts
+async function ensureAliceIsBanned(sdk: SDK, blockNumber: number) {
+    await expectPossibleAuthors(sdk, otherDynValidators, blockNumber);
+    const bannedAfter = (await stake.getBanned(sdk, blockNumber)).map(
+        platformAddr => platformAddr.toString()
+    );
+    expect(bannedAfter).to.includes(alice.platformAddress.toString());
+    const delegteesAfter = (await stake.getDelegations(
+        sdk,
+        faucetAddress,
+        blockNumber
+    )).map(delegation => delegation.delegatee.toString());
+    expect(delegteesAfter).not.to.includes(alice.platformAddress.toString());
+}
+
+describe("Report Double Vote", function() {
+    const promiseExpect = new PromiseExpect();
+
+    afterEach(async function() {
+        promiseExpect.checkFulfilled();
+    });
+});