com.jayway.restassured.assertion.JSONAssertion.groovy Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of rest-assured Show documentation

Java DSL for easy testing of REST services

There is a newer version: 2.9.0

/*
 * Copyright 2010 the original author or authors.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.jayway.restassured.assertion

import net.sf.json.JSONArray
import net.sf.json.JSONNull
import static com.jayway.restassured.assertion.AssertionSupport.escapeMinus
import static com.jayway.restassured.assertion.AssertionSupport.generateWhitespace
import static java.util.Arrays.asList

class JSONAssertion implements Assertion {
  String key;

  def Object getResult(Object object) {
    key = escapeMinus(key);
    def result;
    def root = 'restAssuredJsonRootObject'
    try {
      result = Eval.me(root, object, "$root.$key")
    } catch (Exception e) {
      throw new IllegalArgumentException(e.getMessage().replace("startup failed:", "Invalid JSON expression:").replace("$root.", generateWhitespace(root.length())));
    }

    return convertToJavaListIfNeeded(result);
  }

  private Object convertToJavaListIfNeeded(current) {
    if (current instanceof JSONArray) {
      current = asList(current.toArray());
    }
    if(current instanceof List) {
      for(int i = 0; i < current.size(); i++) {
        if(current.get(i) instanceof JSONNull) {
          current.set(i, null);
        }
      }
    }
    return current


  }

  def String description() {
    return "JSON element"
  }
}