Home | History | Annotate | Download | only in devicepolicy 
      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 
     17 package com.android.server.devicepolicy;
     18 
     19 import static com.android.server.devicepolicy.TransferOwnershipMetadataManager
     20         .ADMIN_TYPE_DEVICE_OWNER;
     21 import static com.android.server.devicepolicy.TransferOwnershipMetadataManager
     22         .OWNER_TRANSFER_METADATA_XML;
     23 import static com.android.server.devicepolicy.TransferOwnershipMetadataManager.TAG_ADMIN_TYPE;
     24 import static com.android.server.devicepolicy.TransferOwnershipMetadataManager.TAG_SOURCE_COMPONENT;
     25 import static com.android.server.devicepolicy.TransferOwnershipMetadataManager.TAG_TARGET_COMPONENT;
     26 import static com.android.server.devicepolicy.TransferOwnershipMetadataManager.TAG_USER_ID;
     27 
     28 import static org.junit.Assert.assertEquals;
     29 import static org.junit.Assert.assertFalse;
     30 import static org.junit.Assert.assertTrue;
     31 
     32 import android.os.Environment;
     33 import android.support.test.runner.AndroidJUnit4;
     34 import android.util.Log;
     35 
     36 import com.android.server.devicepolicy.TransferOwnershipMetadataManager.Injector;
     37 import com.android.server.devicepolicy.TransferOwnershipMetadataManager.Metadata;
     38 
     39 import org.junit.After;
     40 import org.junit.Before;
     41 import org.junit.Test;
     42 import org.junit.runner.RunWith;
     43 
     44 import java.io.File;
     45 import java.io.IOException;
     46 import java.nio.charset.Charset;
     47 import java.nio.file.Files;
     48 import java.nio.file.Path;
     49 import 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)
     59 public 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