// Copyright 2017 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package org.chromium.chrome.browser.media.router.cast;

import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.ArgumentMatchers.same;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.timeout;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

import android.support.v7.media.MediaRouter;

import org.junit.Test;
import org.junit.runner.RunWith;
import org.robolectric.annotation.Config;

import org.chromium.base.test.BaseRobolectricTestRunner;
import org.chromium.base.test.util.Feature;
import org.chromium.chrome.browser.media.router.ChromeMediaRouter;
import org.chromium.chrome.browser.media.router.MediaRoute;
import org.chromium.chrome.browser.media.router.MediaRouteManager;
import org.chromium.chrome.browser.media.router.MediaSink;

import java.util.ArrayList;

/**
 * Robolectric tests for {@link CastMediaRouteProvider}.
 */
@RunWith(BaseRobolectricTestRunner.class)
@Config(manifest = Config.NONE)
public class CastMediaRouteProviderTest {
    private static final String SUPPORTED_SOURCE = "cast:DEADBEEF";

    private static final String SUPPORTED_AUTOJOIN_SOURCE = "cast:DEADBEEF"
            + "?clientId=12345&autoJoinPolicy=" + CastMediaSource.AUTOJOIN_TAB_AND_ORIGIN_SCOPED;

    // TODO(crbug.com/672704): Android does not currently support 1-UA mode.
    private static final String UNSUPPORTED_SOURCE = "https://example.com";

    private MediaRouteManager mMockManager;
    private CastMediaRouteProvider mProvider;

    protected void setUpMediaRouter(MediaRouter router) {
        ChromeMediaRouter.setAndroidMediaRouterForTest(router);
        mMockManager = mock(MediaRouteManager.class);
        mProvider = CastMediaRouteProvider.create(mMockManager);
    }

    @Test
    @Feature({"MediaRouter"})
    public void testStartObservingMediaSinksNoMediaRouter() {
        setUpMediaRouter(null);

        mProvider.startObservingMediaSinks(SUPPORTED_SOURCE);

        verify(mMockManager, timeout(100))
                .onSinksReceived(
                        eq(SUPPORTED_SOURCE), same(mProvider), eq(new ArrayList<MediaSink>()));
    }

    @Test
    @Feature({"MediaRouter"})
    public void testStartObservingMediaSinksUnsupportedSource() {
        setUpMediaRouter(mock(MediaRouter.class));

        mProvider.startObservingMediaSinks(UNSUPPORTED_SOURCE);

        verify(mMockManager, timeout(100))
                .onSinksReceived(
                        eq(UNSUPPORTED_SOURCE), same(mProvider), eq(new ArrayList<MediaSink>()));
    }

    @Test
    @Feature({"MediaRouter"})
    public void testOnSessionClosedNoClientRecord() {
        setUpMediaRouter(mock(MediaRouter.class));

        CastSession mockSession = mock(CastSession.class);
        mProvider.onSessionStarted(mockSession);

        MediaRoute route = new MediaRoute("sink", SUPPORTED_SOURCE, "");
        mProvider.addRoute(route, "", -1);
        mProvider.onSessionEnded();

        verify(mMockManager).onRouteTerminated(route.id);
    }

    @Test
    @Feature({"MediaRouter"})
    public void testCloseRouteWithNoSession() {
        setUpMediaRouter(mock(MediaRouter.class));

        MediaRoute route = new MediaRoute("sink", SUPPORTED_SOURCE, "");
        mProvider.addRoute(route, "", -1);
        mProvider.closeRoute(route.id);

        verify(mMockManager).onRouteTerminated(route.id);
    }

    @Test
    @Feature({"MediaRouter"})
    public void testAutoJoinWithSameOrigin() {
        setUpMediaRouter(mock(MediaRouter.class));

        CastSession mockSession = mock(CastSession.class);
        when(mockSession.getSinkId()).thenReturn("sinkId");
        when(mockSession.getSourceId()).thenReturn(SUPPORTED_AUTOJOIN_SOURCE);

        MediaRoute route = new MediaRoute("sinkId", SUPPORTED_AUTOJOIN_SOURCE, "presentationId");
        mProvider.addRoute(route, "https://example.com", 1);
        mProvider.onSessionStarted(mockSession);
        mProvider.joinRoute(SUPPORTED_AUTOJOIN_SOURCE, "auto-join", "https://example.com", 1, -1);

        verify(mMockManager)
                .onRouteCreated("route:auto-join/sinkId/" + SUPPORTED_AUTOJOIN_SOURCE, "sinkId", -1,
                        mProvider, false);
    }

    @Test
    @Feature({"MediaRouter"})
    public void testNoAutoJoinWithOneUniqueOrigin() {
        setUpMediaRouter(mock(MediaRouter.class));

        CastSession mockSession = mock(CastSession.class);
        when(mockSession.getSinkId()).thenReturn("sinkId");
        when(mockSession.getSourceId()).thenReturn(SUPPORTED_AUTOJOIN_SOURCE);

        MediaRoute route = new MediaRoute("sinkId", SUPPORTED_AUTOJOIN_SOURCE, "presentationId");
        mProvider.addRoute(route, "https://example.com", 1);
        mProvider.onSessionStarted(mockSession);
        mProvider.joinRoute(SUPPORTED_AUTOJOIN_SOURCE, "auto-join", "", 1, -1);

        verify(mMockManager).onRouteRequestError("No matching route", -1);
    }

    @Test
    @Feature({"MediaRouter"})
    public void testNoAutoJoinWithTwoUniqueOrigins() {
        setUpMediaRouter(mock(MediaRouter.class));

        CastSession mockSession = mock(CastSession.class);
        when(mockSession.getSinkId()).thenReturn("sinkId");
        when(mockSession.getSourceId()).thenReturn(SUPPORTED_AUTOJOIN_SOURCE);

        MediaRoute route = new MediaRoute("sinkId", SUPPORTED_AUTOJOIN_SOURCE, "presentationId");
        mProvider.addRoute(route, "", 1);
        mProvider.onSessionStarted(mockSession);
        mProvider.joinRoute(SUPPORTED_AUTOJOIN_SOURCE, "auto-join", "", 1, -1);

        verify(mMockManager).onRouteRequestError("No matching route", -1);
    }
}