EnumNumberFormatShape3580PojoTest.java

package tools.jackson.databind.format;

import org.junit.jupiter.api.Test;

import com.fasterxml.jackson.annotation.JsonFormat;
import com.fasterxml.jackson.annotation.JsonValue;

import tools.jackson.databind.ObjectMapper;
import tools.jackson.databind.exc.InvalidFormatException;
import tools.jackson.databind.exc.MismatchedInputException;
import tools.jackson.databind.json.JsonMapper;
import tools.jackson.databind.testutil.DatabindTestUtil;

import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertThrows;

// [databind#3580] Enum (de)serialization in conjunction with JsonFormat.Shape.NUMBER_INT
public class EnumNumberFormatShape3580PojoTest
    extends DatabindTestUtil
{
    public static class Pojo3580 {
        public PojoStateInt3580 state;
        public Pojo3580() {}
        public Pojo3580(PojoStateInt3580 state) {this.state = state;}
    }

    @JsonFormat(shape = JsonFormat.Shape.NUMBER_INT)
    public enum PojoStateInt3580 {
        OFF(17),
        ON(31),
        UNKNOWN(99);

        private int value;

        PojoStateInt3580(int value) { this.value = value; }

        @JsonValue
        public int value() {return this.value;}
    }

    public static class PojoNum3580 {
        public PojoStateNum3580 state;
        public PojoNum3580() {}
        public PojoNum3580(PojoStateNum3580 state) {this.state = state;}
    }

    @JsonFormat(shape = JsonFormat.Shape.NUMBER)
    public enum PojoStateNum3580 {
        OFF(17),
        ON(31),
        UNKNOWN(99);

        private int value;

        PojoStateNum3580(int value) { this.value = value; }

        @JsonValue
        public int value() {return this.value;}
    }

    @Test
    public void testEnumNumberIntFormatShape3580() throws Exception
    {
        ObjectMapper mapper = JsonMapper.builder().build();

        // Serialize
        assertEquals("{\"state\":17}", mapper.writeValueAsString(new Pojo3580(PojoStateInt3580.OFF))); //
        assertEquals("{\"state\":31}", mapper.writeValueAsString(new Pojo3580(PojoStateInt3580.ON))); //
        assertEquals("{\"state\":99}", mapper.writeValueAsString(new Pojo3580(PojoStateInt3580.UNKNOWN))); //

        // Pass Deserialize
        assertEquals(PojoStateInt3580.OFF, mapper.readValue("{\"state\":17}", Pojo3580.class).state); // Pojo[state=OFF]
        assertEquals(PojoStateInt3580.ON, mapper.readValue("{\"state\":31}", Pojo3580.class).state); // Pojo[state=OFF]
        assertEquals(PojoStateInt3580.UNKNOWN, mapper.readValue("{\"state\":99}", Pojo3580.class).state); // Pojo[state=OFF]

        // Fail : Try to use ordinal number
        assertThrows(InvalidFormatException.class, () -> mapper.readValue("{\"state\":0}", Pojo3580.class));
    }

    @Test
    public void testEnumNumberFormatShape3580() throws Exception
    {
        ObjectMapper mapper = JsonMapper.builder().build();

        // Serialize
        assertEquals("{\"state\":17}", mapper.writeValueAsString(new PojoNum3580(PojoStateNum3580.OFF))); //
        assertEquals("{\"state\":31}", mapper.writeValueAsString(new PojoNum3580(PojoStateNum3580.ON))); //
        assertEquals("{\"state\":99}", mapper.writeValueAsString(new PojoNum3580(PojoStateNum3580.UNKNOWN))); //

        // Pass Deserialize
        assertEquals(PojoStateNum3580.OFF, mapper.readValue("{\"state\":17}", PojoNum3580.class).state); // Pojo[state=OFF]
        assertEquals(PojoStateNum3580.ON, mapper.readValue("{\"state\":31}", PojoNum3580.class).state); // Pojo[state=OFF]
        assertEquals(PojoStateNum3580.UNKNOWN, mapper.readValue("{\"state\":99}", PojoNum3580.class).state); // Pojo[state=OFF]

        // Fail : Try to use ordinal number
        assertThrows(MismatchedInputException.class, () -> mapper.readValue("{\"state\":0}", PojoStateNum3580.class));
    }
}