1/* 2 * Copyright (C) 2018 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 17package com.android.server.devicepolicy; 18 19import static com.android.server.devicepolicy.TransferOwnershipMetadataManager 20 .ADMIN_TYPE_DEVICE_OWNER; 21import static com.android.server.devicepolicy.TransferOwnershipMetadataManager 22 .OWNER_TRANSFER_METADATA_XML; 23import static com.android.server.devicepolicy.TransferOwnershipMetadataManager.TAG_ADMIN_TYPE; 24import static com.android.server.devicepolicy.TransferOwnershipMetadataManager.TAG_SOURCE_COMPONENT; 25import static com.android.server.devicepolicy.TransferOwnershipMetadataManager.TAG_TARGET_COMPONENT; 26import static com.android.server.devicepolicy.TransferOwnershipMetadataManager.TAG_USER_ID; 27 28import static org.junit.Assert.assertEquals; 29import static org.junit.Assert.assertFalse; 30import static org.junit.Assert.assertTrue; 31 32import android.os.Environment; 33import android.support.test.runner.AndroidJUnit4; 34import android.util.Log; 35 36import com.android.server.devicepolicy.TransferOwnershipMetadataManager.Injector; 37import com.android.server.devicepolicy.TransferOwnershipMetadataManager.Metadata; 38 39import org.junit.After; 40import org.junit.Before; 41import org.junit.Test; 42import org.junit.runner.RunWith; 43 44import java.io.File; 45import java.io.IOException; 46import java.nio.charset.Charset; 47import java.nio.file.Files; 48import java.nio.file.Path; 49import java.nio.file.Paths; 50 51/** 52* Unit tests for {@link TransferOwnershipMetadataManager}. 53 * 54 * bit FrameworksServicesTests:com.android.server.devicepolicy.TransferOwnershipMetadataManagerTest 55 * runtest -x frameworks/base/services/tests/servicestests/src/com/android/server/devicepolicy/TransferOwnershipMetadataManagerTest.java 56* */ 57 58@RunWith(AndroidJUnit4.class) 59public class TransferOwnershipMetadataManagerTest { 60 private final static String TAG = TransferOwnershipMetadataManagerTest.class.getName(); 61 private final static String SOURCE_COMPONENT = 62 "com.dummy.admin.package/com.dummy.admin.package.SourceClassName"; 63 private final static String TARGET_COMPONENT = 64 "com.dummy.target.package/com.dummy.target.package.TargetClassName"; 65 private final static int USER_ID = 123; 66 private final static Metadata TEST_PARAMS = new Metadata(SOURCE_COMPONENT, 67 TARGET_COMPONENT, USER_ID, ADMIN_TYPE_DEVICE_OWNER); 68 69 private MockInjector mMockInjector; 70 71 @Before 72 public void setUp() { 73 mMockInjector = new MockInjector(); 74 getOwnerTransferParams().deleteMetadataFile(); 75 } 76 77 @Test 78 public void testSave() { 79 TransferOwnershipMetadataManager paramsManager = getOwnerTransferParams(); 80 assertTrue(paramsManager.saveMetadataFile(TEST_PARAMS)); 81 assertTrue(paramsManager.metadataFileExists()); 82 } 83 84 @Test 85 public void testFileContentValid() { 86 TransferOwnershipMetadataManager paramsManager = getOwnerTransferParams(); 87 assertTrue(paramsManager.saveMetadataFile(TEST_PARAMS)); 88 Path path = Paths.get(new File(mMockInjector.getOwnerTransferMetadataDir(), 89 OWNER_TRANSFER_METADATA_XML).getAbsolutePath()); 90 try { 91 String contents = new String(Files.readAllBytes(path), Charset.forName("UTF-8")); 92 assertEquals( 93 "<?xml version='1.0' encoding='utf-8' standalone='yes' ?>\n" 94 + "<" + TAG_USER_ID + ">" + USER_ID + "</" + TAG_USER_ID + ">\n" 95 + "<" + TAG_SOURCE_COMPONENT + ">" + SOURCE_COMPONENT + "</" 96 + TAG_SOURCE_COMPONENT + ">\n" 97 + "<" + TAG_TARGET_COMPONENT + ">" + TARGET_COMPONENT + "</" 98 + TAG_TARGET_COMPONENT + ">\n" 99 + "<" + TAG_ADMIN_TYPE + ">" + ADMIN_TYPE_DEVICE_OWNER + "</" 100 + TAG_ADMIN_TYPE + ">\n", 101 contents); 102 } catch (IOException e) { 103 e.printStackTrace(); 104 } 105 } 106 107 @Test 108 public void testLoad() { 109 TransferOwnershipMetadataManager paramsManager = getOwnerTransferParams(); 110 final File transferOwnershipMetadataFile = 111 new File(mMockInjector.getOwnerTransferMetadataDir(), OWNER_TRANSFER_METADATA_XML); 112 Log.d(TAG, "testLoad: file path is " + transferOwnershipMetadataFile.getAbsolutePath()); 113 Log.d(TAG, "testLoad: file exists? " + transferOwnershipMetadataFile.exists()); 114 Log.d(TAG, "testLoad: file mkdir?" + transferOwnershipMetadataFile.mkdir()); 115 try { 116 File canonicalFile = transferOwnershipMetadataFile.getCanonicalFile(); 117 File parentFile = canonicalFile.getParentFile(); 118 Log.d(TAG, "testLoad: file getCanonicalFile?" + canonicalFile); 119 Log.d(TAG, "testLoad: getCanonicalFile.getParentFile " + parentFile); 120 Log.d(TAG, "testLoad: parent mkdirs? " + parentFile.mkdirs()); 121 Log.d(TAG, "testLoad: parent exists? " + parentFile.exists()); 122 Log.d(TAG, "testLoad: canonical file.mkdir()? " + canonicalFile.mkdir()); 123 } catch (IOException e) { 124 Log.d(TAG, "testLoad: failed to get canonical file"); 125 } 126 paramsManager.saveMetadataFile(TEST_PARAMS); 127 assertEquals(TEST_PARAMS, paramsManager.loadMetadataFile()); 128 } 129 130 @Test 131 public void testDelete() { 132 TransferOwnershipMetadataManager paramsManager = getOwnerTransferParams(); 133 paramsManager.saveMetadataFile(TEST_PARAMS); 134 paramsManager.deleteMetadataFile(); 135 assertFalse(paramsManager.metadataFileExists()); 136 } 137 138 @After 139 public void tearDown() { 140 getOwnerTransferParams().deleteMetadataFile(); 141 } 142 143 private TransferOwnershipMetadataManager getOwnerTransferParams() { 144 return new TransferOwnershipMetadataManager(mMockInjector); 145 } 146 147 static class MockInjector extends Injector { 148 @Override 149 public File getOwnerTransferMetadataDir() { 150 return Environment.getExternalStorageDirectory(); 151 } 152 } 153 154} 155