retrofit2.mock.NetworkBehavior Java Examples

<T> T adapt(Method method, NetworkBehavior networkBehavior, Call<?> call) {
  try {
    Class<?> behaviorCallClass = Class.forName("retrofit2.mock.BehaviorCall");

    Constructor<?> constructor = behaviorCallClass.getDeclaredConstructor(NetworkBehavior.class,
        ExecutorService.class, Call.class);
    constructor.setAccessible(true);

    Call<?> behaviorCall = (Call<?>) constructor.newInstance(networkBehavior,
        Executors.newCachedThreadPool(),
        call);

    return (T) retrofit
        .callAdapter(method.getGenericReturnType(), method.getAnnotations())
        .adapt(behaviorCall);
  } catch (Exception e) {
    throw new RuntimeException(e);
  }
}
 

@Provides
@Singleton
MockRetrofit provideRetrofit(GsonConverterFactory gson, OkHttpClient okHttpClient) {
    NetworkBehavior behavior = NetworkBehavior.create();
    behavior.setDelay(1, TimeUnit.SECONDS);
    behavior.setVariancePercent(40);
    behavior.setErrorPercent(2);
    Retrofit retrofit = new Retrofit.Builder()
            .baseUrl(BuildConfig.API_URL)
            .addCallAdapterFactory(RxJava2CallAdapterFactory.create())
            .addCallAdapterFactory(LiveDataCallAdapterFactory.create())
            .client(okHttpClient)
            .build();
    return new MockRetrofit.Builder(retrofit)
            .networkBehavior(behavior)
            .build();
}
 

private NetworkBehavior networkBehaviour(Metadata<Retrofit> metadata) {
  Retrofit retrofit = metadata.getAnnotation();

  NetworkBehavior networkBehavior = NetworkBehavior.create();
  networkBehavior.setDelay(retrofit.delay(), TimeUnit.MILLISECONDS);
  networkBehavior.setVariancePercent(retrofit.variancePercentage());
  networkBehavior.setFailurePercent(retrofit.failurePercent());

  return networkBehavior;
}
 

@Provides
@ApplicationScope
Picasso providePicasso(OkHttpClient client, NetworkBehavior behavior,
                       @IsMockMode boolean isMockMode, Application app) {
    Picasso.Builder builder = new Picasso.Builder(app).downloader(new OkHttp3Downloader(client));
    if (isMockMode) {
        builder.addRequestHandler(new MockRequestHandler(behavior, app.getAssets()));
    }
    builder.listener((picasso, uri, exception) -> Timber.e(exception, "Error while loading image %s", uri));
    return builder.build();
}
 

@Provides
@ApplicationScope
MockRetrofit provideMockRetrofit(Retrofit retrofit, NetworkBehavior behavior) {
    return new MockRetrofit.Builder(retrofit)
            .networkBehavior(behavior)
            .build();
}
 

@Provides
@ApplicationScope
NetworkBehavior provideBehavior(@NetworkDelay Preference<Long> networkDelay,
                                @NetworkFailurePercent Preference<Integer> networkFailurePercent,
                                @NetworkVariancePercent Preference<Integer> networkVariancePercent) {
    NetworkBehavior behavior = NetworkBehavior.create();
    behavior.setDelay(networkDelay.get(), MILLISECONDS);
    behavior.setFailurePercent(networkFailurePercent.get());
    behavior.setVariancePercent(networkVariancePercent.get());
    return behavior;
}
 

@Override
public void setUp() throws Exception {
    super.setUp();
    retrofit = new Retrofit.Builder().baseUrl("http://test.com")
            .client(new OkHttpClient())
            .addConverterFactory(JacksonConverterFactory.create())
            .build();

    NetworkBehavior behavior = NetworkBehavior.create();

    mockRetrofit = new MockRetrofit.Builder(retrofit)
            .networkBehavior(behavior)
            .build();
}
 

private static NetworkBehavior getNetworkBehavior(int delayMsec, int delayVariance,
                                                  int failurePercent, int errorPercent) {
    NetworkBehavior networkBehavior = NetworkBehavior.create();
    networkBehavior.setDelay(delayMsec, TimeUnit.MILLISECONDS);
    networkBehavior.setVariancePercent(delayVariance);
    // "failure" means network layer failure
    networkBehavior.setFailurePercent(failurePercent);
    // "error" means HTTP error
    networkBehavior.setErrorPercent(errorPercent);
    return networkBehavior;
}
 

