| Index: chrome/android/javatests/src/org/chromium/chrome/browser/gcore/MockConnectedTask.java
|
| diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/gcore/MockConnectedTask.java b/chrome/android/javatests/src/org/chromium/chrome/browser/gcore/MockConnectedTask.java
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..7cae1b04eb95d194b1183aa2e42c57f61d0d5a7a
|
| --- /dev/null
|
| +++ b/chrome/android/javatests/src/org/chromium/chrome/browser/gcore/MockConnectedTask.java
|
| @@ -0,0 +1,57 @@
|
| +// Copyright 2015 The Chromium Authors. All rights reserved.
|
| +// Use of this source code is governed by a BSD-style license that can be
|
| +// found in the LICENSE file.
|
| +
|
| +package org.chromium.chrome.browser.gcore;
|
| +
|
| +import static junit.framework.Assert.assertEquals;
|
| +
|
| +/** Spying mock for ConnectedTask. */
|
| +class MockConnectedTask<T extends ChromeGoogleApiClient> extends ConnectedTask<T> {
|
| + private int mDoWhenConnectedCount;
|
| + private int mCleanUpCount;
|
| + private int mRescheduleCount;
|
| +
|
| + public MockConnectedTask(T client) {
|
| + super(client);
|
| + }
|
| +
|
| + @Override
|
| + protected final String getName() {
|
| + return "MockConnectedTask";
|
| + }
|
| +
|
| + @Override
|
| + protected final void doWhenConnected(T client) {
|
| + mDoWhenConnectedCount++;
|
| + }
|
| +
|
| + @Override
|
| + protected final void cleanUp() {
|
| + mCleanUpCount++;
|
| + }
|
| +
|
| + @Override
|
| + protected final void retry(Runnable task, long delayMs) {
|
| + mRescheduleCount++;
|
| + }
|
| +
|
| + public void assertDoWhenConnectedCalled(int times) {
|
| + assertEquals(times, mDoWhenConnectedCount);
|
| + mDoWhenConnectedCount = 0;
|
| + }
|
| +
|
| + public void assertCleanUpCalled(int times) {
|
| + assertEquals(times, mCleanUpCount);
|
| + mCleanUpCount = 0;
|
| + }
|
| +
|
| + public void assertRescheduleCalled(int times) {
|
| + assertEquals(times, mRescheduleCount);
|
| + mRescheduleCount = 0;
|
| + }
|
| +
|
| + public void assertNoOtherMethodsCalled() {
|
| + assertEquals(0, mDoWhenConnectedCount + mCleanUpCount + mRescheduleCount);
|
| + }
|
| +}
|
|
|