package org.json.junit;

import static org.junit.Assert.*;

import java.util.*;

import org.json.*;

public class Util {

    
    ///////////////////////////  UTILITY METHODS /////////////////////////
    
    /**
     * Compares two json arrays for equality
     * @param jsonArray created by the code to be tested
     * @param expectedJsonArray created specifically for compar
     */
    public static void compareActualVsExpectedJsonArrays(JSONArray jsonArray,
            JSONArray expectedJsonArray) {
        assertTrue("jsonArray lengths should be equal",
                jsonArray.length() == expectedJsonArray.length());
        for (int i = 0; i < jsonArray.length(); ++i) {
            JSONObject jsonObject = jsonArray.getJSONObject(i);
            JSONObject expectedJsonObject = expectedJsonArray.getJSONObject(i);
            assertTrue("jsonObjects should have the same length",
                    jsonObject.length() == expectedJsonObject.length());
            Iterator<String> keys = jsonObject.keys();
            while (keys.hasNext()) {
                // TODO: check for nonstring types
                String key = keys.next();
                Object value = jsonObject.get(key);
                String testStr = "row: "+i+" key: "+key+" val: "+value.toString();
                String actualStr = expectedJsonObject .get(key).toString();
                assertTrue("values should be equal for actual: "+testStr+
                        " expected: "+actualStr,
                        value.equals(expectedJsonArray.getJSONObject(i).
                                get(key).toString()));
            }
        }
    }

    public static void compareActualVsExpectedJsonObjects(
            JSONObject jsonObject, JSONObject expectedJsonObject) {
        assertTrue("jsonObjects should have the same length",
                jsonObject.length() == expectedJsonObject.length());
        Iterator<String> keys = jsonObject.keys();
        while (keys.hasNext()) {
            String key = keys.next();
            Object value = jsonObject.get(key);
            Object expectedValue = expectedJsonObject.get(key);
            if (value instanceof JSONObject) {
                JSONObject childJsonObject = jsonObject.getJSONObject(key);
                JSONObject expectedChildJsonObject =
                        expectedJsonObject.getJSONObject(key);
                compareActualVsExpectedJsonObjects(
                        childJsonObject, expectedChildJsonObject);
            } else if (value instanceof JSONArray) {
                JSONArray childJsonArray = jsonObject.getJSONArray(key);
                JSONArray expectedChildJsonArray =
                        expectedJsonObject.getJSONArray(key);
                compareActualVsExpectedJsonArrays(
                        childJsonArray, expectedChildJsonArray);
            } else {
                String testStr = "key: "+key+" val: "+value.toString();
                String actualStr = expectedValue.toString();
                assertTrue("string values should be equal for actual: "+
                        testStr+" expected: "+actualStr,
                        value.equals(expectedValue.toString()));
            }
        }
    }
}