components/cronet/android/test/src/org/chromium/net/MockUrlRequestJobFactory.java

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 package org.chromium.net;
   6
   7 import static junit.framework.Assert.assertTrue;
   8
   9 import org.chromium.base.annotations.JNINamespace;
  10 import org.chromium.net.test.FailurePhase;
  11
  12 /**
  13  * Helper class to set up url interceptors for testing purposes.
  14  */
  15 @JNINamespace("cronet")
  16 public final class MockUrlRequestJobFactory {
  17     /**
  18      * Sets up URL interceptors.
  19      */
  20     public static void setUp() {
  21         nativeAddUrlInterceptors();
  22     }
  23
  24     /**
  25      * Constructs a mock URL that hangs or fails at certain phase.
  26      *
  27      * @param phase at which request fails. It should be a value in
  28      *              org.chromium.net.test.FailurePhase.
  29      * @param netError reported by UrlRequestJob. Passing -1, results in hang.
  30      */
  31     public static String getMockUrlWithFailure(int phase, int netError) {
  32         assertTrue(netError < 0);
  33         switch (phase) {
  34             case FailurePhase.START:
  35             case FailurePhase.READ_SYNC:
  36             case FailurePhase.READ_ASYNC:
  37                 break;
  38             default:
  39                 throw new IllegalArgumentException(
  40                         "phase not in org.chromium.net.test.FailurePhase");
  41         }
  42         return nativeGetMockUrlWithFailure(phase, netError);
  43     }
  44
  45     /**
  46      * Constructs a mock URL that synchronously responds with data repeated many
  47      * times.
  48      *
  49      * @param data to return in response.
  50      * @param dataRepeatCount number of times to repeat the data.
  51      */
  52     public static String getMockUrlForData(String data, int dataRepeatCount) {
  53         return nativeGetMockUrlForData(data, dataRepeatCount);
  54     }
  55
  56     /**
  57      * Constructs a mock URL that will fail with an SSL certificate error.
  58      */
  59     public static String getMockUrlForSSLCertificateError() {
  60         return nativeGetMockUrlForSSLCertificateError();
  61     }
  62
  63     private static native void nativeAddUrlInterceptors();
  64
  65     private static native String nativeGetMockUrlWithFailure(int phase, int netError);
  66
  67     private static native String nativeGetMockUrlForData(String data,
  68             int dataRepeatCount);
  69
  70     private static native String nativeGetMockUrlForSSLCertificateError();
  71 }