Index: chrome/android/javatests/src/org/chromium/chrome/browser/TabCountLabelTest.java |
diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/TabCountLabelTest.java b/chrome/android/javatests/src/org/chromium/chrome/browser/TabCountLabelTest.java |
index a3e3fe62f1c9dad640136c043fc571c1fb276627..19317b8b05150a76fb2c01b1846836f0061149eb 100644 |
--- a/chrome/android/javatests/src/org/chromium/chrome/browser/TabCountLabelTest.java |
+++ b/chrome/android/javatests/src/org/chromium/chrome/browser/TabCountLabelTest.java |
@@ -4,16 +4,14 @@ |
package org.chromium.chrome.browser; |
-import static org.chromium.base.test.util.Restriction.RESTRICTION_TYPE_PHONE; |
- |
import android.test.suitebuilder.annotation.MediumTest; |
import android.widget.ImageButton; |
import org.chromium.base.test.util.Feature; |
-import org.chromium.base.test.util.Restriction; |
import org.chromium.chrome.R; |
import org.chromium.chrome.browser.toolbar.TabSwitcherDrawable; |
import org.chromium.chrome.test.ChromeTabbedActivityTestBase; |
+import org.chromium.chrome.test.util.ChromeRestriction; |
import org.chromium.chrome.test.util.ChromeTabUtils; |
/** |
@@ -39,7 +37,7 @@ public class TabCountLabelTest extends ChromeTabbedActivityTestBase { |
*/ |
@MediumTest |
@Feature({"Browser", "Main"}) |
- @Restriction(RESTRICTION_TYPE_PHONE) |
+ @ChromeRestriction(ChromeRestriction.RESTRICTION_TYPE_PHONE) |
public void testTabCountLabel() throws InterruptedException { |
final int tabCount = getActivity().getCurrentTabModel().getCount(); |
tabCountLabelCheck("Initial state", tabCount); |