1 /* 2 * Copyright (C) 2020 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 android.dynamicmime.testapp.assertions; 18 19 import android.content.Context; 20 import android.dynamicmime.testapp.util.AppMimeGroups; 21 import android.util.ArraySet; 22 23 import java.util.Set; 24 25 public class AppAssertionsByGroupData extends AssertionsByGroupData { 26 private final AppMimeGroups mAppMimeGroups; 27 AppAssertionsByGroupData(Context context, String targetPackage)28 AppAssertionsByGroupData(Context context, String targetPackage) { 29 mAppMimeGroups = AppMimeGroups.with(context, targetPackage); 30 } 31 32 @Override getMimeGroup(String mimeGroup)33 protected Set<String> getMimeGroup(String mimeGroup) { 34 try { 35 String[] mimeTypes = mAppMimeGroups.get(mimeGroup); 36 return mimeTypes != null ? new ArraySet<>(mimeTypes) : null; 37 } catch (Throwable exception) { 38 return null; 39 } 40 } 41 } 42