components/web_restrictions/browser/javatest/src/org/chromium/components/webrestrictions/browser/MockWebRestrictionsClient.java

// Copyright 2015 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.components.webrestrictions.browser;

import android.database.AbstractCursor;

import org.chromium.base.annotations.CalledByNative;

/**
 * This is a dummy version of the real WebRestrictionsClient, to allow unit testing of the C++
 * code.
 */
public class MockWebRestrictionsClient extends WebRestrictionsClient {
    private String mAuthority;

    @CalledByNative
    static void registerAsMockForTesting() {
        WebRestrictionsClient.registerMockForTesting(new MockWebRestrictionsClient());
    }

    @CalledByNative
    static void unregisterAsMockForTesting() {
        WebRestrictionsClient.registerMockForTesting(null);
    }
    /**
     * Start the web restriction provider and setup the content resolver.
     */
    private MockWebRestrictionsClient() {}

    @Override
    void init(String authority, long nativeProvider) {
        mAuthority = authority;
    }

    /**
     * @return whether the web restriction provider supports requesting access for a blocked url.
     */
    @Override
    boolean supportsRequest() {
        return mAuthority.contains("Good");
    }

    /**
     * Called when the ContentResolverWebRestrictionsProvider is about to be destroyed.
     */
    @Override
    void onDestroy() {}

    /**
     * Whether we can proceed with loading the {@code url}.
     * In case the url is not to be loaded, the web restriction provider can return an optional
     * error page to show instead.
     */
    @Override
    WebRestrictionsClientResult shouldProceed(final String url) {
        return new WebRestrictionsClientResult(new AbstractCursor() {

            @Override
            public int getCount() {
                return 0;
            }

            @Override
            public String[] getColumnNames() {
                return new String[] {"Result", "Error number", "Error string"};
            }

            @Override
            public int getColumnCount() {
                return mAuthority.contains("Good") ? 1 : 3;
            }

            @Override
            public String getString(int column) {
                if (column == 2 && !mAuthority.contains("Good")) return url;
                return null;
            }

            @Override
            public short getShort(int column) {
                return 0;
            }

            @Override
            public int getInt(int column) {
                if (column == 0 && mAuthority.contains("Good")) return 1;
                if (column == 1 && !mAuthority.contains("Good")) return 42;
                return 0;
            }

            @Override
            public long getLong(int column) {
                return 0;
            }

            @Override
            public float getFloat(int column) {
                return 0;
            }

            @Override
            public double getDouble(int column) {
                return 0;
            }

            @Override
            public boolean isNull(int column) {
                return false;
            }

        });
    }

    /**
     * Request permission to load the {@code url}.
     * The web restriction provider returns a {@code boolean} variable indicating whether it was
     * able to forward the request to the appropriate authority who can approve it.
     */
    @Override
    boolean requestPermission(final String url) {
        return mAuthority.contains("Good");
    }
}