uk.ac.jorum.integration.matchers.ContainsJSONKey.java Source code

Introduction

Here is the source code for uk.ac.jorum.integration.matchers.ContainsJSONKey.java

Source

/**
 * The contents of this file are subject to the license and copyright
 * detailed in the LICENSE and NOTICE files at the root of the source
 * tree and available online at
 *
 * http://www.dspace.org/license/
 */
package uk.ac.jorum.integration.matchers;

import org.hamcrest.Description;
import org.hamcrest.Factory;
import org.hamcrest.Matcher;
import org.json.simple.JSONObject;
import org.junit.internal.matchers.TypeSafeMatcher;

public class ContainsJSONKey<T> extends TypeSafeMatcher<JSONObject> {
    @Factory
    public static Matcher<JSONObject> hasKey(String key) {
        return new ContainsJSONKey<Object>(key);
    }

    @Factory
    public static <T> Matcher<JSONObject> hasKey(String key, T value) {
        return new ContainsJSONKey<T>(key, value);
    }

    private String key;
    private T expectedValue;
    private boolean isKeyPresent;
    private boolean run = false;
    private boolean success;

    @Override
    public void describeTo(Description description) {
        if (!run || success)
            return;
        if (!isKeyPresent) {
            description.appendText("Key not present: " + key);
        } else {
            description.appendText("Value of " + key + " to be " + expectedValue);
        }
    }

    @Override
    public boolean matchesSafely(JSONObject item) {
        run = true;
        success = runMatch(item);
        return success;
    }

    private boolean runMatch(JSONObject item) {
        isKeyPresent = item.containsKey(key);
        if (!isKeyPresent) {
            return false;
        }
        if (expectedValue == null) {
            return true;
        }

        T actualValue = (T) item.get(key);

        if (actualValue == null) {
            return (expectedValue == null);
        }
        return actualValue.equals(expectedValue);
    }

    ContainsJSONKey(String key) {
        this.key = key;
    }

    ContainsJSONKey(String key, T value) {
        this.key = key;
        this.expectedValue = value;
    }
}