Populate selected network in RecommendationRequest

Bug: 34467596
Test: ./runtests.sh
Change-Id: Ide0b4cbc9114bb8f79a2671103ab7f1cf0590dd0

1 files changed, 17 insertions, 0 deletions

diff --git a/tests/wifitests/src/com/android/server/wifi/RecommendedNetworkEvaluatorTest.java b/tests/wifitests/src/com/android/server/wifi/RecommendedNetworkEvaluatorTest.java
index 719412952..d5ba0961d 100644
--- a/tests/wifitests/src/com/android/server/wifi/RecommendedNetworkEvaluatorTest.java
+++ b/tests/wifitests/src/com/android/server/wifi/RecommendedNetworkEvaluatorTest.java
@@ -405,6 +405,23 @@ public class RecommendedNetworkEvaluatorTest {
         assertTrue(request.getScanResults()[0].untrusted);
     }
 
+    @Test
+    public void testEvaluateNetworks_requestContainsLastSelectedNetwork() {
+        int lastSelectedNetworkId = 5;
+        long lastSelectedTimestamp = 1000;
+        when(mWifiConfigManager.wasEphemeralNetworkDeleted(anyString())).thenReturn(false);
+        when(mWifiConfigManager.getLastSelectedNetwork()).thenReturn(lastSelectedNetworkId);
+        when(mWifiConfigManager.getLastSelectedTimeStamp()).thenReturn(lastSelectedTimestamp);
+
+        mRecommendedNetworkEvaluator.evaluateNetworks(Lists.newArrayList(TRUSTED_SCAN_DETAIL),
+                null, null, false, false /* untrustedNetworkAllowed */, null);
+
+        verify(mNetworkScoreManager).requestRecommendation(mRecommendationRequestCaptor.capture());
+        RecommendationRequest request = mRecommendationRequestCaptor.getValue();
+        assertEquals(lastSelectedNetworkId, request.getLastSelectedNetworkId());
+        assertEquals(lastSelectedTimestamp, request.getLastSelectedNetworkTimestamp());
+    }
+
     private static ScanDetail buildScanDetail(String ssid) {
         return new ScanDetail(WifiSsid.createFromAsciiEncoded(ssid),
                 "00:00:00:00:00:00", "", 0, 0, 0, 0);