com.github.fge.jsonschema.TestUtils.java Source code

Introduction

Here is the source code for com.github.fge.jsonschema.TestUtils.java

Source

/*
 * Copyright (c) 2014, Francis Galiegue (fgaliegue@gmail.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 this file and of both licenses is available at the root of this
 * project or, if you have the jar distribution, in directory META-INF/, 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.fge.jsonschema;

import com.fasterxml.jackson.databind.JsonNode;
import com.github.fge.jackson.NodeType;
import com.github.fge.jsonschema.core.report.ProcessingMessage;
import com.github.fge.jsonschema.core.report.ProcessingReport;
import com.github.fge.jsonschema.core.tree.SchemaTree;
import com.github.fge.msgsimple.bundle.MessageBundle;
import com.google.common.collect.Lists;
import org.mockito.verification.VerificationMode;

import java.util.List;

import static org.mockito.Mockito.*;

public final class TestUtils {
    private TestUtils() {
    }

    public static VerificationMode onlyOnce() {
        return times(1);
    }

    public static ProcessingReport anyReport() {
        return any(ProcessingReport.class);
    }

    public static SchemaTree anySchema() {
        return any(SchemaTree.class);
    }

    public static ProcessingMessage anyMessage() {
        return any(ProcessingMessage.class);
    }

    // TODO: while we're at it, the @DataProvider could be factorized!
    public static String buildMessage(final MessageBundle BUNDLE, final String key, final JsonNode params,
            final JsonNode data) {
        final ProcessingMessage message = new ProcessingMessage().setMessage(BUNDLE.getMessage(key));
        if (params != null) {
            String name;
            JsonNode value;
            for (final JsonNode node : params) {
                name = node.textValue();
                value = data.get(name);
                message.putArgument(name, valueToArgument(value));
            }
        }
        return message.getMessage();
    }

    private static Object valueToArgument(final JsonNode value) {
        final NodeType type = NodeType.getNodeType(value);

        switch (type) {
        case STRING:
            return value.textValue();
        case INTEGER:
            return value.bigIntegerValue();
        case NUMBER:
        case NULL:
            return value;
        case BOOLEAN:
            return value.booleanValue();
        case ARRAY:
            final List<Object> list = Lists.newArrayList();
            for (final JsonNode element : value)
                list.add(valueToArgument(element));
            return list;
        default:
            throw new UnsupportedOperationException();
        }
    }
}