commit 8ddd3b9ffd347bf95414cf061a0a821948b9cace
author Erik Kline <ek@google.com> Thu Jul 02 16:08:15 2015 +0900
committer Erik Kline <ek@google.com> Thu Jul 30 12:56:56 2015 +0900
tree 478730300502725231ecfc81983584c07358a774
parent a209544b63ca766af379337bd97ca19091f631c2

Tests for candidate source address restriction

Bug: 20733156
Change-Id: I4118cc4c4dc8e9730bc95e681c14c29240f1af77

tests/net_test/srcaddr_selection_test.py

diff --git a/tests/net_test/srcaddr_selection_test.py b/tests/net_test/srcaddr_selection_test.py
index 0781b2d..b2f3152 100755
--- a/tests/net_test/srcaddr_selection_test.py
+++ b/tests/net_test/srcaddr_selection_test.py
@@ -36,6 +36,9 @@
 
 class IPv6SourceAddressSelectionTest(multinetwork_base.MultiNetworkBaseTest):
 
+  def SetIPv6Sysctl(self, ifname, sysctl, value):
+    self.SetSysctl("/proc/sys/net/ipv6/conf/%s/%s" % (ifname, sysctl), value)
+
   def SetDAD(self, ifname, value):
     self.SetSysctl("/proc/sys/net/ipv6/conf/%s/accept_dad" % ifname, value)
     self.SetSysctl("/proc/sys/net/ipv6/conf/%s/dad_transmits" % ifname, value)
@@ -115,10 +118,12 @@
     # [0]  Make sure DAD, optimistic DAD, and the use_optimistic option
     # are all consistently disabled at the outset.
     for netid in self.tuns:
-      self.SetDAD(self.GetInterfaceName(netid), 0)
-      self.SetOptimisticDAD(self.GetInterfaceName(netid), 0)
-      self.SetUseTempaddrs(self.GetInterfaceName(netid), 0)
-      self.SetUseOptimistic(self.GetInterfaceName(netid), 0)
+      ifname = self.GetInterfaceName(netid)
+      self.SetDAD(ifname, 0)
+      self.SetOptimisticDAD(ifname, 0)
+      self.SetUseTempaddrs(ifname, 0)
+      self.SetUseOptimistic(ifname, 0)
+      self.SetIPv6Sysctl(ifname, "use_oif_addrs_only", 0)
 
     # [1]  Pick an interface on which to test.
     self.test_netid = random.choice(self.tuns.keys())
@@ -301,5 +306,25 @@
 # TODO(ek): add tests listening for netlink events.
 
 
+class DefaultCandidateSrcAddrsTest(MultiInterfaceSourceAddressSelectionTest):
+
+  def testChoosesNonInterfaceSourceAddress(self):
+    self.SetIPv6Sysctl(self.test_ifname, "use_oif_addrs_only", 0)
+    src_ip = self.GetSourceIP(self.test_netid)
+    self.assertFalse(src_ip in [self.test_ip, self.test_lladdr])
+    self.assertTrue(src_ip in
+        [self.MyAddress(6, netid)
+         for netid in self.tuns if netid != self.test_netid])
+
+
+class RestrictedCandidateSrcAddrsTest(MultiInterfaceSourceAddressSelectionTest):
+
+  def testChoosesOnlyInterfaceSourceAddress(self):
+    self.SetIPv6Sysctl(self.test_ifname, "use_oif_addrs_only", 1)
+    # self.test_ifname does not have a global IPv6 address, so the only
+    # candidate is the existing link-local address.
+    self.assertAddressSelected(self.test_lladdr, self.test_netid)
+
+
 if __name__ == "__main__":
   unittest.main()