com.redhat.lightblue.metadata.PredefinedFieldsTest.java Source code

Introduction

Here is the source code for com.redhat.lightblue.metadata.PredefinedFieldsTest.java

Source

/*
 Copyright 2013 Red Hat, Inc. and/or its affiliates.
    
 This file is part of lightblue.
    
 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.
    
 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.
    
 You should have received a copy of the GNU General Public License
 along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package com.redhat.lightblue.metadata;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.JsonNodeFactory;
import com.fasterxml.jackson.databind.node.ObjectNode;
import com.redhat.lightblue.metadata.parser.Extensions;
import com.redhat.lightblue.metadata.parser.JSONMetadataParser;
import com.redhat.lightblue.metadata.types.DefaultTypes;
import com.redhat.lightblue.metadata.types.IntegerType;
import com.redhat.lightblue.metadata.types.StringType;
import com.redhat.lightblue.util.JsonDoc;
import com.redhat.lightblue.util.Path;
import org.junit.Assert;
import org.junit.Test;

public class PredefinedFieldsTest {

    private EntityMetadata getMD1() {
        EntityMetadata entityMetadata = new EntityMetadata("test");

        entityMetadata.getFields().addNew(new SimpleField("simpleInteger", IntegerType.TYPE));
        entityMetadata.getFields().addNew(new SimpleField("simpleString", StringType.TYPE));
        ObjectField objectField1 = new ObjectField("obj1");
        entityMetadata.getFields().addNew(objectField1);
        objectField1.getFields().addNew(new SimpleField("nestedSimpleInteger", IntegerType.TYPE));
        objectField1.getFields().addNew(new SimpleField("nestedSimpleString", StringType.TYPE));
        ObjectField objectField2 = new ObjectField("nested");
        objectField1.getFields().addNew(objectField2);
        objectField2.getFields().addNew(new SimpleField("doubleNestedString", StringType.TYPE));
        ArrayField arrayField1 = new ArrayField("simpleArr", new SimpleArrayElement(StringType.TYPE));
        //arrayField1.addNew(new SimpleField("nestedSimpleString", StringType.TYPE));
        entityMetadata.getFields().addNew(arrayField1);

        ObjectArrayElement objectArrayElement = new ObjectArrayElement();
        objectArrayElement.getFields().addNew(new SimpleField("nestedArrObjString1", StringType.TYPE));
        objectArrayElement.getFields().addNew(new SimpleField("nestedArrObjString2", StringType.TYPE));
        ArrayField arrayField2 = new ArrayField("objArr", objectArrayElement);
        objectField2.getFields().addNew(arrayField2);

        return entityMetadata;
    }

    private EntityMetadata getMD2() {
        EntityMetadata entityMetadata = new EntityMetadata("test");

        entityMetadata.getFields().addNew(new SimpleField("simpleInteger", IntegerType.TYPE));
        entityMetadata.getFields().addNew(new SimpleField("simpleString", StringType.TYPE));
        entityMetadata.getFields().addNew(new SimpleField("objectType", StringType.TYPE));
        entityMetadata.getFields().addNew(new SimpleField("_id", IntegerType.TYPE));
        ObjectField objectField1 = new ObjectField("obj1");
        entityMetadata.getFields().addNew(objectField1);
        objectField1.getFields().addNew(new SimpleField("nestedSimpleInteger", IntegerType.TYPE));
        objectField1.getFields().addNew(new SimpleField("nestedSimpleString", StringType.TYPE));
        ObjectField objectField2 = new ObjectField("nested");
        objectField1.getFields().addNew(objectField2);
        objectField2.getFields().addNew(new SimpleField("doubleNestedString", StringType.TYPE));
        ArrayField arrayField1 = new ArrayField("simpleArr", new SimpleArrayElement(StringType.TYPE));
        //arrayField1.addNew(new SimpleField("nestedSimpleString", StringType.TYPE));
        entityMetadata.getFields().addNew(arrayField1);

        ObjectArrayElement objectArrayElement = new ObjectArrayElement();
        objectArrayElement.getFields().addNew(new SimpleField("nestedArrObjString1", StringType.TYPE));
        objectArrayElement.getFields().addNew(new SimpleField("nestedArrObjString2", StringType.TYPE));
        ArrayField arrayField2 = new ArrayField("objArr", objectArrayElement);
        objectField2.getFields().addNew(arrayField2);
        objectField2.getFields().addNew(new SimpleField("objArr#", IntegerType.TYPE));

        return entityMetadata;
    }

    @Test
    public void testAddPredef() throws Exception {
        EntityMetadata md = getMD1();
        try {
            md.resolve(new Path("objectType"));
            Assert.fail();
        } catch (Exception e) {
        }
        try {
            md.resolve(new Path("simpleArr#"));
        } catch (Exception e) {
        }
        try {
            md.resolve(new Path("nested.objArr#"));
        } catch (Exception e) {
        }

        PredefinedFields.ensurePredefinedFields(md);
        Extensions<JsonNode> x = new Extensions<>();
        x.addDefaultExtensions();
        JSONMetadataParser p = new JSONMetadataParser(x, new DefaultTypes(),
                JsonNodeFactory.withExactBigDecimals(false));
        System.out.println(p.convert(md).toString());

        Field f = (SimpleField) md.resolve(new Path("objectType"));
        Assert.assertEquals(StringType.TYPE, f.getType());
        Assert.assertTrue(f.getAccess().getFind().getRoles().contains(MetadataConstants.ROLE_ANYONE));

        f = (SimpleField) md.resolve(new Path("simpleArr#"));
        Assert.assertEquals(IntegerType.TYPE, f.getType());
        Assert.assertTrue(f.getAccess().getFind().getRoles().contains(MetadataConstants.ROLE_ANYONE));

        f = (SimpleField) md.resolve(new Path("obj1.nested.objArr#"));
        Assert.assertEquals(IntegerType.TYPE, f.getType());
        Assert.assertTrue(f.getAccess().getFind().getRoles().contains(MetadataConstants.ROLE_ANYONE));
    }

    @Test
    public void testModPredef() throws Exception {
        EntityMetadata md = getMD2();

        PredefinedFields.ensurePredefinedFields(md);

        Field f = (SimpleField) md.resolve(new Path("objectType"));
        Assert.assertEquals(StringType.TYPE, f.getType());
        Assert.assertTrue(f.getAccess().getFind().getRoles().contains(MetadataConstants.ROLE_ANYONE));
        Assert.assertTrue(f.getAccess().getUpdate().getRoles().contains(MetadataConstants.ROLE_NOONE));

        f = (SimpleField) md.resolve(new Path("simpleArr#"));
        Assert.assertEquals(IntegerType.TYPE, f.getType());
        Assert.assertTrue(f.getAccess().getFind().getRoles().contains(MetadataConstants.ROLE_ANYONE));

        f = (SimpleField) md.resolve(new Path("obj1.nested.objArr#"));
        Assert.assertEquals(IntegerType.TYPE, f.getType());
        Assert.assertTrue(f.getAccess().getFind().getRoles().contains(MetadataConstants.ROLE_ANYONE));
    }

    @Test
    public void testInvalidObjectType() throws Exception {
        EntityMetadata md = new EntityMetadata("test");
        md.getFields().addNew(new SimpleField("objectType", IntegerType.TYPE));
        try {
            PredefinedFields.ensurePredefinedFields(md);
            Assert.fail();
        } catch (Exception e) {
        }
    }

    @Test
    public void testInvalidArr() throws Exception {
        EntityMetadata md = new EntityMetadata("test");
        md.getFields().addNew(new ArrayField("x", new SimpleArrayElement(IntegerType.TYPE)));
        md.getFields().addNew(new SimpleField("x#", StringType.TYPE));
        try {
            PredefinedFields.ensurePredefinedFields(md);
            Assert.fail();
        } catch (Exception e) {
        }
    }

    @Test
    public void testDocModify() throws Exception {
        JsonNodeFactory factory = JsonNodeFactory.withExactBigDecimals(false);
        ObjectNode node = factory.objectNode();
        JsonDoc doc = new JsonDoc(node);
        node.put("fld1", "value").put("arr#", 1);
        node.set("arr", factory.arrayNode().add(1).add(2));
        node.set("arr2", factory.arrayNode().add(1).add(2).add(3));
        ObjectNode obj = factory.objectNode();
        obj.put("fld3", "val");
        obj.set("arr3", factory.arrayNode().add(1).add(2).add(3).add(4));
        node.set("obj", obj);

        PredefinedFields.updateArraySizes(factory, node);

        Assert.assertEquals(2, doc.get(new Path("arr#")).intValue());
        Assert.assertEquals(3, doc.get(new Path("arr2#")).intValue());
        Assert.assertEquals(4, doc.get(new Path("obj.arr3#")).intValue());
    }
}