add some tests for net (#1936)

Signed-off-by: Lan Liang <[email protected]>

Author: liangyuanpeng

kclvm/tests/test_units/runtime/kclvm_runtime.py

@@ -97,3 +97,4 @@ def Invoke(self, method: str, *args, **kwargs) -> any:
 if __name__ == "__main__":
     dylib = KclvmRuntimeDylib()
     dylib.Invoke(f"print", "hello kclvm")
+

kclvm/tests/test_units/runtime/net/test_net.py

@@ -2,23 +2,57 @@
 
 import typing
 import unittest
+import sys
+import os
 
+# Add the parent directory to the path to import kclvm_runtime
+parent_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+sys.path.append(parent_dir)
 import kclvm_runtime
 
 # https://github.com/python/cpython/blob/main/Lib/test
-
 _Dylib = kclvm_runtime.KclvmRuntimeDylib()
 
 
 class kclx_Net:
     def __init__(self, dylib_=None):
         self.dylib = dylib_ if dylib_ else _Dylib
-
+    def is_global_unicast_IP(self, value: str) -> bool:
+        return self.dylib.Invoke(f"net.is_global_unicast_IP", value)
+    def is_link_local_multicast_IP(self, value: str) -> bool:
+        return self.dylib.Invoke(f"net.is_link_local_multicast_IP", value)
+    def is_interface_local_multicast_IP(self, value: str) -> bool:
+        return self.dylib.Invoke(f"net.is_interface_local_multicast_IP", value)
+    def is_multicast_IP(self, value: str) -> bool:
+        return self.dylib.Invoke(f"net.is_multicast_IP", value)
+    def is_loopback_IP(self, value: str) -> bool:
+        return self.dylib.Invoke(f"net.is_loopback_IP", value)
+    def is_link_local_unicast_IP(self, value: str) -> bool:
+        return self.dylib.Invoke(f"net.is_link_local_unicast_IP", value)
+    def is_unspecified_IP(self, value: str) -> bool:
+        return self.dylib.Invoke(f"net.is_unspecified_IP", value)
+
+
+kclxnet = kclx_Net(_Dylib)
 
 class BaseTest(unittest.TestCase):
-    def test_foo(self):
-        self.assertTrue(True)
-
+    def test_is_interface_local_multicast_IP(self):
+        self.assertFalse(kclxnet.is_interface_local_multicast_IP("224.0.0.0"))
+        self.assertTrue(kclxnet.is_interface_local_multicast_IP("ff11::1"))
+    def test_is_link_local_multicast_IP(self):
+        self.assertTrue(kclxnet.is_link_local_multicast_IP("ff12::1"))
+    def test_is_global_unicast_IP(self):
+        self.assertTrue(kclxnet.is_global_unicast_IP("2607:f8b0:4005:802::200e"))
+        self.assertTrue(kclxnet.is_global_unicast_IP("64:ff9b::800:1"))
+        self.assertTrue(kclxnet.is_global_unicast_IP("220.181.108.89"))
+    def test_is_multicast_IP(self):
+        self.assertTrue(kclxnet.is_multicast_IP("239.255.255.255"))
+    def test_is_loopback_IP(self):
+        self.assertTrue(kclxnet.is_loopback_IP("127.0.0.1"))
+    def test_is_link_local_unicast_IP(self):
+        self.assertTrue(kclxnet.is_link_local_unicast_IP("fe80::2012:1"))
+    def test_is_unspecified_IP(self):
+        self.assertTrue(kclxnet.is_unspecified_IP("0.0.0.0"))
 
 if __name__ == "__main__":
     unittest.main()

test/grammar/builtins/net/is_ip_2/main.k

@@ -6,3 +6,9 @@ isip2 = net.is_link_local_multicast_IP("224.0.0.0")
 isip3 = net.is_link_local_unicast_IP("fe80::2012:1")
 isip4 = net.is_global_unicast_IP("220.181.108.89")
 isip5 = net.is_unspecified_IP("0.0.0.0")
+
+isip6 = net.is_interface_local_multicast_IP("224.0.0.0")
+isip7 = net.is_interface_local_multicast_IP("ff11::1")
+isip8 = net.is_link_local_multicast_IP("ff12::1")
+isip9 = net.is_global_unicast_IP("2607:f8b0:4005:802::200e")
+isip10 = net.is_global_unicast_IP("64:ff9b::800:1")

test/grammar/builtins/net/is_ip_2/stdout.golden

@@ -3,4 +3,9 @@ isip1: true
 isip2: true
 isip3: true
 isip4: true
-isip5: true
+isip5: true
+isip6: false
+isip7: true
+isip8: true
+isip9: true
+isip10: true