1 /*
2  * Copyright (C) 2016 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.documentsui.testing;
18 
19 import androidx.annotation.BoolRes;
20 import androidx.annotation.NonNull;
21 import androidx.annotation.PluralsRes;
22 import androidx.annotation.StringRes;
23 import android.content.res.Resources;
24 import android.util.SparseArray;
25 import android.util.SparseBooleanArray;
26 
27 import com.android.documentsui.R;
28 import com.android.documentsui.files.QuickViewIntentBuilder;
29 
30 import org.mockito.Mockito;
31 
32 import javax.annotation.Nullable;
33 
34 /**
35  * Abstract to avoid having to implement unnecessary Activity stuff.
36  * Instances are created using {@link #create()}.
37  */
38 public abstract class TestResources extends Resources {
39 
40     public SparseBooleanArray bools;
41     public SparseArray<String> strings;
42     public SparseArray<String> plurals;
43 
TestResources()44     public TestResources() {
45         super(ClassLoader.getSystemClassLoader());
46     }
47 
create()48     public static TestResources create() {
49         TestResources res = Mockito.mock(
50                 TestResources.class, Mockito.CALLS_REAL_METHODS);
51         res.bools = new SparseBooleanArray();
52         res.strings = new SparseArray<>();
53         res.plurals = new SparseArray<>();
54 
55         res.setProductivityDeviceEnabled(false);
56 
57         // quick view package can be set via system property on debug builds.
58         // unfortunately that interfers with testing. For that reason we have
59         // this little hack....QuickViewIntentBuilder will check for this value
60         // and ignore
61         res.setQuickViewerPackage(QuickViewIntentBuilder.IGNORE_DEBUG_PROP);
62         res.setDefaultDocumentsUri(TestProvidersAccess.DOWNLOADS.getUri().toString());
63         return res;
64     }
65 
setQuickViewerPackage(String packageName)66     public void setQuickViewerPackage(String packageName) {
67         strings.put(R.string.trusted_quick_viewer_package, packageName);
68     }
69 
setDefaultDocumentsUri(String uri)70     public void setDefaultDocumentsUri(String uri) {
71         strings.put(R.string.default_root_uri, uri);
72     }
73 
setProductivityDeviceEnabled(boolean enabled)74     public void setProductivityDeviceEnabled(boolean enabled) {
75         bools.put(R.bool.show_documents_root, enabled);
76     }
77 
78     @Override
getBoolean(@oolRes int id)79     public final boolean getBoolean(@BoolRes int id) throws NotFoundException {
80         return bools.get(id);
81     }
82 
83     @Override
getString(@tringRes int id)84     public final @Nullable String getString(@StringRes int id) throws NotFoundException {
85         return strings.get(id);
86     }
87 
88     @Override
89     @NonNull
getString( @tringRes int id, Object... formatArgs)90     public final String getString(
91             @StringRes int id, Object... formatArgs) throws NotFoundException {
92         final String raw = getString(id);
93         return String.format(raw, formatArgs);
94     }
95 
96     @Override
getQuantityString(@luralsRes int id, int size)97     public final @Nullable String getQuantityString(@PluralsRes int id, int size) {
98         return plurals.get(id);
99     }
100 
101     @Override
getQuantityString(@luralsRes int id, int size, Object... args)102     public final @Nullable String getQuantityString(@PluralsRes int id, int size, Object... args) {
103         String format = getQuantityString(id, size);
104         if (format != null) {
105             return String.format(format, args);
106         }
107 
108         return null;
109     }
110 
111     @Override
getText(@tringRes int resId)112     public final CharSequence getText(@StringRes int resId) {
113         return getString(resId);
114     }
115 }
116