Todo: 集成多平台 解决因SaiNiu线程抢占资源问题 本地提交测试环境打包 和 正式打包脚本与正式环境打包bat 提交Python32环境包 改进多日志文件生成情况修改打包日志细节

Utils/PythonNew32/Lib/site-packages/win32/test/test_win32wnet.py

@@ -0,0 +1,174 @@
+import unittest
+
+import netbios
+import win32api
+import win32wnet
+
+RESOURCE_CONNECTED = 0x00000001
+RESOURCE_GLOBALNET = 0x00000002
+RESOURCE_REMEMBERED = 0x00000003
+RESOURCE_RECENT = 0x00000004
+RESOURCE_CONTEXT = 0x00000005
+RESOURCETYPE_ANY = 0x00000000
+RESOURCETYPE_DISK = 0x00000001
+RESOURCETYPE_PRINT = 0x00000002
+RESOURCETYPE_RESERVED = 0x00000008
+RESOURCETYPE_UNKNOWN = 0xFFFFFFFF
+RESOURCEUSAGE_CONNECTABLE = 0x00000001
+RESOURCEUSAGE_CONTAINER = 0x00000002
+RESOURCEDISPLAYTYPE_GENERIC = 0x00000000
+RESOURCEDISPLAYTYPE_DOMAIN = 0x00000001
+RESOURCEDISPLAYTYPE_SERVER = 0x00000002
+RESOURCEDISPLAYTYPE_SHARE = 0x00000003
+
+
+NETRESOURCE_attributes = [
+    ("dwScope", int),
+    ("dwType", int),
+    ("dwDisplayType", int),
+    ("dwUsage", int),
+    ("lpLocalName", str),
+    ("lpRemoteName", str),
+    ("lpComment", str),
+    ("lpProvider", str),
+]
+
+NCB_attributes = [
+    ("Command", int),
+    ("Retcode", int),
+    ("Lsn", int),
+    ("Num", int),
+    #    ("Bufflen", int), - read-only
+    ("Callname", str),
+    ("Name", str),
+    ("Rto", int),
+    ("Sto", int),
+    ("Lana_num", int),
+    ("Cmd_cplt", int),
+    ("Event", int),
+    ("Post", int),
+]
+
+
+class TestCase(unittest.TestCase):
+    def testGetUser(self):
+        self.assertEqual(win32api.GetUserName(), win32wnet.WNetGetUser())
+
+    def _checkItemAttributes(self, item, attrs):
+        for attr, typ in attrs:
+            val = getattr(item, attr)
+            if typ is int:
+                self.assertTrue(
+                    isinstance(val, int), f"Attr {attr!r} has value {val!r}"
+                )
+                new_val = val + 1
+            elif typ is str:
+                if val is not None:
+                    # must be string
+                    self.assertTrue(
+                        isinstance(val, str), f"Attr {attr!r} has value {val!r}"
+                    )
+                    new_val = val + " new value"
+                else:
+                    new_val = "new value"
+            else:
+                self.fail(f"Don't know what {typ} is")
+            # set the attribute just to make sure we can.
+            setattr(item, attr, new_val)
+
+    def testNETRESOURCE(self):
+        nr = win32wnet.NETRESOURCE()
+        self._checkItemAttributes(nr, NETRESOURCE_attributes)
+
+    def testWNetEnumResource(self):
+        handle = win32wnet.WNetOpenEnum(RESOURCE_GLOBALNET, RESOURCETYPE_ANY, 0, None)
+        try:
+            while 1:
+                items = win32wnet.WNetEnumResource(handle, 0)
+                if len(items) == 0:
+                    break
+                for item in items:
+                    self._checkItemAttributes(item, NETRESOURCE_attributes)
+        finally:
+            handle.Close()
+
+    def testNCB(self):
+        ncb = win32wnet.NCB()
+        self._checkItemAttributes(ncb, NCB_attributes)
+
+    def testNetbios(self):
+        # taken from the demo code in netbios.py
+        ncb = win32wnet.NCB()
+        ncb.Command = netbios.NCBENUM
+        la_enum = netbios.LANA_ENUM()
+        ncb.Buffer = la_enum
+        rc = win32wnet.Netbios(ncb)
+        self.assertEqual(rc, 0)
+        for i in range(la_enum.length):
+            ncb.Reset()
+            ncb.Command = netbios.NCBRESET
+            ncb.Lana_num = la_enum.lana[i]
+            rc = win32wnet.Netbios(ncb)
+            self.assertEqual(rc, 0)
+            ncb.Reset()
+            ncb.Command = netbios.NCBASTAT
+            ncb.Lana_num = la_enum.lana[i]
+            ncb.Callname = b"*               "
+            adapter = netbios.ADAPTER_STATUS()
+            ncb.Buffer = adapter
+            win32wnet.Netbios(ncb)
+            # expect 6 bytes in the mac address.
+            self.assertTrue(len(adapter.adapter_address), 6)
+
+    def iterConnectableShares(self):
+        nr = win32wnet.NETRESOURCE()
+        nr.dwScope = RESOURCE_GLOBALNET
+        nr.dwUsage = RESOURCEUSAGE_CONTAINER
+        nr.lpRemoteName = "\\\\" + win32api.GetComputerName()
+
+        handle = win32wnet.WNetOpenEnum(RESOURCE_GLOBALNET, RESOURCETYPE_ANY, 0, nr)
+        while 1:
+            items = win32wnet.WNetEnumResource(handle, 0)
+            if len(items) == 0:
+                break
+            for item in items:
+                if item.dwDisplayType == RESOURCEDISPLAYTYPE_SHARE:
+                    yield item
+
+    def findUnusedDriveLetter(self):
+        existing = [
+            x[0].lower() for x in win32api.GetLogicalDriveStrings().split("\0") if x
+        ]
+        handle = win32wnet.WNetOpenEnum(RESOURCE_REMEMBERED, RESOURCETYPE_DISK, 0, None)
+        try:
+            while 1:
+                items = win32wnet.WNetEnumResource(handle, 0)
+                if len(items) == 0:
+                    break
+                xtra = [i.lpLocalName[0].lower() for i in items if i.lpLocalName]
+                existing.extend(xtra)
+        finally:
+            handle.Close()
+        for maybe in "defghijklmnopqrstuvwxyz":
+            if maybe not in existing:
+                return maybe
+        self.fail("All drive mappings are taken?")
+
+    def testAddConnection(self):
+        localName = self.findUnusedDriveLetter() + ":"
+        for share in self.iterConnectableShares():
+            share.lpLocalName = localName
+            win32wnet.WNetAddConnection2(share)
+            win32wnet.WNetCancelConnection2(localName, 0, 0)
+            break
+
+    def testAddConnectionOld(self):
+        localName = self.findUnusedDriveLetter() + ":"
+        for share in self.iterConnectableShares():
+            win32wnet.WNetAddConnection2(share.dwType, localName, share.lpRemoteName)
+            win32wnet.WNetCancelConnection2(localName, 0, 0)
+            break
+
+
+if __name__ == "__main__":
+    unittest.main()