package com.foo.bar;

import mockit.Expectations;
import mockit.Mocked;
import org.junit.Test;

import java.time.Duration;

import static org.hamcrest.Matchers.is;
import static org.hamcrest.core.IsEqual.equalTo;
import static org.junit.Assert.assertThat;

public class ConcerningDurationProvider {

    @Test
    public void shouldThrowIllegalArgumentException(@Mocked DurationProvider durationProvider) {
        DurationConsumer durationConsumer = new DurationConsumer(durationProvider);
        new Expectations() {{
            durationProvider.getDuration();
            result = Duration.ofSeconds(123);
        }};
        assertThat(durationConsumer.getDuration(), is(equalTo(Duration.ofSeconds(123))));
    }

    public interface DurationProvider {
        Duration getDuration();
    }

    public static class DurationConsumer {
        private final DurationProvider durationProvider;

        public DurationConsumer(DurationProvider durationProvider) {

            this.durationProvider = durationProvider;
        }

        public Duration getDuration() {
            return durationProvider.getDuration();
        }
    }


}