import tripadministratorjava.Company;
import tripadministratorjava.Transportation;
import tripadministratorjava.Trip;
import tripadministratorjava.TripAdministratorImpl;

import org.junit.Before;
import org.junit.Test;

import java.util.Arrays;

import static org.junit.Assert.*;

public class Test033 {

    private TripAdministratorImpl tripAdministrations;

    private Company c1 = new Company("a", 2);
    private Company c2 = new Company("b", 1);
    private Company c3 = new Company("c", 1);
    private Company c4 = new Company("d", 2);

    private Trip t1 = new Trip("a", 1, Transportation.NONE, 1);
    private Trip t2 = new Trip("b", 1, Transportation.BUS, 1);
    private Trip t3 = new Trip("c", 1, Transportation.BUS, 1);
    private Trip t4 = new Trip("d", 1, Transportation.BUS, 1);

    @Before
    public void Setup() {
        this.tripAdministrations = new TripAdministratorImpl();
    }

    @Test
    public void TestGetAllTripsInPriceRange() {
        this.tripAdministrations.addCompany(c1);
        this.tripAdministrations.addCompany(c2);
        this.tripAdministrations.addCompany(c3);
        this.tripAdministrations.addCompany(c4);
        this.tripAdministrations.addTrip(c1, t1);
        this.tripAdministrations.addTrip(c1, t2);
        this.tripAdministrations.addTrip(c4, t3);
        this.tripAdministrations.addTrip(c4, t4);

        var res = this.tripAdministrations.getAllTripsInPriceRange(1, 1);
        var expected = Arrays.asList(t1, t2, t3, t4);
        assertEquals(res, expected);
    }
}