@Provides @Singleton NetworkBehavior provideBehavior(@NetworkDelay Preference<Long> networkDelay,
    @NetworkVariancePercent Preference<Integer> networkVariancePercent,
    @NetworkFailurePercent Preference<Integer> networkFailurePercent,
    @NetworkErrorPercent Preference<Integer> networkErrorPercent,
    Preference<NetworkErrorCode> networkErrorCode) {
  NetworkBehavior behavior = NetworkBehavior.create();
  behavior.setDelay(networkDelay.get(), MILLISECONDS);
  behavior.setVariancePercent(networkVariancePercent.get());
  behavior.setFailurePercent(networkFailurePercent.get());
  behavior.setErrorPercent(networkErrorPercent.get());
  behavior.setErrorFactory(
      () -> Response.error(networkErrorCode.get().code, ResponseBody.create(null, new byte[0])));
  return behavior;
}
 

@Provides @Singleton Picasso providePicasso(OkHttpClient client, NetworkBehavior behavior,
    @IsMockMode boolean isMockMode, Application app) {
  Picasso.Builder builder = new Picasso.Builder(app).downloader(new OkHttp3Downloader(client));
  if (isMockMode) {
    builder.addRequestHandler(new MockRequestHandler(behavior, app.getAssets()));
  }
  builder.listener((picasso, uri, exception) -> {
    Timber.e(exception, "Error while loading image %s", uri);
  });
  return builder.build();
}
 

private static LoginOrchestrator makeOrchestrator(CredentialSource credSource,
                                                  CredentialSink credSink,
                                                  boolean useGhostAuth,
                                                  NetworkBehavior networkBehavior) {
    BlogUrlValidator blogUrlValidator = blogUrl -> Observable.just("http://" + blogUrl);
    HACKListener hackListener = mock(HACKListener.class);
    final MockApiProviderFactory apiProviderFactory = new MockApiProviderFactory(
            useGhostAuth, networkBehavior);
    return new LoginOrchestrator(blogUrlValidator, apiProviderFactory, credSource,
            credSink, hackListener);
}
 

@Test
public void ghostAuth_networkFailure() {
    NetworkBehavior failingNetworkBehavior = Helpers.getFailingNetworkBehaviour();
    LoginOrchestrator orchestrator = makeOrchestrator(credSource, credSink, true, failingNetworkBehavior);
    when(credSource.getGhostAuthCode(any())).thenReturn(Observable.just("auth-code"));
    orchestrator.listen(listener);

    orchestrator.start(BLOG_URL_WITHOUT_PROTOCOL);

    verify(listener).onNetworkError(any(), argThat(sameInstance(failingNetworkBehavior.failureException())));
}
 

@Override public Call onIllegalMock(AssertionError assertionError, Metadata<Retrofit> metadata) {
  checkReturnMethodTypeIsCall(metadata);

  String errorMessage = assertionError.getMessage() != null ? assertionError.getMessage() : "";
  String safeErrorMessage = adaptErrorResponse(errorMessage, metadata);
  ResponseBody responseBody = ResponseBody
      .create(MediaType.parse("application/json"), safeErrorMessage);

  NetworkBehavior networkBehavior = networkBehaviour(metadata);
  return callAdapter.adapt(metadata.getMethod(),
      networkBehavior, Calls.response(Response.error(404, responseBody)));
}
 

@Override public Call onLegalMock(Object mock, Metadata<Retrofit> metadata) {
  checkReturnMethodTypeIsCall(metadata);
  checkTypeMockIsNotCall(metadata, mock);

  NetworkBehavior networkBehavior = networkBehaviour(metadata);
  return callAdapter.adapt(metadata.getMethod(),
      networkBehavior, Calls.response(mock));
}
 

private NetworkBehavior networkBehaviour(Metadata<Rx2Retrofit> metadata) {
  Rx2Retrofit retrofit = metadata.getAnnotation();

  NetworkBehavior networkBehavior = NetworkBehavior.create();
  networkBehavior.setDelay(retrofit.delay(), TimeUnit.MILLISECONDS);
  networkBehavior.setVariancePercent(retrofit.variancePercentage());
  networkBehavior.setFailurePercent(retrofit.failurePercent());

  return networkBehavior;
}
 

@Override public Object onIllegalMock(final AssertionError assertionError, final Metadata<Rx2Retrofit> metadata) {
  checkReturnMethodTypeIsSingleOrCompletable(metadata);

  final String errorMessage = assertionError.getMessage() != null ? assertionError.getMessage() : "";
  final String adaptedErrorMessage = adaptErrorResponse(errorMessage, metadata);
  NetworkBehavior networkBehavior = networkBehaviour(metadata);

  final ResponseBody responseBody = ResponseBody
      .create(MediaType.parse("application/json"), adaptedErrorMessage);

  return callAdapter.adapt(metadata.getMethod(),
      networkBehavior, Calls.response(Response.error(404, responseBody)));
}
 

