Home | History | Annotate | Download | only in util 
      1 /*
      2  * Copyright (C) 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 package com.android.compatibility.common.util;
     17 
     18 import android.util.Log;
     19 
     20 import org.junit.rules.TestRule;
     21 import org.junit.runner.Description;
     22 import org.junit.runners.model.Statement;
     23 
     24 /**
     25  * Custom JUnit4 rule that provides a callback upon test failures.
     26  */
     27 public abstract class OnFailureRule implements TestRule {
     28     private String mLogTag = "OnFailureRule";
     29 
     30     public OnFailureRule() {
     31     }
     32 
     33     public OnFailureRule(String logTag) {
     34         mLogTag = logTag;
     35     }
     36 
     37     @Override
     38     public Statement apply(Statement base, Description description) {
     39         return new Statement() {
     40 
     41             @Override
     42             public void evaluate() throws Throwable {
     43                 try {
     44                     base.evaluate();
     45                 } catch (Throwable t) {
     46                     Log.e(mLogTag, "Test failed: description=" +  description + "\nThrowable=" + t);
     47                     onTestFailure(base, description, t);
     48                     throw t;
     49                 }
     50             }
     51         };
     52     }
     53 
     54     protected abstract void onTestFailure(Statement base, Description description, Throwable t);
     55 }