// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.jdisc.handler;

import org.junit.Test;
import org.mockito.Mockito;

import java.nio.ByteBuffer;
import java.util.concurrent.Executor;
import java.util.concurrent.TimeUnit;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.fail;


/**
 * @author <a href="mailto:simon@yahoo-inc.com">Simon Thoresen Hult</a>
 */
public class FastContentOutputStreamTestCase {

    @Test
    public void requireThatNullConstructorArgumentThrows() {
        try {
            new FastContentOutputStream((ContentChannel)null);
            fail();
        } catch (NullPointerException e) {
            assertEquals("out", e.getMessage());
        }
        try {
            new FastContentOutputStream((FastContentWriter)null);
            fail();
        } catch (NullPointerException e) {
            assertEquals("out", e.getMessage());
        }
    }

    @Test
    public void requireThatAllMethodsDelegateToWriter() throws Exception {
        FastContentWriter writer = Mockito.mock(FastContentWriter.class);
        FastContentOutputStream out = new FastContentOutputStream(writer);

        out.write(new byte[] { 6, 9 });
        out.flush();
        Mockito.verify(writer).write(Mockito.any(ByteBuffer.class));

        out.close();
        Mockito.verify(writer).close();

        out.cancel(true);
        Mockito.verify(writer).cancel(true);
        out.cancel(false);
        Mockito.verify(writer).cancel(false);

        out.isCancelled();
        Mockito.verify(writer).isCancelled();

        out.isDone();
        Mockito.verify(writer).isDone();

        out.get();
        Mockito.verify(writer).get();

        out.get(600, TimeUnit.SECONDS);
        Mockito.verify(writer).get(600, TimeUnit.SECONDS);

        Runnable listener = Mockito.mock(Runnable.class);
        Executor executor = Mockito.mock(Executor.class);
        out.addListener(listener, executor);
        Mockito.verify(writer).addListener(listener, executor);
    }
}