1 files changed, 120 insertions, 0 deletions
diff --git a/test/tuntap/test_interface.py b/test/tuntap/test_interface.py
new file mode 100644
index 0000000..7cf19b2
--- /dev/null
+++ b/test/tuntap/test_interface.py
@@ -0,0 +1,120 @@
+# Copyright (c) 2011 Mattias Nissler <mattias.nissler@gmx.de>
+#
+# Redistribution and use in source and binary forms, with or without modification, are permitted
+# provided that the following conditions are met:
+#
+#   1. Redistributions of source code must retain the above copyright notice, this list of
+#      conditions and the following disclaimer.
+#   2. Redistributions in binary form must reproduce the above copyright notice, this list of
+#      conditions and the following disclaimer in the documentation and/or other materials provided
+#      with the distribution.
+#   3. The name of the author may not be used to endorse or promote products derived from this
+#      software without specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES,
+# INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
+# PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT,
+# INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
+# TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+import errno
+import socket
+import unittest
+
+from tuntap.char_dev_harness import TunCharDevHarness, TapCharDevHarness
+from tuntap.interface_harness import Address, InterfaceHarness
+from tuntap.sockaddr import SockaddrDl, SockaddrIn, SockaddrIn6
+from tuntap.tun_tap_harness import TunHarness, TapHarness
+
+class TestInterface(unittest.TestCase):
+
+    def __init__(self, name, harness):
+        super(TestInterface, self).__init__(name)
+        self.harness = harness
+
+    def setUp(self):
+        self.harness.start()
+
+    def tearDown(self):
+        self.harness.stop()
+
+    def test_CloseWhileUp(self):
+        self.harness.interface.flags |= InterfaceHarness.IFF_UP
+        self.harness.char_dev.close()
+        self.harness.start()
+
+    def test_UpDown(self):
+        self.harness.interface.flags |= InterfaceHarness.IFF_UP
+        self.assertEquals(InterfaceHarness.IFF_UP,
+                          self.harness.interface.flags & InterfaceHarness.IFF_UP)
+        self.harness.interface.flags &= ~InterfaceHarness.IFF_UP
+        self.assertEquals(0,
+                          self.harness.interface.flags & InterfaceHarness.IFF_UP)
+
+    def test_NetmaskAFFix(self):
+        self.harness.interface.addIfAddr(local = self.harness.addr.sa_local,
+                                         dst = self.harness.addr.sa_dst,
+                                         mask = SockaddrIn(af = 0, addr = self.harness.addr.mask))
+        for addr in self.harness.interface.getAddrs(socket.AF_INET):
+            if addr[1] == self.harness.addr.sa_mask:
+                return;
+        self.fail()
+
+    def test_Address(self):
+        self.harness.interface.addIfAddr(local = self.harness.addr.sa_local,
+                                         dst = self.harness.addr.sa_dst,
+                                         mask = self.harness.addr.sa_mask)
+        for addr in self.harness.interface.getAddrs(socket.AF_INET):
+            if (addr[0] == self.harness.addr.sa_local and
+                addr[1] == self.harness.addr.sa_mask and
+                addr[2] == self.harness.addr.sa_dst):
+                return
+        self.fail()
+
+    def test_Address6(self):
+        def compare(expected, actual):
+            return (expected or SockaddrIn6(af = 0, addr = None)) == actual
+
+        self.harness.interface.addIfAddr6(local = self.harness.addr6.sa_local,
+                                          dst = self.harness.addr6.sa_dst,
+                                          mask = self.harness.addr6.sa_mask)
+        for addr in self.harness.interface.getAddrs(socket.AF_INET6):
+            if (compare(addr[0], self.harness.addr6.sa_local) and
+                compare(addr[1], self.harness.addr6.sa_mask) and
+                compare(addr[2], self.harness.addr6.sa_dst)):
+                return
+        self.fail()
+
+
+class TestTunInterface(TestInterface):
+
+    def __init__(self, name):
+        super(TestTunInterface, self).__init__(name, TunHarness())
+
+    def test_Flags(self):
+        self.assertEquals(InterfaceHarness.IFF_POINTOPOINT |
+                          InterfaceHarness.IFF_RUNNING |
+                          InterfaceHarness.IFF_SIMPLEX |
+                          InterfaceHarness.IFF_MULTICAST,
+                          self.harness.interface.flags)
+
+
+class TestTapInterface(TestInterface):
+
+    def __init__(self, name):
+        super(TestTapInterface, self).__init__(name, TapHarness())
+
+    def test_Flags(self):
+        self.assertEquals(InterfaceHarness.IFF_BROADCAST |
+                          InterfaceHarness.IFF_RUNNING |
+                          InterfaceHarness.IFF_SIMPLEX |
+                          InterfaceHarness.IFF_MULTICAST,
+                          self.harness.interface.flags)
+
+    def test_SetLladdr(self):
+        addr = SockaddrDl(name = '', addr = '\x11\x22\x33\x44\x55\x66', type = 0)
+        self.harness.interface.lladdr = addr
+        self.assertEquals(addr.addr, self.harness.interface.lladdr.addr)