| /* |
| * Copyright (C) 2017 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.server.wm; |
| |
| import static android.content.res.Configuration.ORIENTATION_PORTRAIT; |
| import static android.graphics.GraphicBuffer.USAGE_HW_TEXTURE; |
| import static android.graphics.GraphicBuffer.USAGE_SW_READ_RARELY; |
| |
| import android.app.ActivityManager.TaskSnapshot; |
| import android.graphics.Canvas; |
| import android.graphics.Color; |
| import android.graphics.GraphicBuffer; |
| import android.graphics.PixelFormat; |
| import android.graphics.Rect; |
| import android.os.UserManager; |
| import android.support.test.InstrumentationRegistry; |
| |
| import org.junit.After; |
| import org.junit.Before; |
| import org.junit.BeforeClass; |
| |
| import java.io.File; |
| |
| /** |
| * Base class for tests that use a {@link TaskSnapshotPersister}. |
| */ |
| class TaskSnapshotPersisterTestBase extends WindowTestsBase { |
| |
| private static final Rect TEST_INSETS = new Rect(10, 20, 30, 40); |
| |
| TaskSnapshotPersister mPersister; |
| TaskSnapshotLoader mLoader; |
| int mTestUserId; |
| static File sFilesDir; |
| |
| @BeforeClass |
| public static void setUpUser() { |
| sFilesDir = InstrumentationRegistry.getContext().getFilesDir(); |
| } |
| |
| @Before |
| public void setUp() throws Exception { |
| super.setUp(); |
| final UserManager um = UserManager.get(InstrumentationRegistry.getContext()); |
| mTestUserId = um.getUserHandle(); |
| mPersister = new TaskSnapshotPersister(userId -> sFilesDir); |
| mLoader = new TaskSnapshotLoader(mPersister); |
| mPersister.start(); |
| } |
| |
| @After |
| public void tearDown() throws Exception { |
| cleanDirectory(); |
| } |
| |
| private void cleanDirectory() { |
| final File[] files = new File(sFilesDir, "snapshots").listFiles(); |
| if (files == null) { |
| return; |
| } |
| for (File file : files) { |
| if (!file.isDirectory()) { |
| file.delete(); |
| } |
| } |
| } |
| |
| TaskSnapshot createSnapshot() { |
| final GraphicBuffer buffer = GraphicBuffer.create(100, 100, PixelFormat.RGBA_8888, |
| USAGE_HW_TEXTURE | USAGE_SW_READ_RARELY | USAGE_SW_READ_RARELY); |
| Canvas c = buffer.lockCanvas(); |
| c.drawColor(Color.RED); |
| buffer.unlockCanvasAndPost(c); |
| return new TaskSnapshot(buffer, ORIENTATION_PORTRAIT, TEST_INSETS, |
| false /* reducedResolution */, 1f /* scale */); |
| } |
| } |