Truffle — testing enum values

Question

I am using truffle to develop a simple smart contract.

I have the following contract:

contract Puppy {

  enum State { good, bad }

  State public status;
  State public constant INITIAL_STATUS = State.good;

  function Puppy() {
    status = INITIAL_STATUS;
  }
}

And I wish to test it as follows:

const Puppy = artifacts.require('./Puppy.sol')

contract('Puppy', () => {
  it('sets the initial status to \'good\'', () => Puppy.deployed()
    .then(instance => instance.status())
    .then((status) => {
      assert.equal(status, Puppy.State.good, 'Expected the status to be \'good\'')
    }))
})

This throws TypeError: Cannot read property 'good' of undefined

If I change the test to

const Puppy = artifacts.require('./Puppy.sol')

contract('Puppy', () => {
  it('sets the initial status to \'good\'', () => Puppy.deployed()
    .then(instance => instance.status())
    .then((status) => {
      assert.equal(status, 0, 'Expected the status to be \'good\'')
    }))
})

it passes.

How do I refer to the enum from within the test?

score 5 · Answer 1 · answered Oct 27 '17 at 08:18

You can't (for now), as enums are not supported by the ABI.

See https://solidity.readthedocs.io/en/latest/frequently-asked-questions.html#if-i-return-an-enum-i-only-get-integer-values-in-web3-js-how-to-get-the-named-values

There is a proposal to change that https://github.com/ethereum/EIPs/issues/47

score 3 · Accepted Answer · edited Sep 15 '21 at 16:34

Let's take the test-contract:

// SPDX-License-Identifier: MIT
pragma solidity ^0.8.3;
contract Test {
    enum Stages {
        stage_01,
        stage_02,
        stage_03,
        stage_04,
        stage_05
    }
Stages public stage = Stages.stage_01;

function setStage(Stages _stage) public {
    stage = _stage;
}

}

and test it such way:

const TestContract = artifacts.require('Test');
contract('Test', function (accounts) {
    const owner = accounts[0];
    const txParams = { from: owner };
beforeEach(async function () {
    this.testContract = await TestContract.new(txParams);
});

it('test initial stage', async function () {
    expect((await this.testContract.stage()).toString()).to.equal(TestContract.Stages.stage_01.toString());
});

it('assign custom stage', async function () {
    await this.testContract.setStage(TestContract.Stages.stage_05);
    expect((await this.testContract.stage()).toString()).to.equal(TestContract.Stages.stage_05.toString());
});

it('assign custom number to stage', async function () {
    await this.testContract.setStage(3); // take into account that enum indexed from zero
    expect((await this.testContract.stage()).toString()).to.equal(TestContract.Stages.stage_04.toString());
});      

});

Truffle — testing enum values

2 Answers2