| /* |
| * Copyright (C) 2021 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 android.mediapc.cts; |
| |
| import android.os.Environment; |
| |
| import androidx.test.platform.app.InstrumentationRegistry; |
| |
| import org.junit.Assert; |
| |
| import java.io.File; |
| |
| class WorkDir { |
| private static final String MEDIA_PATH_INSTR_ARG_KEY = "media-path"; |
| static private final File getTopDir() { |
| Assert.assertEquals(Environment.getExternalStorageState(), Environment.MEDIA_MOUNTED); |
| return Environment.getExternalStorageDirectory(); |
| } |
| static private final String getTopDirString() { |
| return (getTopDir().getAbsolutePath() + File.separator); |
| } |
| static final String getMediaDirString() { |
| android.os.Bundle bundle = InstrumentationRegistry.getArguments(); |
| String mediaDirString = bundle.getString(MEDIA_PATH_INSTR_ARG_KEY); |
| if (mediaDirString != null) { |
| // user has specified the mediaDirString via instrumentation-arg |
| return mediaDirString + ((mediaDirString.endsWith("/")) ? "" : "/"); |
| } else { |
| return (getTopDirString() + "test/CtsMediaPerformanceClassTestCases-2.2/"); |
| } |
| } |
| } |