chrome/test/android/javatests/src/org/chromium/chrome/test/ChromeTabbedActivityTestRule.java

// 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.test;

import org.junit.runner.Description;
import org.junit.runners.model.Statement;

import org.chromium.base.test.util.UrlUtils;
import org.chromium.chrome.browser.ChromeTabbedActivity;
import org.chromium.chrome.browser.feed.FeedNewTabPage;
import org.chromium.chrome.browser.feed.FeedProcessScopeFactory;
import org.chromium.chrome.browser.feed.TestNetworkClient;

/**
 * Custom ActivityTestRule for test using ChromeTabbedActivity
 */
public class ChromeTabbedActivityTestRule extends ChromeActivityTestRule<ChromeTabbedActivity> {
    // Response file for Feed's network stub.
    // TODO(crbug.com/943874): Make this configurable.
    public static final String FEED_TEST_RESPONSE_FILE_PATH =
            "/chrome/test/data/android/feed/feed_large.gcl.bin";

    public ChromeTabbedActivityTestRule() {
        super(ChromeTabbedActivity.class);
    }

    @Override
    public Statement apply(final Statement base, Description description) {
        Statement tabbedActivityStatement = new Statement() {
            @Override
            public void evaluate() throws Throwable {
                // Setup Feed stubs if Feed is enabled.
                boolean feedStubsInitialized = false;
                if (!FeedNewTabPage.isDummy()) {
                    feedStubsInitialized = true;
                    TestNetworkClient client = new TestNetworkClient();
                    client.setNetworkResponseFile(
                            UrlUtils.getIsolatedTestFilePath(FEED_TEST_RESPONSE_FILE_PATH));
                    FeedProcessScopeFactory.setTestNetworkClient(client);
                }

                base.evaluate();

                // Teardown the network stubs for Feed if they've been setup.
                if (feedStubsInitialized) FeedProcessScopeFactory.setTestNetworkClient(null);
            }
        };

        return super.apply(tabbedActivityStatement, description);
    }
}