OLD | NEW |
(Empty) | |
| 1 // Copyright 2015 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.chrome.browser.gcore; |
| 6 |
| 7 import static junit.framework.Assert.assertEquals; |
| 8 |
| 9 /** Spying mock for ConnectedTask. */ |
| 10 class MockConnectedTask<T extends ChromeGoogleApiClient> extends ConnectedTask<T
> { |
| 11 private int mDoWhenConnectedCount; |
| 12 private int mCleanUpCount; |
| 13 private int mRescheduleCount; |
| 14 |
| 15 public MockConnectedTask(T client) { |
| 16 super(client); |
| 17 } |
| 18 |
| 19 @Override |
| 20 protected final String getName() { |
| 21 return "MockConnectedTask"; |
| 22 } |
| 23 |
| 24 @Override |
| 25 protected final void doWhenConnected(T client) { |
| 26 mDoWhenConnectedCount++; |
| 27 } |
| 28 |
| 29 @Override |
| 30 protected final void cleanUp() { |
| 31 mCleanUpCount++; |
| 32 } |
| 33 |
| 34 @Override |
| 35 protected final void retry(Runnable task, long delayMs) { |
| 36 mRescheduleCount++; |
| 37 } |
| 38 |
| 39 public void assertDoWhenConnectedCalled(int times) { |
| 40 assertEquals(times, mDoWhenConnectedCount); |
| 41 mDoWhenConnectedCount = 0; |
| 42 } |
| 43 |
| 44 public void assertCleanUpCalled(int times) { |
| 45 assertEquals(times, mCleanUpCount); |
| 46 mCleanUpCount = 0; |
| 47 } |
| 48 |
| 49 public void assertRescheduleCalled(int times) { |
| 50 assertEquals(times, mRescheduleCount); |
| 51 mRescheduleCount = 0; |
| 52 } |
| 53 |
| 54 public void assertNoOtherMethodsCalled() { |
| 55 assertEquals(0, mDoWhenConnectedCount + mCleanUpCount + mRescheduleCount
); |
| 56 } |
| 57 } |
OLD | NEW |