1 /* 2 * Copyright (C) 2010 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package com.android.email; 18 19 import android.content.Context; 20 import android.os.Bundle; 21 import android.test.AndroidTestCase; 22 import android.test.suitebuilder.annotation.SmallTest; 23 24 import com.android.emailcommon.VendorPolicyLoader; 25 import com.android.emailcommon.VendorPolicyLoader.Provider; 26 27 @SmallTest 28 public class VendorPolicyLoaderTest extends AndroidTestCase { 29 private String mTestApkPackageName; 30 31 @Override setUp()32 protected void setUp() throws Exception { 33 super.setUp(); 34 mTestApkPackageName = getContext().getPackageName() + ".tests"; 35 } 36 37 @Override tearDown()38 protected void tearDown() throws Exception { 39 super.tearDown(); 40 VendorPolicyLoader.clearInstanceForTest(); 41 } 42 43 /** 44 * Test for the case where the helper package doesn't exist. 45 */ testPackageNotExist()46 public void testPackageNotExist() { 47 VendorPolicyLoader pl = new VendorPolicyLoader(getContext(), "no.such.package", 48 "no.such.Class", true); 49 50 // getPolicy() shouldn't throw any exception. 51 assertEquals(Bundle.EMPTY, pl.getPolicy(null, null)); 52 } 53 testIsSystemPackage()54 public void testIsSystemPackage() { 55 final Context c = getContext(); 56 assertEquals(false, VendorPolicyLoader.isSystemPackage(c, "no.such.package")); 57 assertEquals(false, VendorPolicyLoader.isSystemPackage(c, mTestApkPackageName)); 58 assertEquals(true, VendorPolicyLoader.isSystemPackage(c, "com.android.settings")); 59 } 60 61 /** 62 * Actually call {@link VendorPolicyLoader#getPolicy}, using MockVendorPolicy as a vendor 63 * policy. 64 */ testGetPolicy()65 public void testGetPolicy() { 66 MockVendorPolicy.inject(getContext()); 67 VendorPolicyLoader pl = VendorPolicyLoader.getInstance(getContext()); 68 69 // Prepare result 70 Bundle result = new Bundle(); 71 result.putInt("ret", 1); 72 MockVendorPolicy.mockResult = result; 73 74 // Arg to pass 75 Bundle args = new Bundle(); 76 args.putString("arg1", "a"); 77 78 // Call! 79 Bundle actualResult = pl.getPolicy("policy1", args); 80 81 // Check passed args 82 assertEquals("policy", "policy1", MockVendorPolicy.passedPolicy); 83 assertEquals("arg", "a", MockVendorPolicy.passedBundle.getString("arg1")); 84 85 // Check return value 86 assertEquals("result", 1, actualResult.getInt("ret")); 87 } 88 89 /** 90 * Same as {@link #testGetPolicy}, but with the system-apk check. It's a test for the case 91 * where we have a non-system vendor policy installed, which shouldn't be used. 92 */ testGetPolicyNonSystem()93 public void testGetPolicyNonSystem() { 94 VendorPolicyLoader pl = new VendorPolicyLoader(getContext(), mTestApkPackageName, 95 MockVendorPolicy.class.getName(), false); 96 97 MockVendorPolicy.passedPolicy = null; 98 99 // getPolicy() shouldn't throw any exception. 100 assertEquals(Bundle.EMPTY, pl.getPolicy("policy1", null)); 101 102 // MockVendorPolicy.getPolicy() shouldn't get called. 103 assertNull(MockVendorPolicy.passedPolicy); 104 } 105 106 /** 107 * Test that any vendor policy that happens to be installed returns legal values 108 * for getImapIdValues() per its API. 109 * 110 * Note, in most cases very little will happen in this test, because there is 111 * no vendor policy package. Most of this test exists to test a vendor policy 112 * package itself, to make sure that its API returns reasonable values. 113 */ testGetImapIdValues()114 public void testGetImapIdValues() { 115 VendorPolicyLoader pl = VendorPolicyLoader.getInstance(getContext()); 116 String id = pl.getImapIdValues("user-name", "server.yahoo.com", 117 "IMAP4rev1 STARTTLS AUTH=GSSAPI"); 118 // null is a reasonable result 119 if (id == null) return; 120 121 // if non-null, basic sanity checks on format 122 assertEquals("\"", id.charAt(0)); 123 assertEquals("\"", id.charAt(id.length()-1)); 124 // see if we can break it up properly 125 String[] elements = id.split("\""); 126 assertEquals(0, elements.length % 4); 127 for (int i = 0; i < elements.length; ) { 128 // Because we split at quotes, we expect to find: 129 // [i] = null or one or more spaces 130 // [i+1] = key 131 // [i+2] = one or more spaces 132 // [i+3] = value 133 // Here are some incomplete checks of the above 134 assertTrue(elements[i] == null || elements[i].startsWith(" ")); 135 assertTrue(elements[i+1].charAt(0) != ' '); 136 assertTrue(elements[i+2].startsWith(" ")); 137 assertTrue(elements[i+3].charAt(0) != ' '); 138 i += 4; 139 } 140 } 141 142 /** 143 * Test that findProviderForDomain() returns legal values, or functions properly when 144 * none is installed. 145 */ testFindProviderForDomain()146 public void testFindProviderForDomain() { 147 VendorPolicyLoader pl = VendorPolicyLoader.getInstance(getContext()); 148 Provider p = pl.findProviderForDomain("yahoo.com"); 149 // null is a reasonable result (none installed) 150 if (p == null) return; 151 152 // if non-null, basic sanity checks on format 153 assertNull(p.id); 154 assertNull(p.label); 155 assertEquals("yahoo.com", p.domain); 156 assertNotNull(p.incomingUriTemplate); 157 assertNotNull(p.incomingUsernameTemplate); 158 assertNotNull(p.outgoingUriTemplate); 159 assertNotNull(p.outgoingUsernameTemplate); 160 assertTrue(p.note == null || p.note.length() > 0); // no empty string 161 } 162 } 163