测试人必会:Python带你上手WebSocket

测试人必会:Python带你上手WebSocket

关于Socket那些事

Socket,即网络套接字,是双向通信通道的端点(是抽象的)。

套接字可以在一个进程内、同一台机器上的进程之间,或者在不同机器上的进程之间进行通信。

网络套接字可以通过多种不同的通道类型得以实现,例如TCP、UDP等。

换句话说,网络套接字,是在计算机网络中,两个运行程序之间通信流中的一个端点。这个端点是一个虚拟抽象的概念,并不存在任何硬件。

网络套接字可以通过IP地址和端口号的组合来唯一标识。下面我们对于Socket套接字相关术语做简单介绍,以便于后续内容的理解。

Domain(域)

作为网络中的传输机制,域是协议中的一员。这些域的值是常量,例如AF_INET、PF_INET、PF_UNIX、PF_X25等,这些常量都属于地址系列,用于指定套接字可以与之通信的地址类型。

Type(类型)

表示两个端点之间的通信类型,通常有SOCK_STREAM和SOCK_DGRAM。

SOCK_STREAM

基于TCP,数据传输比较有保障,是有保障的面向连接的Socket,即能保证数据正确传送到对方;基于数据流的传输;使用TCP/IP协议的网络编程。

SOCK_DGRAM

基于UDP,用于无向连接协议,是无保障的面向消息的socket,主要用于在网络上发放广播信息;基于数据包的传输;使用UDP协议的网络编程。

Protocol(协议)

用于标识域和类型内的协议变形。它的默认值是0,通常被省略。

Hostname(主机名)

网络接口的标识符。主机名可以是字符串、点分四组地址或冒号(也可能是点)表示法的IPV6地址。

Port(端口)

每个服务器监听调用一个或多个客户端的端口。

PythonSocket模块

在Python中实现Socket编程,就需要用到Socket模块。我们可以通过以下语法创建Socket:

# 导入socket库,创建一个简单的socket
import socket
s = socket.socket (socket_family, socket_type, protocol = 0)

创建Socket时,可以使用不同参数(相关概念可参见上文):

  • socket_family:AF_UNIX或AF_INET
  • socket_type:SOCK_STREAM或SOCK_DGRAM
  • protocol:这通常被忽略,默认为0

SocketMethods套接字方法

Python中为套接字提供了三种不同的方法,分别是"服务端套接字方法"、“客户端套接字方法”、“通用套接字方法”。

服务端套接字方法

在客户端-服务器架构中(C/S架构),有一个*服务器,来提供服务,众多不同的客户端则从该*服务器接收服务。众多不同客户端也会向该*服务器发出请求。

C/S架构中的一些重要服务器套接字方法有:

socket.bind()

将服务器地址(主机名,端口号)绑定到套接字。

socket.listen()

监听与套接字的连接,该方法启动了TCP监听器;此方法中的参数用于指定排队链接的最大数量(最小值为0,最大值为5)。

socket.accept()

接受TCP客户端连接,该方法将返回一对值(conn,address),其中conn是一个新的套接字对象,用于发送和接受数据;address是绑定到套接字的地址。在使用该方法之前,必须先使用socket.bind()和socket.listen()这两个方法。

客户端套接字方法

在客户端-服务器架构中(C/S架构),客户端会请求服务器,并且也会从服务器接收服务。

socket.connect(address)

该方法主动建立服务器连接,简而言之,该方法的作用就是将客户端连接到服务器。

通用套接字方法

除了客户端和服务端套接字方法,还有一些通用的套接字方法,这些方法在Socket套接字编程中也非常有用。

常用的通用套接字方法有:

.recv(bufsize)

用于从套接字接收TCP消息;

参数bufsize代表缓冲区大小,定义了该方法在任何时候可以接收的最大数据量。

socket.send(bytes)

用于将数据发送到连接着远程机器的套接字上;

参数bytes表示发送到套接字的字节数。

socket.recvfrom(data,address)

用于从套接字接收数据;

该方法返回两对值(data、address):其中,data表示接收到的数据,address表示发送数据的套接字地址。

socket.sendto(data,address)

用于向套接字发送数据;

该方法返回两对值(data、address):其中,data表示发送的字节数,address表示远程机器的地址。

socket.close()

用于关闭套接字。

socket.gethostname()

返回主机的名称。

socket.sendall(data)

  • 将所有数据发送到连接着远程机器的套接字,直到发生错误;
  • 期间,如果发生错误,则使用socket.close()方法关闭套接字。

建立连接

为了能够在服务器和客户端之间建立连接,我们需要通过Python编写两个程序,分别用于服务器和客户端。

服务端程序

socket.bind()方法

在服务端Socket套接字程序中,通过使用socket.bind()方法,将其绑定到指定的IP地址及端口号上,这样一来,就可以监听到传入该IP和端口上的请求。

