import static org.junit.jupiter.api.Assertions.assertEquals;

import org.junit.jupiter.api.Test;

public class OneTest
{
    @Test
    public void testPartA()
    {
        DogWalkCompany company = new DogWalkCompany();
        company.availableDogs = new int[24];
        company.availableDogs[5] = 10;
        company.availableDogs[6] = 3;

        DogWalker walker = new DogWalker(4, company);

        assertEquals(4, walker.walkDogs(5));
        assertEquals(6, company.availableDogs[5]);

        assertEquals(3, walker.walkDogs(6));
        assertEquals(0, company.availableDogs[6]);
    }

    @Test
    public void testPartB()
    {
        DogWalkCompany company = new DogWalkCompany();
        company.availableDogs = new int[24];
        company.availableDogs[7] = 3;
        company.availableDogs[8] = 2;
        company.availableDogs[9] = 2;
        company.availableDogs[10] = 10;

        DogWalker walker = new DogWalker(3, company);
        assertEquals(59, walker.dogWalkShift(7, 10));
    }
}