mirror of
https://gitee.com/Lyon1998/pikapython.git
synced 2025-01-15 17:02:53 +08:00
38 lines
1.0 KiB
Python
38 lines
1.0 KiB
Python
import socket
|
|
import random
|
|
import unittest
|
|
|
|
|
|
class TestUnittestAssertions(unittest.TestCase):
|
|
def testSocket(self):
|
|
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
host = "127.0.0.1"
|
|
port = 9999 + random.randint(0, 1000) + 1
|
|
print("port:", port)
|
|
server.bind((host, port))
|
|
server.listen(5)
|
|
|
|
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
client.connect((host, port))
|
|
|
|
accept, addr = server.accept()
|
|
self.assertEqual(addr, "127.0.0.1")
|
|
|
|
print("recv from client: %s" % str(addr))
|
|
|
|
client.send("send test from client".encode())
|
|
print("server recv:", accept.recv(1024).decode())
|
|
|
|
accept.send("send test from server".encode())
|
|
print("client recv:", client.recv(1024).decode())
|
|
|
|
accept.close()
|
|
client.close()
|
|
server.close()
|
|
|
|
|
|
suit = unittest.TestSuite("test1")
|
|
suit.addTest(TestUnittestAssertions())
|
|
runner = unittest.TextTestRunner()
|
|
res = runner.run(suit)
|