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 com.android.nn.crashtest.app;
18 
19 import android.content.Intent;
20 
21 import com.android.nn.crashtest.core.CrashTest;
22 import com.android.nn.crashtest.core.CrashTestCoordinator;
23 import com.android.nn.crashtest.core.test.RandomGraphTest;
24 
25 public class NNRandomGraphTestActivity extends NNCrashTestActivity {
26   private static final String TAG = "NN_RAND_MODEL";
27 
28 
29   @Override
getTag()30   protected String getTag() {
31     return TAG;
32   }
33 
34   @Override
getTestName(Intent intent)35   protected String getTestName(Intent intent) {
36     return intent.getStringExtra(RandomGraphTest.TEST_NAME);
37   }
38 
39   @Override
getTestDurationMillis(Intent intent)40   protected long getTestDurationMillis(Intent intent) {
41     return intent.getLongExtra(
42             RandomGraphTest.MAX_TEST_DURATION, RandomGraphTest.DEFAULT_MAX_TEST_DURATION_MILLIS);
43   }
44 
45   @Override
getIntentInitializer(Intent intent)46   protected CrashTestCoordinator.CrashTestIntentInitializer getIntentInitializer(Intent intent) {
47     return RandomGraphTest.intentInitializer(intent);
48   }
49 
50   @Override
getTestClass()51   protected Class<? extends CrashTest> getTestClass() {
52     return RandomGraphTest.class;
53   }
54 }