Home | History | Annotate | Download | only in testutis
      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 package com.android.server.testutis;
     17 
     18 import android.test.MoreAsserts;
     19 
     20 import junit.framework.Assert;
     21 
     22 public class TestUtils {
     23     private TestUtils() {
     24     }
     25 
     26     public interface ExceptionRunnable {
     27         void run() throws Exception;
     28     }
     29 
     30     public static void assertExpectException(Class<? extends Throwable> expectedExceptionType,
     31             String expectedExceptionMessageRegex, ExceptionRunnable r) {
     32         try {
     33             r.run();
     34         } catch (Throwable e) {
     35             Assert.assertTrue(
     36                     "Expected exception type was " + expectedExceptionType.getName()
     37                     + " but caught " + e.getClass().getName(),
     38                     expectedExceptionType.isAssignableFrom(e.getClass()));
     39             if (expectedExceptionMessageRegex != null) {
     40                 MoreAsserts.assertContainsRegex(expectedExceptionMessageRegex, e.getMessage());
     41             }
     42             return; // Pass.
     43         }
     44         Assert.fail("Expected exception type " + expectedExceptionType.getName()
     45                 + " was not thrown");
     46     }
     47 }
     48