import org.junit.Assert;
import org.junit.Test;

import java.util.*;

public class SimpleTest {

    private Scheduler scheduler = new FlightScheduler();

    @Test
    public void test1() {

        Flight vlucht = new Flight(100.0, 10);
        Airplane vliegtuig = new Airplane(200.0, 20);

        List<Airplane> planning = scheduler.schedule(Collections.singletonList(vlucht), Collections.singleton(vliegtuig));

        Assert.assertNotNull(planning);
        Assert.assertEquals(1, planning.size());
        Assert.assertSame(vliegtuig, planning.get(0));

    }

    @Test
    public void test2() {

        Flight vlucht = new Flight(100.0, 10);
        Airplane vliegtuig = new Airplane(50.0, 20);

        List<Airplane> planning = scheduler.schedule(Collections.singletonList(vlucht), Collections.singleton(vliegtuig));

        Assert.assertNull(planning);

    }

    @Test
    public void test3() {

        Flight vlucht = new Flight(100.0, 10);
        Airplane vliegtuig = new Airplane(200.0, 5);

        List<Airplane> planning = scheduler.schedule(Collections.singletonList(vlucht), Collections.singleton(vliegtuig));

        Assert.assertNull(planning);

    }

    @Test
    public void test4() {

        List<Flight> vluchten = Arrays.asList(
                new Flight(100.0, 10),
                new Flight(50.0, 50),
                new Flight(150.0, 5)
        );

        Set<Airplane> vliegtuigen = new HashSet<>();
        Airplane vlieg1 = new Airplane(180.0, 10);
        Airplane vlieg2 = new Airplane(120.0, 5);
        Airplane vlieg3 = new Airplane(25.0, 100);
        Airplane vlieg4 = new Airplane(100.0, 100);
        Airplane vlieg5 = new Airplane(125.0, 10);
        vliegtuigen.addAll(Arrays.asList(vlieg1, vlieg2, vlieg3, vlieg4, vlieg5));

        List<Airplane> planning = scheduler.schedule(vluchten, vliegtuigen);

        Assert.assertNotNull(planning);
        Assert.assertEquals(3, planning.size());
        Assert.assertSame(vlieg5, planning.get(0));
        Assert.assertSame(vlieg4, planning.get(1));
        Assert.assertSame(vlieg1, planning.get(2));

    }

}