1 files changed, 8 insertions, 3 deletions
diff --git a/libs/WifiTrackerLib/tests/src/com/android/wifitrackerlib/NetworkDetailsTrackerTest.java b/libs/WifiTrackerLib/tests/src/com/android/wifitrackerlib/NetworkDetailsTrackerTest.java
index eedd1a66c..ef2c003a2 100644
--- a/libs/WifiTrackerLib/tests/src/com/android/wifitrackerlib/NetworkDetailsTrackerTest.java
+++ b/libs/WifiTrackerLib/tests/src/com/android/wifitrackerlib/NetworkDetailsTrackerTest.java
@@ -17,6 +17,8 @@
 package com.android.wifitrackerlib;
 
 import static com.android.wifitrackerlib.NetworkDetailsTracker.createNetworkDetailsTracker;
+import static com.android.wifitrackerlib.StandardWifiEntry.ScanResultKey;
+import static com.android.wifitrackerlib.StandardWifiEntry.StandardWifiEntryKey;
 
 import static com.google.common.truth.Truth.assertThat;
 
@@ -39,6 +41,7 @@ import org.mockito.MockitoAnnotations;
 
 import java.time.Clock;
 import java.util.ArrayList;
+import java.util.Collections;
 
 public class NetworkDetailsTrackerTest {
 
@@ -108,9 +111,11 @@ public class NetworkDetailsTrackerTest {
      * StandardWifiEntry key is passed in.
      */
     @Test
-    public void testCreateNetworkDetailsTracker_returnsStandardNetworkDetailsTracker() {
-        final NetworkDetailsTracker tracker =
-                createTestNetworkDetailsTracker(StandardWifiEntry.KEY_PREFIX + "ssid,0");
+    public void testCreateNetworkDetailsTracker_returnsStandardNetworkDetailsTracker()
+            throws Exception {
+        final NetworkDetailsTracker tracker = createTestNetworkDetailsTracker(
+                new StandardWifiEntryKey(new ScanResultKey("ssid",
+                        Collections.singletonList(WifiEntry.SECURITY_NONE))).toString());
         assertThat(tracker).isInstanceOf(StandardNetworkDetailsTracker.class);
     }
 }