1 /*
2  * Copyright 2018 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 androidx.build.dependencyTracker
18 
19 import org.junit.rules.TestRule
20 import org.junit.runner.Description
21 import org.junit.runners.model.Statement
22 
23 /**
24  * Special rule for dependency detector tests that will attach logs to a failure.
25  */
26 class AttachLogsTestRule() : TestRule {
27     val logger = ToStringLogger()
applynull28     override fun apply(base: Statement, description: Description): Statement {
29         return object : Statement() {
30             override fun evaluate() {
31                 try {
32                     base.evaluate()
33                 } catch (t: Throwable) {
34                     throw Exception(
35                             """
36                                 test failed with msg: ${t.message}
37                                 logs:
38                                 ${logger.buildString()}
39                             """.trimIndent(),
40                             t
41                     )
42                 }
43             }
44         }
45     }
46 }