xref: /aosp_15_r20/cts/common/device-side/bedstead/nene/src/main/java/com/android/bedstead/nene/utils/FailureDumper.kt (revision b7c941bb3fa97aba169d73cee0bed2de8ac964bf)
1 /*
2  * Copyright (C) 2024 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 package com.android.bedstead.nene.utils
17 
18 import java.util.concurrent.ConcurrentHashMap
19 import java.util.concurrent.atomic.AtomicReference
20 
21 interface FailureDumper {
22 
23     /**
24      * Called when a test has failed.
25      */
onTestFailednull26     fun onTestFailed(exception: Throwable) {}
27 
28     companion object {
29         /**
30          * A set of classes which each implement the [FailureDumper] interface - which will be
31          * instantiated (via a no-args constructor) and called when a Bedstead test fails.
32          */
33         val failureDumpers: MutableSet<String> =
34             ConcurrentHashMap<String, Boolean>().keySet(true)
35 
36         private val currentTestName = AtomicReference("")
37 
38         /**
39          * Register currently started test name
40          */
registerCurrentTestnull41         fun registerCurrentTest(testName: String) {
42             currentTestName.set(testName)
43         }
44 
45         /**
46          * Get the name of the last started test method
47          */
getCurrentTestNamenull48         fun getCurrentTestName(): String = currentTestName.get()
49     }
50 }
51