From 0e42d0c0adf3e7e9bf72474e0a869e4bb586635e Mon Sep 17 00:00:00 2001
From: Felipe Leme <felipeal@google.com>
Date: Thu, 8 Sep 2016 09:50:50 -0700
Subject: [PATCH] Restore right number of expected intents.

They were changed during a refactoring, which is now finished.

Test: cts-tradefed run commandAndExit cts -m CtsHostsideNetworkTests -t com.android.cts.net.HostsideRestrictBackgroundNetworkTests#testDataSaverMode_enabled

Fixes: 28791717
Change-Id: Ia5e99c0c3d421b7d3b58e11ddde4da222d5f8c15
---
 .../src/com/android/cts/net/hostside/DataSaverModeTest.java | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/tests/cts/hostside/app/src/com/android/cts/net/hostside/DataSaverModeTest.java b/tests/cts/hostside/app/src/com/android/cts/net/hostside/DataSaverModeTest.java
index 8efea01844..7ca302f52a 100644
--- a/tests/cts/hostside/app/src/com/android/cts/net/hostside/DataSaverModeTest.java
+++ b/tests/cts/hostside/app/src/com/android/cts/net/hostside/DataSaverModeTest.java
@@ -133,6 +133,7 @@ public class DataSaverModeTest extends AbstractRestrictBackgroundNetworkTestCase
         assertDataSaverStatusOnBackground(RESTRICT_BACKGROUND_STATUS_ENABLED);
 
         assertsForegroundAlwaysHasNetworkAccess();
+        assertRestrictBackgroundChangedReceived(1);
         assertDataSaverStatusOnBackground(RESTRICT_BACKGROUND_STATUS_ENABLED);
 
         // UID policies live by the Highlander rule: "There can be only one".
@@ -147,16 +148,17 @@ public class DataSaverModeTest extends AbstractRestrictBackgroundNetworkTestCase
         // Check status after removing blacklist.
         // ...re-enables first
         addRestrictBackgroundBlacklist(mUid);
+        assertRestrictBackgroundChangedReceived(4);
         assertDataSaverStatusOnBackground(RESTRICT_BACKGROUND_STATUS_ENABLED);
         assertsForegroundAlwaysHasNetworkAccess();
         // ... remove blacklist - access's still rejected because Data Saver is on
         removeRestrictBackgroundBlacklist(mUid);
-        assertRestrictBackgroundChangedReceived(3);
+        assertRestrictBackgroundChangedReceived(4);
         assertDataSaverStatusOnBackground(RESTRICT_BACKGROUND_STATUS_ENABLED);
         assertsForegroundAlwaysHasNetworkAccess();
         // ... finally, disable Data Saver
         setRestrictBackground(false);
-        assertRestrictBackgroundChangedReceived(4);
+        assertRestrictBackgroundChangedReceived(5);
         assertDataSaverStatusOnBackground(RESTRICT_BACKGROUND_STATUS_DISABLED);
         assertsForegroundAlwaysHasNetworkAccess();
     }