package io.trygvis.persistence;

import io.trygvis.container.compiler.SqlUnitModel;
import io.trygvis.container.compiler.model.TypeRef;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;

import static io.trygvis.persistence.FieldMirror.PrimitiveFieldMirror;
import static java.lang.String.format;
import static org.testng.Assert.assertEquals;

public class EntityMirrorTest {

    static TypeRef entityType = new TypeRef("Wat");
    static FieldMirror idLong = new PrimitiveFieldMirror(new TypeRef(Long.class), "id", "id", true, false, true);
    static FieldMirror idString = new PrimitiveFieldMirror(new TypeRef(String.class), "id", "id", true, false, true);
    static FieldMirror name = new PrimitiveFieldMirror(new TypeRef(String.class), "name", "name", false, false, false);
    static FieldMirror ref = new FieldMirror.ReferenceFieldMirror(entityType, "parent", "parent", false, false);

    @DataProvider(name = "insertIntoSql", parallel = true)
    public static Object[][] insertIntoProvider() {
        return new Object[][]{
                new Object[]{
                        new FieldMirror[]{idString},
                        "INSERT INTO my_table(id) VALUES(?);",
                        "DELETE FROM my_table WHERE id=?;", format(
                        "CREATE TABLE my_table(%n" +
                                "  id VARCHAR(1000) PRIMARY KEY%n" +
                                ");")},
                new Object[]{
                        new FieldMirror[]{idLong},
                        "INSERT INTO my_table(id) VALUES(?);",
                        "DELETE FROM my_table WHERE id=?;", format(
                        "CREATE TABLE my_table(%n" +
                                "  id BIGINT PRIMARY KEY%n" +
                                ");")},
                new Object[]{
                        new FieldMirror[]{idLong, name},
                        "INSERT INTO my_table(id, name) VALUES(?, ?);",
                        "DELETE FROM my_table WHERE id=?;", format(
                        "CREATE TABLE my_table(%n" +
                                "  id BIGINT PRIMARY KEY,%n" +
                                "  name VARCHAR(1000)%n" +
                                ");")},
                new Object[]{
                        new FieldMirror[]{idLong, ref},
                        "INSERT INTO my_table(id, parent) VALUES(?, ?);",
                        "DELETE FROM my_table WHERE id=?;", format(
                        "CREATE TABLE my_table(%n" +
                                "  id BIGINT PRIMARY KEY,%n" +
                                "  parent BIGINT REFERENCES my_table(id)%n" +
                                ");")},
        };
    }

    @Test(dataProvider = "insertIntoSql")
    public void testInsertIntoSql(FieldMirror[] fields, String insert, String delete, String create) throws Exception {
        EntityMirror myTable = new EntityMirror(new GeneratorConfiguration(), entityType, "my_table");
        for (FieldMirror field : fields) {
            myTable.add(field);
        }

        SqlUnitModel unit = new SqlUnitModel();
        unit.add(myTable);

        assertEquals(myTable.insertIntoSql(), insert);
        assertEquals(myTable.deleteFromSql(), delete);
        assertEquals(myTable.createTableSql(unit), create);
    }
}