com.github.reinert.jjschema.v1.SchemaIgnoreTest.java Source code

Introduction

Here is the source code for com.github.reinert.jjschema.v1.SchemaIgnoreTest.java

Source

/*
 * Copyright (c) 2014, Danilo Reinert (daniloreinert@growbit.com)
 *
 * This software is dual-licensed under:
 *
 * - the Lesser General Public License (LGPL) version 3.0 or, at your option, any
 *   later version;
 * - the Apache Software License (ASL) version 2.0.
 *
 * The text of both licenses is available under the src/resources/ directory of
 * this project (under the names LGPL-3.0.txt and ASL-2.0.txt respectively).
 *
 * Direct link to the sources:
 *
 * - LGPL 3.0: https://www.gnu.org/licenses/lgpl-3.0.txt
 * - ASL 2.0: http://www.apache.org/licenses/LICENSE-2.0.txt
 */

package com.github.reinert.jjschema.v1;

import com.fasterxml.jackson.annotation.JsonBackReference;
import com.fasterxml.jackson.annotation.JsonManagedReference;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.github.reinert.jjschema.JsonSchemaGenerator;
import com.github.reinert.jjschema.SchemaGeneratorBuilder;
import com.github.reinert.jjschema.SchemaIgnore;
import junit.framework.TestCase;

import java.util.List;

/**
 * @author reinert
 */
public class SchemaIgnoreTest extends TestCase {

    static ObjectMapper MAPPER = new ObjectMapper();
    JsonSchemaFactory v4generator = new JsonSchemaV4Factory();

    public SchemaIgnoreTest(String testName) {
        super(testName);
    }

    /**
     * Test if @SchemaIgnore works correctly
     */
    public void testGenerateSchema() {

        JsonNode schema = v4generator.createSchema(Sale.class);
        //System.out.println(MAPPER.writerWithDefaultPrettyPrinter().writeValueAsString(schema));
        JsonNode properties = schema.get("properties");
        assertEquals(1, properties.size());

        schema = v4generator.createSchema(SaleItem.class);
        //System.out.println(MAPPER.writerWithDefaultPrettyPrinter().writeValueAsString(schema));
        properties = schema.get("properties");
        assertEquals(2, properties.size());
    }

    static class Sale {
        int id;
        @SchemaIgnore
        @JsonManagedReference
        List<SaleItem> saleItems;

        public int getId() {
            return id;
        }

        public void setId(int id) {
            this.id = id;
        }

        public List<SaleItem> getSaleItems() {
            return saleItems;
        }

        public void setSaleItems(List<SaleItem> saleItems) {
            this.saleItems = saleItems;
        }
    }

    static class SaleItem {
        int idSale;
        int seqNumber;
        @SchemaIgnore
        @JsonBackReference
        Sale parent;

        public int getIdSale() {
            return idSale;
        }

        public void setIdSale(int idSale) {
            this.idSale = idSale;
        }

        public int getSeqNumber() {
            return seqNumber;
        }

        public void setSeqNumber(int seqNumber) {
            this.seqNumber = seqNumber;
        }

        public Sale getParent() {
            return parent;
        }

        public void setParent(Sale parent) {
            this.parent = parent;
        }
    }
}