import graphlib.edges.DirectedEdge;
import graphlib.graphs.DirectedGraph;
import graphlib.util.Graphs;
import graphlib.nodes.Node;

import java.util.Arrays;
import java.util.Comparator;
import java.util.List;

import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.BeforeAll;

public class SimpleTest {

    private static TopologischSorteren ts;


    @BeforeAll
    public static void init() {
        ts = new MijnTopologischSorteren();
    }

    @Test
    public void test1() {

        DirectedGraph<String> graph = new DirectedGraph<>();

        graph.addNode(new Node<>("n"));

        List<Node<String>> topolOrdStud = ts.bepaalTopologischeOrdening(Graphs.unmodifiableDirectedGraph(graph));

        //Bij deze graaf is de oplossing uniek!
        List<Node<String>> expectedSolution = Arrays.asList(
                graph.getNode("n")
        );

        Assertions.assertEquals(expectedSolution, topolOrdStud);

    }

    @Test
    public void test2() {

        DirectedGraph<String> graph = new DirectedGraph<>();

        for(int i = 0; i < 3; i++){
            graph.addNode(new Node<>("n" + i));
        }

        voegBoogToe(graph,"n1","n0");
        voegBoogToe(graph,"n1","n2");
        voegBoogToe(graph,"n2","n0");

        List<Node<String>> topolOrdStud = ts.bepaalTopologischeOrdening(Graphs.unmodifiableDirectedGraph(graph));

        //Bij deze graaf is de oplossing uniek!
        List<Node<String>> expectedSolution = Arrays.asList(
                graph.getNode("n1"),
                graph.getNode("n2"),
                graph.getNode("n0")
        );

        Assertions.assertEquals(expectedSolution, topolOrdStud);

    }

    @Test
    public void test3() {
        Assertions.assertThrows(IllegalArgumentException.class, () -> {

        DirectedGraph<String> graph = new DirectedGraph<>();

        for(int i = 0; i < 3; i++){
            graph.addNode(new Node<>("n" + i));
        }

        voegBoogToe(graph,"n1","n0");
        voegBoogToe(graph,"n0","n1");
        voegBoogToe(graph,"n1","n2");
        voegBoogToe(graph,"n2","n0");

        List<Node<String>> topolOrdStud = ts.bepaalTopologischeOrdening(Graphs.unmodifiableDirectedGraph(graph));

    });
    }

    @Test
    public void test4() {

        DirectedGraph<String> graph = new DirectedGraph<>();

        for(int i = 0; i < 4; i++){
            graph.addNode(new Node<>("n" + i));
        }

        voegBoogToe(graph,"n1","n0");
        voegBoogToe(graph,"n2","n0");
        voegBoogToe(graph,"n2","n1");
        voegBoogToe(graph,"n3","n0");
        voegBoogToe(graph,"n3","n1");
        voegBoogToe(graph,"n3","n2");

        List<Node<String>> topolOrdStud = ts.bepaalTopologischeOrdening(Graphs.unmodifiableDirectedGraph(graph));

        //Bij deze graaf is de oplossing uniek!
        List<Node<String>> topolOrdOpl = Arrays.asList(
                graph.getNode("n3"),
                graph.getNode("n2"),
                graph.getNode("n1"),
                graph.getNode("n0")
        );

        Assertions.assertEquals(topolOrdOpl, topolOrdStud);

    }