socket.listen()方法

然后,使用socket.listen()方法将服务器设置为监听模式。

该方法有一个参数,表示当服务器正忙时,当前允许几个连接保持等待,例如socket.listen(2),则表示当前允许2个连接保持等待,如果第3个套接字尝试连接,则该连接请求会被拒绝。

socket.send()方法

使用socket.send()方法向客户端发送消息。

socket.accept方法

使用socket.accept()方法启动连接。

socket.close()方法

使用socket.close()方法关闭连接。

服务端程序_Demo:

import socket
def Server_side():
 host = socket.gethostname()
 port = 23333
 serversocket = socket.socket()
 serversocket.bind((host, port))
 serversocket.listen(1)
 print('socket 监听中')
 while True:
 conn, addr = serversocket.accept()
 print("收到连接请求 %s" % str(addr))
 msg = '连接已创建' + "\r\n"
 conn.send(msg.encode('ascii'))
 conn.close()
if __name__ == '__main__':
 Server_side()

客户端程序

创建socket对象

在客户端socket程序中,需要先创建一个socket对象。然后将该socket对象连接到正在运行着的服务器端口(即上面的23333端口)。

socket.connect()方法

使用socket.connect()方法建立连接。

socket.recv()方法

socket.recv()方法用于客户端接收来自服务器的消息。

socket.close()方法

socket.close()方法关闭客户端。

客户端程序_Demo:

import socket
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
host = socket.gethostname()
port = 23333
s.connect((host, port))
msg = s.recv(1024) s.close()
print (msg.decode('ascii'))

分别启动服务端和客户端程序,在运行服务端程序后,在终端上获得以下输出:

socket is listening
Got connection from ('XXX.XX.XXX.XXX', 65426) # 此处IP为客户端IP

在运行客户端程序后,在终端上获得以下输出:

Connecting Established 

网络套接字异常处理

try和except,可用于处理网络套接字异常,示例脚本如下:

import socket
host = socket.gethostname()
port = 23333
# 创建套接字对象 s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
try:
 # 绑定服务端IP及端口
 s.bind((host,port))
 
 # 设置客户端的等待时间
 s.settimeout(3)
 data, addr = s.recvfrom(1024)
 print ("recevied from ",addr)
 print ("obtained ", data)
 s.close()
except socket.timeout :
 print ("No connection between client and server")
 s.close()
 

脚本解析:

(1)通过当前服务端IP及指定端口号(23333),创建了一个套接字对象。

(2)在 try - except 异常处理块中,使用 socket.bind() 方法,尝试绑定服务端IP及端口;

(3)使用 socket.settimeout() 方法设置客户端的等待时间,示例中设置为3秒;

(4)在except块中对异常进行处理,如果服务器和客户端之间没有建立连接,将打印一条消息。

以上程序运行后,在终端上获得以下输出:

No connection between client and server

总结

Web Socket是测试人员知识储备库中不可或缺的一项,本次分享对Web Socket来龙去脉做了梳理,结合Python进行了实战演练。希望通过本文的学习,能够加深你对Web Socket认知,让它不再成为你的知识盲区。

下面是配套资料,对于做【软件测试】的朋友来说应该是最全面最完整的备战仓库,这个仓库也陪伴我走过了最艰难的路程,希望也能帮助到你!
测试人必会:Python带你上手WebSocket

最后: 可以在公众号:伤心的辣条 ! 免费领取一份216页软件测试工程师面试宝典文档资料。以及相对应的视频学习教程免费分享!,其中包括了有基础知识、Linux必备、Shell、互联网程序原理、Mysql数据库、抓包工具专题、接口测试工具、测试进阶-Python编程、Web自动化测试、APP自动化测试、接口自动化测试、测试高级持续集成、测试架构开发测试框架、性能测试、安全测试等。

学习不要孤军奋战,最好是能抱团取暖,相互成就一起成长,群众效应的效果是非常强大的,大家一起学习,一起打卡,会更有学习动力,也更能坚持下去。你可以加入我们的测试技术交流扣扣群:914172719(里面有各种软件测试资源和技术讨论)

喜欢软件测试的小伙伴们,如果我的博客对你有帮助、如果你喜欢我的博客内容,请 “点赞” “评论” “收藏” 一键三连哦!


好文推荐

转行面试,跳槽面试,软件测试人员都必须知道的这几种面试技巧!

面试经:一线城市搬砖!又面软件测试岗,5000就知足了…

面试官:工作三年,还来面初级测试?恐怕你的软件测试工程师的头衔要加双引号…

什么样的人适合从事软件测试工作?

那个准点下班的人,比我先升职了…

测试岗反复跳槽,跳着跳着就跳没了…

上一篇:转发自 python 接口自动化测试之 websocket 接口.


下一篇:websocket 接口如何测试?