arbitrum-chain-sdk/src/getValidators.integration.test.ts at 616c28c8c7965efde623a27a93471120516303cd · OffchainLabs/arbitrum-chain-sdk · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
import { describe, it, expect } from 'vitest';
import { Address, createPublicClient, http } from 'viem';
import { generatePrivateKey, privateKeyToAccount } from 'viem/accounts';

import { nitroTestnodeL2 } from './chains';
import { rollupAdminLogicPublicActions } from './decorators/rollupAdminLogicPublicActions';
import { getInformationFromTestnode, getNitroTestnodePrivateKeyAccounts } from './testHelpers';
import { getValidators } from './getValidators';

const { l3RollupOwner } = getNitroTestnodePrivateKeyAccounts();
const { l3Rollup, l3UpgradeExecutor } = getInformationFromTestnode();

const client = createPublicClient({
  chain: nitroTestnodeL2,
  transport: http(),
}).extend(
  rollupAdminLogicPublicActions({
    rollup: l3Rollup,
  }),
);

async function setValidator(validator: Address, state: boolean) {
  const tx = await client.rollupAdminLogicPrepareTransactionRequest({
    functionName: 'setValidator',
    args: [[validator], [state]],
    account: l3RollupOwner.address,
    upgradeExecutor: l3UpgradeExecutor,
    rollup: l3Rollup,
  });

  const txHash = await client.sendRawTransaction({
    serializedTransaction: await l3RollupOwner.signTransaction(tx),
  });

  await client.waitForTransactionReceipt({
    hash: txHash,
  });
}

// Tests can be enabled once we run one node per integration test
describe('successfully get validators', () => {
  it('when disabling the same validator multiple time', async () => {
    const randomAccount = privateKeyToAccount(generatePrivateKey()).address;

    const { isAccurate: isAccurateInitially, validators: initialValidators } = await getValidators(
      client,
      {
        rollup: l3Rollup,
      },
    );
    // By default, chains from nitro testnode has 10 validators
    expect(initialValidators).toHaveLength(10);
    expect(isAccurateInitially).toBeTruthy();

    await setValidator(randomAccount, false);
    await setValidator(randomAccount, false);

    const { isAccurate: isStillAccurate, validators: newValidators } = await getValidators(client, {
      rollup: l3Rollup,
    });
    // Setting the same validator multiple time to false doesn't add new validators
    expect(newValidators).toEqual(initialValidators);
    expect(isStillAccurate).toBeTruthy();

    await setValidator(randomAccount, true);
    const { validators, isAccurate } = await getValidators(client, { rollup: l3Rollup });
    expect(validators).toEqual(initialValidators.concat(randomAccount));
    expect(isAccurate).toBeTruthy();

    // Reset state for future tests
    await setValidator(randomAccount, false);
    const { isAccurate: isAccurateFinal, validators: validatorsFinal } = await getValidators(
      client,
      {
        rollup: l3Rollup,
      },
    );
    expect(validatorsFinal).toEqual(initialValidators);
    expect(isAccurateFinal).toBeTruthy();
  });

  it('when enabling the same validators multiple time', async () => {
    const randomAccount = privateKeyToAccount(generatePrivateKey()).address;

    const { isAccurate: isAccurateInitially, validators: initialValidators } = await getValidators(
      client,
      {
        rollup: l3Rollup,
      },
    );
    // By default, chains from nitro testnode has 10 validators
    expect(initialValidators).toHaveLength(10);
    expect(isAccurateInitially).toBeTruthy();

    await setValidator(randomAccount, true);
    await setValidator(randomAccount, true);
    const { isAccurate: isStillAccurate, validators: newValidators } = await getValidators(client, {
      rollup: l3Rollup,
    });

    expect(newValidators).toEqual(initialValidators.concat(randomAccount));
    expect(isStillAccurate).toBeTruthy();

    // Reset state for futures tests
    await setValidator(randomAccount, false);
    const { validators, isAccurate } = await getValidators(client, { rollup: l3Rollup });
    expect(validators).toEqual(initialValidators);
    expect(isAccurate).toBeTruthy();
  });

  it('when adding an existing validator', async () => {
    const { isAccurate: isAccurateInitially, validators: initialValidators } = await getValidators(
      client,
      { rollup: l3Rollup },
    );
    expect(initialValidators).toHaveLength(10);
    expect(isAccurateInitially).toBeTruthy();

    const firstValidator = initialValidators[0];
    await setValidator(firstValidator, true);

    const { isAccurate, validators } = await getValidators(client, { rollup: l3Rollup });
    expect(validators).toEqual(initialValidators);
    expect(isAccurate).toBeTruthy();
  });

  it('when removing an existing validator', async () => {
    const { isAccurate: isAccurateInitially, validators: initialValidators } = await getValidators(
      client,
      { rollup: l3Rollup },
    );
    expect(initialValidators).toHaveLength(10);
    expect(isAccurateInitially).toBeTruthy();

    const lastValidator = initialValidators[initialValidators.length - 1];
    await setValidator(lastValidator, false);
    const { isAccurate, validators } = await getValidators(client, { rollup: l3Rollup });
    expect(validators).toEqual(initialValidators.slice(0, -1));
    expect(isAccurate).toBeTruthy();

    await setValidator(lastValidator, true);
    const { isAccurate: isAccurateFinal, validators: validatorsFinal } = await getValidators(
      client,
      { rollup: l3Rollup },
    );
    expect(validatorsFinal).toEqual(initialValidators);
    expect(isAccurateFinal).toBeTruthy();
  });
});