authorize.test.ts

import { newKitFromProvider } from '@celo/contractkit'
import { testWithAnvilL2 } from '@celo/dev-utils/anvil-test'
import { addressToPublicKey } from '@celo/utils/lib/signatureUtils'
import { stripAnsiCodesFromNestedArray, testLocallyWithNode } from '../../test-utils/cliUtils'
import { PROOF_OF_POSSESSION_SIGNATURE } from '../../test-utils/constants'
import Lock from '../lockedcelo/lock'
import ValidatorRegister from '../validator/register'
import Authorize from './authorize'
import Register from './register'

process.env.NO_SYNCCHECK = 'true'

testWithAnvilL2('account:authorize cmd', (provider) => {
  const logMock = jest.spyOn(console, 'log')
  const errorMock = jest.spyOn(console, 'error')

  beforeEach(() => {
    logMock.mockClear().mockImplementation()
    errorMock.mockClear().mockImplementation()
  })

  afterEach(() => jest.clearAllMocks())

  test('can authorize vote signer', async () => {
    const kit = newKitFromProvider(provider)
    const accounts = await kit.connection.getAccounts()
    const notRegisteredAccount = accounts[0]
    const signerNotRegisteredAccount = accounts[1]

    await testLocallyWithNode(Register, ['--from', notRegisteredAccount], provider)

    logMock.mockClear()

    await testLocallyWithNode(
      Authorize,
      [
        '--from',
        notRegisteredAccount,
        '--role',
        'vote',
        '--signer',
        signerNotRegisteredAccount,
        '--signature',
        PROOF_OF_POSSESSION_SIGNATURE,
      ],
      provider
    )

    expect(stripAnsiCodesFromNestedArray(logMock.mock.calls)).toMatchInlineSnapshot(`
      [
        [
          "Running Checks:",
        ],
        [
          "   ✔  0x5409ED021D9299bf6814279A6A1411A7e866A631 is a registered Account ",
        ],
        [
          "All checks passed",
        ],
        [
          "SendTransaction: authorizeTx",
        ],
        [
          "txHash: 0xtxhash",
        ],
      ]
    `)
  })

  test('can authorize attestation signer', async () => {
    const kit = newKitFromProvider(provider)
    const accounts = await kit.connection.getAccounts()
    const notRegisteredAccount = accounts[0]
    const signerNotRegisteredAccount = accounts[1]

    await testLocallyWithNode(Register, ['--from', notRegisteredAccount], provider)

    logMock.mockClear()

    await testLocallyWithNode(
      Authorize,
      [
        '--from',
        notRegisteredAccount,
        '--role',
        'attestation',
        '--signer',
        signerNotRegisteredAccount,
        '--signature',
        PROOF_OF_POSSESSION_SIGNATURE,
      ],
      provider
    )

    expect(stripAnsiCodesFromNestedArray(logMock.mock.calls)).toMatchInlineSnapshot(`
      [
        [
          "Running Checks:",
        ],
        [
          "   ✔  0x5409ED021D9299bf6814279A6A1411A7e866A631 is a registered Account ",
        ],
        [
          "All checks passed",
        ],
        [
          "SendTransaction: authorizeTx",
        ],
        [
          "txHash: 0xtxhash",
        ],
      ]
    `)
  })

  test('can authorize validator signer before validator is registered', async () => {
    const kit = newKitFromProvider(provider)
    const accounts = await kit.connection.getAccounts()
    const notRegisteredAccount = accounts[0]
    const signerNotRegisteredAccount = accounts[1]

    await testLocallyWithNode(Register, ['--from', notRegisteredAccount], provider)

    logMock.mockClear()

    await testLocallyWithNode(
      Authorize,
      [
        '--from',
        notRegisteredAccount,
        '--role',
        'validator',
        '--signer',
        signerNotRegisteredAccount,
        '--signature',
        PROOF_OF_POSSESSION_SIGNATURE,
      ],
      provider
    )

    expect(stripAnsiCodesFromNestedArray(logMock.mock.calls)).toMatchInlineSnapshot(`
      [
        [
          "Running Checks:",
        ],
        [
          "   ✔  0x5409ED021D9299bf6814279A6A1411A7e866A631 is a registered Account ",
        ],
        [
          "All checks passed",
        ],
        [
          "SendTransaction: authorizeTx",
        ],
        [
          "txHash: 0xtxhash",
        ],
      ]
    `)
    expect(stripAnsiCodesFromNestedArray(errorMock.mock.calls)).toMatchInlineSnapshot(`[]`)
  })

  it('can authorize validator signer after validator is registered', async () => {
    const kit = newKitFromProvider(provider)
    const accounts = await kit.connection.getAccounts()
    const notRegisteredAccount = accounts[0]
    const signerNotRegisteredAccount = accounts[1]
    const ecdsaPublicKey = await addressToPublicKey(notRegisteredAccount, kit.connection.sign)
    await testLocallyWithNode(Register, ['--from', notRegisteredAccount], provider)
    await testLocallyWithNode(
      Lock,
      ['--from', notRegisteredAccount, '--value', '10000000000000000000000'],
      provider
    )
    await testLocallyWithNode(
      ValidatorRegister,
      ['--from', notRegisteredAccount, '--ecdsaKey', ecdsaPublicKey, '--yes'],
      provider
    )

    logMock.mockClear()

    await testLocallyWithNode(
      Authorize,
      [
        '--from',
        notRegisteredAccount,
        '--role',
        'validator',
        '--signer',
        signerNotRegisteredAccount,
        '--signature',
        PROOF_OF_POSSESSION_SIGNATURE,
      ],
      provider
    )

    expect(stripAnsiCodesFromNestedArray(logMock.mock.calls)).toMatchInlineSnapshot(`
      [
        [
          "Running Checks:",
        ],
        [
          "   ✔  0x5409ED021D9299bf6814279A6A1411A7e866A631 is a registered Account ",
        ],
        [
          "All checks passed",
        ],
        [
          "SendTransaction: authorizeTx",
        ],
        [
          "txHash: 0xtxhash",
        ],
      ]
    `)
  })

  it('fails when using BLS keys on L2', async () => {
    const kit = newKitFromProvider(provider)
    const accounts = await kit.connection.getAccounts()
    const notRegisteredAccount = accounts[0]
    const signerNotRegisteredAccount = accounts[1]
    const ecdsaPublicKey = await addressToPublicKey(notRegisteredAccount, kit.connection.sign)
    await testLocallyWithNode(Register, ['--from', notRegisteredAccount], provider)
    await testLocallyWithNode(
      Lock,
      ['--from', notRegisteredAccount, '--value', '10000000000000000000000'],
      provider
    )
    await testLocallyWithNode(
      ValidatorRegister,
      ['--from', notRegisteredAccount, '--ecdsaKey', ecdsaPublicKey, '--yes'],
      provider
    )

    logMock.mockClear()

    await expect(
      testLocallyWithNode(
        Authorize,
        [
          '--from',
          notRegisteredAccount,
          '--role',
          'validator',
          '--signer',
          signerNotRegisteredAccount,
          '--signature',
          PROOF_OF_POSSESSION_SIGNATURE,
          '--blsKey',
          '0x4fa3f67fc913878b068d1fa1cdddc54913d3bf988dbe5a36a20fa888f20d4894c408a6773f3d7bde11154f2a3076b700d345a42fd25a0e5e83f4db5586ac7979ac2053cd95d8f2efd3e959571ceccaa743e02cf4be3f5d7aaddb0b06fc9aff00',
          '--blsPop',
          '0xcdb77255037eb68897cd487fdd85388cbda448f617f874449d4b11588b0b7ad8ddc20d9bb450b513bb35664ea3923900',
        ],

        provider
      )
    ).rejects.toMatchInlineSnapshot(`
      [Error: Nonexistent flags: --blsKey, --blsPop
      See more help with --help]
    `)

    expect(stripAnsiCodesFromNestedArray(logMock.mock.calls)).toMatchInlineSnapshot(`[]`)
  })

  test('can force authorize validator signer without BLS after validator is registered', async () => {
    const kit = newKitFromProvider(provider)
    const accounts = await kit.connection.getAccounts()
    const notRegisteredAccount = accounts[0]
    const signerNotRegisteredAccount = accounts[1]
    const ecdsaPublicKey = await addressToPublicKey(notRegisteredAccount, kit.connection.sign)
    await testLocallyWithNode(Register, ['--from', notRegisteredAccount], provider)
    await testLocallyWithNode(
      Lock,
      ['--from', notRegisteredAccount, '--value', '10000000000000000000000'],
      provider
    )
    await testLocallyWithNode(
      ValidatorRegister,
      ['--from', notRegisteredAccount, '--ecdsaKey', ecdsaPublicKey, '--yes'],
      provider
    )

    logMock.mockClear()

    await testLocallyWithNode(
      Authorize,
      [
        '--from',
        notRegisteredAccount,
        '--role',
        'validator',
        '--signer',
        signerNotRegisteredAccount,
        '--signature',
        PROOF_OF_POSSESSION_SIGNATURE,
        '--force',
      ],
      provider
    )
    expect(stripAnsiCodesFromNestedArray(errorMock.mock.calls)).toMatchInlineSnapshot(`[]`)
    expect(stripAnsiCodesFromNestedArray(logMock.mock.calls)).toMatchInlineSnapshot(`
      [
        [
          "Running Checks:",
        ],
        [
          "   ✔  0x5409ED021D9299bf6814279A6A1411A7e866A631 is a registered Account ",
        ],
        [
          "All checks passed",
        ],
        [
          "SendTransaction: authorizeTx",
        ],
        [
          "txHash: 0xtxhash",
        ],
      ]
    `)
  })

  test('fails if from is not an account', async () => {
    const kit = newKitFromProvider(provider)
    const accounts = await kit.connection.getAccounts()
    const notRegisteredAccount = accounts[0]
    const signerNotRegisteredAccount = accounts[1]

    logMock.mockClear()

    await expect(
      testLocallyWithNode(
        Authorize,
        [
          '--from',
          notRegisteredAccount,
          '--role',
          'validator',
          '--signer',
          signerNotRegisteredAccount,
          '--signature',
          PROOF_OF_POSSESSION_SIGNATURE,
        ],

        provider
      )
    ).rejects.toThrowErrorMatchingInlineSnapshot(`"Some checks didn't pass!"`)
    expect(stripAnsiCodesFromNestedArray(errorMock.mock.calls)).toMatchInlineSnapshot(`[]`)
    expect(stripAnsiCodesFromNestedArray(logMock.mock.calls)).toMatchInlineSnapshot(`
      [
        [
          "Running Checks:",
        ],
        [
          "   ✘  0x5409ED021D9299bf6814279A6A1411A7e866A631 is a registered Account 0x5409ED021D9299bf6814279A6A1411A7e866A631 is not registered as an account. Try running account:register",
        ],
      ]
    `)
  })
})