    @Test
    public void test5() {

        DirectedGraph<String> graph = new DirectedGraph<>();

        for(int i = 0; i < 7; i++){
            graph.addNode(new Node<>("n" + i));
        }

        voegBoogToe(graph,"n0","n1");
        voegBoogToe(graph,"n1","n3");
        voegBoogToe(graph,"n1","n4");
        voegBoogToe(graph,"n2","n0");
        voegBoogToe(graph,"n2","n3");
        voegBoogToe(graph,"n2","n5");
        voegBoogToe(graph,"n3","n4");
        voegBoogToe(graph,"n3","n5");
        voegBoogToe(graph,"n3","n6");
        voegBoogToe(graph,"n4","n6");
        voegBoogToe(graph,"n6","n5");

        List<Node<String>> topolOrdStud = ts.bepaalTopologischeOrdening(Graphs.unmodifiableDirectedGraph(graph));

        //Bij deze graaf is de oplossing uniek!
        List<Node<String>> topolOrdOpl = Arrays.asList(
                graph.getNode("n2"),
                graph.getNode("n0"),
                graph.getNode("n1"),
                graph.getNode("n3"),
                graph.getNode("n4"),
                graph.getNode("n6"),
                graph.getNode("n5")
        );

        Assertions.assertEquals(topolOrdOpl, topolOrdStud);

    }

    @Test
    public void test6() {

        DirectedGraph<String> graph = new DirectedGraph<>();

        for(int i = 0; i < 7; i++){
            graph.addNode(new Node<>("n" + i));
        }

        voegBoogToe(graph,"n0","n1");
        voegBoogToe(graph,"n0","n2");
        voegBoogToe(graph,"n0","n4");
        voegBoogToe(graph,"n0","n6");
        voegBoogToe(graph,"n1","n5");
        voegBoogToe(graph,"n2","n3");
        voegBoogToe(graph,"n3","n4");
        voegBoogToe(graph,"n4","n1");
        voegBoogToe(graph,"n4","n6");
        voegBoogToe(graph,"n4","n5");
        voegBoogToe(graph,"n6","n1");

        List<Node<String>> topolOrdStud = ts.bepaalTopologischeOrdening(Graphs.unmodifiableDirectedGraph(graph));

        //Bij deze graaf is de oplossing uniek!
        List<Node<String>> topolOrdOpl = Arrays.asList(
                graph.getNode("n0"),
                graph.getNode("n2"),
                graph.getNode("n3"),
                graph.getNode("n4"),
                graph.getNode("n6"),
                graph.getNode("n1"),
                graph.getNode("n5")
        );

        Assertions.assertEquals(topolOrdOpl, topolOrdStud);

    }

    @Test
    public void test7() {
        Assertions.assertThrows(IllegalArgumentException.class, () -> {

        DirectedGraph<String> graph = new DirectedGraph<>();

        for(int i = 0; i < 5; i++){
            graph.addNode(new Node<>("n" + i));
        }

        voegBoogToe(graph,"n0","n1");
        voegBoogToe(graph,"n1","n2");
        voegBoogToe(graph, "n1", "n4");
        voegBoogToe(graph, "n4", "n1");
        voegBoogToe(graph,"n2","n3");

        List<Node<String>> topolOrdStud = ts.bepaalTopologischeOrdening(Graphs.unmodifiableDirectedGraph(graph));

    });
    }

    @Test
    public void test8() {
        DirectedGraph<String> graph = new DirectedGraph<>();

        for(int i = 0; i < 5; i++){
            graph.addNode(new Node<>("n" + i));
        }

        voegBoogToe(graph,"n0","n1");
        voegBoogToe(graph,"n2","n1");
        voegBoogToe(graph,"n2","n3");
        voegBoogToe(graph,"n4","n3");

        List<Node<String>> topolOrdStud = ts.bepaalTopologischeOrdening(Graphs.unmodifiableDirectedGraph(graph));

        List<Node<String>> level0 = Arrays.asList(
                graph.getNode("n0"),
                graph.getNode("n2"),
                graph.getNode("n4")
        );

        List<Node<String>> level1 = Arrays.asList(
                graph.getNode("n1"),
                graph.getNode("n3")
        );

        Assertions.assertEquals(level0, topolOrdStud.subList(0, 3).stream().sorted(Comparator.comparing(Node::key)).toList());
        Assertions.assertEquals(level1, topolOrdStud.subList(3, 5).stream().sorted(Comparator.comparing(Node::key)).toList());
    }

    public void voegBoogToe(DirectedGraph<String> graph, String n1, String n2){
        graph.addEdge(new DirectedEdge<>(graph.getNode(n1), graph.getNode(n2)));
    }

}