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
17package com.android.documentsui.picker;
18
19import android.annotation.RequiresPermission;
20import android.content.Intent;
21import android.util.Pair;
22
23import com.android.documentsui.testing.TestEnv;
24import com.android.documentsui.testing.TestEventListener;
25
26import org.mockito.Mockito;
27
28public abstract class TestActivity extends AbstractBase {
29
30    public TestEventListener<Pair<Integer, Intent>> setResult;
31    public TestEventListener<Pair<Intent, Integer>> startActivityForResult;
32
33    public static TestActivity create(TestEnv env) {
34        TestActivity activity = Mockito.mock(TestActivity.class, Mockito.CALLS_REAL_METHODS);
35        activity.init(env);
36        return activity;
37    }
38
39    @Override
40    public void init(TestEnv env) {
41        super.init(env);
42
43        setResult = new TestEventListener<>();
44        startActivityForResult = new TestEventListener<>();
45    }
46
47    @Override
48    public void setResult(int resultCode, Intent intent, int notUsed) {
49        setResult.accept(Pair.create(resultCode, intent));
50    }
51
52    @Override
53    public final void startActivityForResult(@RequiresPermission Intent intent, int requestCode) {
54        startActivityForResult.accept(Pair.create(intent, requestCode));
55    }
56}
57
58// Trick Mockito into finding our Addons methods correctly. W/o this
59// hack, Mockito thinks Addons methods are not implemented.
60abstract class AbstractBase extends com.android.documentsui.TestActivity
61        implements ActionHandler.Addons {}
62