tests/AndroidTests/src/com/android/unit_tests/UrlRulesTest.java

/*
 * Copyright (C) 2008 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.android.unit_tests;

import android.content.ContentResolver;
import android.provider.Settings;
import android.test.AndroidTestCase;
import android.test.suitebuilder.annotation.MediumTest;
import android.test.suitebuilder.annotation.SmallTest;

import com.google.android.net.UrlRules;
import static com.google.android.net.UrlRules.Rule;

/** Test loading and matching URL rewrite rules for UrlRules.  */
public class UrlRulesTest extends AndroidTestCase {
    @SmallTest
    public void testEmptyRules() {
        UrlRules rules = new UrlRules(new Rule[] { });
        assertTrue(rules.matchRule("http://foo.bar/") == Rule.DEFAULT);
    }

    @SmallTest
    public void testInvalidRule() throws Exception {
        try {
            new Rule("rule", "foo bar");
        } catch (Exception e) {
            // Re-throw any exception except the one we're looking for.
            if (!e.toString().contains("Illegal rule: foo bar")) throw e;
        }
    }

    @SmallTest
    public void testRewriteRule() throws UrlRules.RuleFormatException {
        Rule rule = new Rule("test_rule",
                "http://foo.bar/ rewrite http://bar.foo/");
        assertEquals("test_rule", rule.mName);
        assertEquals("http://foo.bar/", rule.mPrefix);
        assertEquals("http://bar.foo/", rule.mRewrite);
        assertFalse(rule.mBlock);
        assertEquals("http://bar.foo/bat", rule.apply("http://foo.bar/bat"));
    }

    @SmallTest
    public void testBlockRule() throws UrlRules.RuleFormatException {
        Rule rule = new Rule("test_rule",
                "http://foo.bar/ block");
        assertEquals("test_rule", rule.mName);
        assertEquals("http://foo.bar/", rule.mPrefix);
        assertTrue(rule.mRewrite == null);
        assertTrue(rule.mBlock);
        assertTrue(rule.apply("http://foo.bar/bat") == null);
    }

    @SmallTest
    public void testMatchRule() throws UrlRules.RuleFormatException {
        UrlRules rules = new UrlRules(new Rule[] {
            new Rule("12", "http://one.two/ rewrite http://buckle.my.shoe/"),
            new Rule("34", "http://three.four/ rewrite http://close.the.door/"),
            new Rule("56", "http://five.six/ rewrite http://pick.up.sticks/"),
        });

        assertTrue(rules.matchRule("https://one.two/") == Rule.DEFAULT);
        assertTrue(rules.matchRule("http://one.two") == Rule.DEFAULT);
        assertEquals("12", rules.matchRule("http://one.two/foo").mName);

        String u = "http://five.six/bar";
        assertEquals("http://pick.up.sticks/bar", rules.matchRule(u).apply(u));
    }

    @SmallTest
    public void testAmbiguousMatch() throws UrlRules.RuleFormatException {
        // Rule is the longest match wins.
        UrlRules rules = new UrlRules(new Rule[] {
            new Rule("1", "http://xyz/one rewrite http://rewrite/"),
            new Rule("123", "http://xyz/onetwothree rewrite http://rewrite/"),
            new Rule("12", "http://xyz/onetwo rewrite http://rewrite/"),
        });

        assertEquals("1", rules.matchRule("http://xyz/one").mName);
        assertEquals("1", rules.matchRule("http://xyz/one...").mName);
        assertEquals("12", rules.matchRule("http://xyz/onetwo...").mName);
        assertEquals("123", rules.matchRule("http://xyz/onetwothree...").mName);

    }

    @MediumTest
    public void testGservicesRules() {
        // TODO: use a MockContentProvider/MockContentResolver instead.
        ContentResolver r = getContext().getContentResolver();

        // Update the digest, so the UrlRules cache is reloaded.
        Settings.Gservices.putString(r, "digest", "testGservicesRules");
        Settings.Gservices.putString(r, "url:blank_test", "");
        Settings.Gservices.putString(r, "url:test",
                "http://foo.bar/ rewrite http://bar.foo/");

        UrlRules rules = UrlRules.getRules(r);  // Don't crash, please.  :)
        assertTrue(rules.matchRule("http://bar.foo/") == Rule.DEFAULT);

        Rule rule = rules.matchRule("http://foo.bar/bat");
        assertEquals("test", rule.mName);
        assertEquals("http://foo.bar/", rule.mPrefix);
        assertEquals("http://bar.foo/", rule.mRewrite);
        assertFalse(rule.mBlock);
    }
}