package fr.inria.diversify.ut.json.input;

import fr.inria.diversify.diversification.InputProgram;
import fr.inria.diversify.persistence.json.input.JsonAstAddInput;
import fr.inria.diversify.persistence.json.input.JsonAstReplaceInput;
import fr.inria.diversify.transformation.Transformation;
import fr.inria.diversify.transformation.ast.ASTAdd;
import fr.inria.diversify.ut.MockInputProgram;
import fr.inria.diversify.ut.json.SectionTestUtils;
import org.json.JSONException;
import org.json.JSONObject;
import org.junit.Ignore;
import org.junit.Test;

import java.util.HashMap;
import java.util.List;
import java.util.UUID;

import static fr.inria.diversify.persistence.json.output.JsonSectionOutput.*;
import static fr.inria.diversify.ut.json.SectionTestUtils.TEST_ID_1;
import static fr.inria.diversify.ut.json.SectionTestUtils.createAddASTTransformationJSON;
import static fr.inria.diversify.ut.json.SectionTestUtils.createReplaceASTTransformationJSON;
import static fr.inria.diversify.ut.json.input.JsonAstReplaceInputTest.testErrors;
import static org.junit.Assert.*;

/**
 * Created by marodrig on 12/01/2015.
 */
public class JsonAstAddInputTest {

    /**
     * Test that something goes wrong and it reports it right
     */
    @Test
    public void testWithErrors_UnableToFindCodeFragment() throws JSONException {
        String error1 = "ERROR  : Transf " + TEST_ID_1 +". Unable to find code fragment \"return 0\" at \"org.MyClass:200\". ";
        JSONObject o = createAddASTTransformationJSON().getJSONArray(TRANSFORMATIONS).getJSONObject(0);
        o.getJSONObject(TRANSPLANT_POINT).put(POSITION, "org.MyClass:200");
        assertEquals(error1, testErrors(new JsonAstAddInput(null, o), o, 1, 0).get(0));

        error1 = "ERROR  : Transf " + TEST_ID_1 +". Unable to find code fragment \"fullyDifferent()\" at \"org.MyOtherClass:10\". ";
        o = createAddASTTransformationJSON().getJSONArray(TRANSFORMATIONS).getJSONObject(0);
        o.getJSONObject(TRANSPLANT).put(SOURCE_CODE, "fullyDifferent()");
        assertEquals(error1, testErrors(new JsonAstAddInput(null, o), o, 1, 0).get(0));
    }

    @Test
    public void testWithErrors_MismatchCF() throws JSONException {

        String error1 = "WARNING: Transf " + TEST_ID_1 +". Position mismatch -> Storage: " +
                "\"org.MyClass:2\"; Found: \"org.MyClass:1\". ";
        String error2 = "WARNING: Transf " + TEST_ID_1 +". Source mismatch -> Storage: \"return   0\"; Found: \"return 0\". ";

        //Insert ann error
        JSONObject o = createAddASTTransformationJSON().getJSONArray(TRANSFORMATIONS).getJSONObject(0);
        o.getJSONObject(TRANSPLANT_POINT).put(POSITION, "org.MyClass:2");
        //Test that the error was raised
        assertEquals(error1,testErrors(new JsonAstAddInput(null, o), o, 1, 1).get(0));

        o = createAddASTTransformationJSON().getJSONArray(TRANSFORMATIONS).getJSONObject(0);
        o.getJSONObject(TRANSPLANT_POINT).put(SOURCE_CODE, "return   0");
        assertEquals(error2, testErrors(new JsonAstAddInput(null, o), o, 1, 1).get(0));

        o = createAddASTTransformationJSON().getJSONArray(TRANSFORMATIONS).getJSONObject(0);
        o.getJSONObject(TRANSPLANT).put(SOURCE_CODE, "return   0");
        o.getJSONObject(TRANSPLANT).put(POSITION, "org.MyClass:2");
        List<String> errors = testErrors(new JsonAstAddInput(null, o), o, 2, 1);
        assertEquals(error1, errors.get(0));
        assertEquals(error2, errors.get(1));
    }
    /**
     * Test the reading of a transformation
     */
    @Test
    public void testReadAddTransformation() throws JSONException {
        InputProgram p = new MockInputProgram();
        //Create a ASTAdd persisted as json
        JSONObject o = createAddASTTransformationJSON().getJSONArray(TRANSFORMATIONS).getJSONObject(0);
        JsonAstAddInput reader = new JsonAstAddInput(p, o);
        reader.setInputProgram(p);
        HashMap<UUID, Transformation> result = new HashMap<>();
        reader.read(result);

        ASTAdd add = (ASTAdd)result.get(TEST_ID_1);
        assertEquals(TEST_ID_1, add.getIndex());
        assertEquals(-1, add.getStatus());
        assertEquals(1, result.size());
        assertEquals(p.getCodeFragments().get(0), add.getTransplantationPoint());
        assertEquals(p.getCodeFragments().get(1), add.getTransplant());
    }

    @Test
    public void testCanHandleSection() {
        JsonAstAddInput reader = new JsonAstAddInput(new InputProgram(), new JSONObject());
        assertTrue(reader.canRead(TRANSFORMATIONS + ".add"));
        assertTrue(reader.canRead(TRANSFORMATIONS + ".addWitgestein"));
        assertFalse(reader.canRead(TRANSFORMATIONS + ".delete"));
        assertFalse(reader.canRead(TRANSFORMATIONS + ".replace"));
        assertFalse(reader.canRead(TRANSFORMATIONS + ".replaceWitgestein"));
    }

}