| Index: chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/NewTabPageAdapterTest.java
|
| diff --git a/chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/NewTabPageAdapterTest.java b/chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/NewTabPageAdapterTest.java
|
| index 2d31743e17147960433e0a28382cbcdc4c75c916..d1bcd9104a473947fbbd6cfa526dafc25adbb8db 100644
|
| --- a/chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/NewTabPageAdapterTest.java
|
| +++ b/chrome/android/junit/src/org/chromium/chrome/browser/ntp/cards/NewTabPageAdapterTest.java
|
| @@ -44,6 +44,7 @@
|
| import org.chromium.chrome.browser.favicon.LargeIconBridge.LargeIconCallback;
|
| import org.chromium.chrome.browser.ntp.LogoBridge.LogoObserver;
|
| import org.chromium.chrome.browser.ntp.MostVisitedItem;
|
| +import org.chromium.chrome.browser.ntp.NewTabPage.SelfRegistrable;
|
| import org.chromium.chrome.browser.ntp.NewTabPageView.NewTabPageManager;
|
| import org.chromium.chrome.browser.ntp.snippets.CategoryInt;
|
| import org.chromium.chrome.browser.ntp.snippets.CategoryStatus;
|
| @@ -55,6 +56,7 @@
|
| import org.chromium.chrome.browser.preferences.ChromePreferenceManager;
|
| import org.chromium.chrome.browser.profiles.MostVisitedSites.MostVisitedURLsObserver;
|
| import org.chromium.chrome.browser.signin.SigninManager;
|
| +import org.chromium.chrome.browser.signin.SigninManager.SignInAllowedObserver;
|
| import org.chromium.chrome.browser.signin.SigninManager.SignInStateObserver;
|
| import org.chromium.testing.local.LocalRobolectricTestRunner;
|
|
|
| @@ -776,10 +778,18 @@ public void testSigninPromo() {
|
| assertEquals(1, signinPromoGroup.getItemCount());
|
| assertEquals(ItemViewType.PROMO, signinPromoGroup.getItemViewType(0));
|
|
|
| - ntpManager.mSignInStateObserver.onSignedIn();
|
| + ((SignInStateObserver) ntpManager.mSelfRegistrable).onSignedIn();
|
| assertEquals(0, signinPromoGroup.getItemCount());
|
|
|
| - ntpManager.mSignInStateObserver.onSignedOut();
|
| + ((SignInStateObserver) ntpManager.mSelfRegistrable).onSignedOut();
|
| + assertEquals(1, signinPromoGroup.getItemCount());
|
| +
|
| + when(mMockSigninManager.isSignInAllowed()).thenReturn(false);
|
| + ((SignInAllowedObserver) ntpManager.mSelfRegistrable).onSignInAllowedChanged();
|
| + assertEquals(0, signinPromoGroup.getItemCount());
|
| +
|
| + when(mMockSigninManager.isSignInAllowed()).thenReturn(true);
|
| + ((SignInAllowedObserver) ntpManager.mSelfRegistrable).onSignInAllowedChanged();
|
| assertEquals(1, signinPromoGroup.getItemCount());
|
| }
|
|
|
| @@ -840,7 +850,7 @@ private int getCategory(TreeNode itemGroup) {
|
|
|
| private static class MockNewTabPageManager implements NewTabPageManager {
|
| SuggestionsSource mSuggestionsSource;
|
| - SignInStateObserver mSignInStateObserver;
|
| + SelfRegistrable mSelfRegistrable;
|
|
|
| public MockNewTabPageManager(SuggestionsSource suggestionsSource) {
|
| mSuggestionsSource = suggestionsSource;
|
| @@ -1002,8 +1012,8 @@ public void closeContextMenu() {
|
| }
|
|
|
| @Override
|
| - public void registerSignInStateObserver(SignInStateObserver signInStateObserver) {
|
| - mSignInStateObserver = signInStateObserver;
|
| + public void addSelfRegistrable(SelfRegistrable selfRegistrable) {
|
| + mSelfRegistrable = selfRegistrable;
|
| }
|
|
|
| @Override
|
|
|