commit ab1df17af9c0dd3b4d145139276c90ff8bf54a7d
author Jimi Chen <jimictw@google.com> Sat Feb 08 09:44:55 2025 +0000
committer Jimi Chen <jimictw@google.com> Mon Feb 10 09:30:55 2025 +0000
tree 687506205cd4fd33a8a98406c0a82c0aea78fcc2
parent b94a883a7c62e028c5fb9300aa181d08c0c0679d

Add helper function for updating IPv4 multicast addresses in APF tests

Test: atest NetworkStackTests
Change-Id: I0569d8cc15ebd1f4322721e84cc287cef6627b1c

tests/unit/src/android/net/apf/ApfFilterTest.kt

diff --git a/tests/unit/src/android/net/apf/ApfFilterTest.kt b/tests/unit/src/android/net/apf/ApfFilterTest.kt
index 0e3d8e1..f4f9d30 100644
--- a/tests/unit/src/android/net/apf/ApfFilterTest.kt
+++ b/tests/unit/src/android/net/apf/ApfFilterTest.kt
@@ -488,6 +488,11 @@
         return naPacket
     }
 
+    private fun updateIPv4MulticastAddrs(apfFilter: ApfFilter, mcastAddrs: List<Inet4Address>) {
+        doReturn(mcastAddrs).`when`(dependencies).getIPv4MulticastAddresses(any())
+        apfFilter.updateIPv4MulticastAddrs()
+    }
+
     @Test
     @IgnoreUpTo(Build.VERSION_CODES.UPSIDE_DOWN_CAKE)
     fun testV4EtherTypeAllowListFilter() {
@@ -4066,13 +4071,12 @@
         val addr = InetAddress.getByName("239.0.0.2") as Inet4Address
         mcastAddrs.add(addr)
         mcastAddrsExcludeAllHost.add(addr)
-        doReturn(mcastAddrs).`when`(dependencies).getIPv4MulticastAddresses(any())
-        apfFilter.updateIPv4MulticastAddrs()
+        updateIPv4MulticastAddrs(apfFilter, mcastAddrs)
         apfTestHelpers.consumeInstalledProgram(apfController, installCnt = 1)
         assertEquals(mcastAddrs.toSet(), apfFilter.mIPv4MulticastAddresses)
         assertEquals(mcastAddrsExcludeAllHost.toSet(), apfFilter.mIPv4McastAddrsExcludeAllHost)
 
-        apfFilter.updateIPv4MulticastAddrs()
+        updateIPv4MulticastAddrs(apfFilter, mcastAddrs)
         verify(apfController, never()).installPacketFilter(any(), any())
     }