@Override public Object onLegalMock(final Object mock, final Metadata<Rx2Retrofit> metadata) {
  checkReturnMethodTypeIsSingleOrCompletable(metadata);
  checkTypeMockIsNotSingleNeitherResponse(metadata, mock);

  NetworkBehavior networkBehavior = networkBehaviour(metadata);
  return callAdapter.adapt(metadata.getMethod(),
      networkBehavior, Calls.response(mock));
}
 

private NetworkBehavior networkBehaviour(Metadata<RxRetrofit> metadata) {
  RxRetrofit retrofit = metadata.getAnnotation();

  NetworkBehavior networkBehavior = NetworkBehavior.create();
  networkBehavior.setDelay(retrofit.delay(), TimeUnit.MILLISECONDS);
  networkBehavior.setVariancePercent(retrofit.variancePercentage());
  networkBehavior.setFailurePercent(retrofit.failurePercent());

  return networkBehavior;
}
 

@Override public Observable onIllegalMock(final AssertionError assertionError, final Metadata<RxRetrofit> metadata) {
  checkReturnMethodTypeIsObservable(metadata);

  final String errorMessage = assertionError.getMessage() != null ? assertionError.getMessage() : "";
  final String adaptedErrorMessage = adaptErrorResponse(errorMessage, metadata);
  NetworkBehavior networkBehavior = networkBehaviour(metadata);

  final ResponseBody responseBody = ResponseBody
      .create(MediaType.parse("application/json"), adaptedErrorMessage);

  return callAdapter.adapt(metadata.getMethod(),
      networkBehavior, Calls.response(Response.error(404, responseBody)));
}
 

@Override public Observable onLegalMock(final Object mock, final Metadata<RxRetrofit> metadata) {
  checkReturnMethodTypeIsObservable(metadata);
  checkTypeMockIsNotObservableNeitherResponse(metadata, mock);

  NetworkBehavior networkBehavior = networkBehaviour(metadata);
  return callAdapter.adapt(metadata.getMethod(),
      networkBehavior, Calls.response(mock));
}
 

@Before
public void setup() {

    Retrofit retrofit = new Retrofit.Builder()
            .baseUrl(LyftApi.API_ROOT)
            .build();

    NetworkBehavior behavior = NetworkBehavior.create();
    behavior.setDelay(1, TimeUnit.MILLISECONDS);
    behavior.setVariancePercent(0);
    behavior.setFailurePercent(0);
    MockRetrofit mockRetrofit = new MockRetrofit.Builder(retrofit)
            .networkBehavior(behavior)
            .build();
    BehaviorDelegate<LyftApi> lyftApidelegate = mockRetrofit.create(LyftApi.class);
    lyftPublicApi = spy(new MockLyftApi(lyftApidelegate));

    BehaviorDelegate<GoogleApi> googleApiDelegate = mockRetrofit.create(GoogleApi.class);
    googleApi = spy(new MockGoogleApi(googleApiDelegate));
    HashMap<String, GoogleLatLng> addressToLatLngMap = new HashMap<>();
    addressToLatLngMap.put(PICKUP_ADDR, new GoogleLatLng(PICKUP_LAT, PICKUP_LNG));
    addressToLatLngMap.put(DROPOFF_ADDR, new GoogleLatLng(DROPOFF_LAT, DROPOFF_LNG));
    googleApi.setAddressToLatLngMap(addressToLatLngMap);

    mockCallback = mock(LyftButton.ResultLoadedCallback.class);
    callManager = new LyftButtonCallManager("clientId", lyftPublicApi, googleApi, new MockExecutorService());
}
 

public MockRequestHandler(NetworkBehavior behavior, AssetManager assetManager) {
  this.behavior = behavior;
  this.assetManager = assetManager;
}
 

@Provides @Singleton MockRetrofit provideMockRetrofit(Retrofit retrofit,
    NetworkBehavior behavior) {
  return new MockRetrofit.Builder(retrofit)
      .networkBehavior(behavior)
      .build();
}
 

public MockRequestHandler(NetworkBehavior behavior, AssetManager assetManager) {
    this.behavior = behavior;
    this.assetManager = assetManager;
}
 

public static NetworkBehavior getFailingNetworkBehaviour() {
    return getNetworkBehavior(0, 0, 100, 0);
}
 

public static NetworkBehavior getIdealNetworkBehavior() {
    return getNetworkBehavior(0, 0, 0, 0);
}
 

@NonNull
public static MockRetrofit getMockRetrofit(Retrofit retrofit, NetworkBehavior networkBehavior) {
    return new MockRetrofit.Builder(retrofit)
            .networkBehavior(networkBehavior)
            .build();
}
 

public MockApiProviderFactory(boolean useGhostAuth, NetworkBehavior networkBehavior) {
    mUseGhostAuth = useGhostAuth;
    mNetworkBehavior = networkBehavior;
}