Socket又称"套接字"，应用程序通常通过"套接字"向网络发出请求或者应答网络请求，使主机间或者一台计算机上的进程间可以通讯。

Python 中，我们使用 socket 模块的 socket 函数来创建一个 socket 对象。语法格式如下：

socket.socket ( family ，type ，proto)

现在大多数用的都是websocket，那我们就先来安装一下websocket的安装包。

安装完之后，我们就开始我们的websocket之旅了。

在websocket里，我们有常用的这几个方法：

on_message方法

1. def on_message(ws, message):

2.   print(message)

on_message是用来接受消息的，server发送的所有消息都可以用on_message这个方法来收取。

on_error方法：

1. def on_error(ws, error):

2. print(error)

这个方法是用来处理错误异常的，如果一旦socket的程序出现了通信的问题，就可以被这个方法捕捉到。

on_open方法：

1. def on_open(ws):

2. def run(*args):

3. for i in range(30):

4. # send the message, then wait

5. # so thread doesn't exit and socket

6. # isn't closed

7. ws.send("Hello %d" % i)

8. time.sleep(1)

10. time.sleep(1)

11. ws.close()

12. print("Thread terminating...")

14. Thread(target=run).start()

on_open方法是用来保持连接的，上面这样的一个例子，就是保持连接的一个过程，每隔一段时间就会来做一件事,他会在30s内一直发送hello。最后停止。

on_close方法：

1. def on_close(ws):

2. print("### closed ###")

onclose主要就是关闭socket连接的。

如何创建一个websocket应用：

ws = websocket.WebSocketApp("wss://echo.websocket.org")

括号里面就是你要连接的socket的地址，在WebSocketApp这个实例化的方法里面还可以有其他参数，这些参数就是我们刚刚介绍的这些方法。

1. ws = websocket.WebSocketApp("ws://echo.websocket.org/",

2. on_message=on_message,

3. on_error=on_error,

4. on_close=on_close)

完整代码：

1. import websocket

2. from threading import Thread

3. import time

4. import sys

6. def on_message(ws, message):

7. print(message)

9. def on_error(ws, error):

10. print(error)

12. def on_close(ws):

13. print("### closed ###")

15. def on_open(ws):

16. def run(*args):

17. for i in range(3):

18. # send the message, then wait

19. # so thread doesn't exit and socket

20. # isn't closed

21. ws.send("Hello %d" % i)

22. time.sleep(1)

24. time.sleep(1)

25. ws.close()

26. print("Thread terminating...")

28. Thread(target=run).start()

31. if __name__ == "__main__":

33. websocket.enableTrace(True)

34. host = "ws://echo.websocket.org/"

35. ws = websocket.WebSocketApp(host,

36. on_message=on_message,

37. on_error=on_error,

38. on_close=on_close)

39. ws.on_open = on_open

40. ws.run_forever()

也可以使用python进行websocket的客户端压力测试。

1. # -*- coding:utf-8 -*-

2. # __author__ == 'chenmingle'

4. import websocket

5. import time

6. import threading

7. import json

8. import multiprocessing

9. import uuid

10. from threadpool import ThreadPool, makeRequests

12. # 修改成自己的websocket地址

13. WS_URL = "xxxx"

14. # 定义进程数

15. processes = 4

16. # 定义线程数（每个文件可能限制1024个，可以修改fs.file等参数）

17. thread_num = 700

18. index = 1

21. def on_message(ws, message):

22. # print(message)

23. pass

26. def on_error(ws, error):

27. print(error)

28. pass

31. def on_close(ws):

32. # print("### closed ###")

33. pass

36. def on_open(ws):

37. global index

38. index = index + 1

40. def send_thread():

41. # 设置你websocket的内容

42. # 每隔10秒发送一下数据使链接不中断

43. while True:

44. ws.send(u'hello服务器')

45. time.sleep(10)

47. t = threading.Thread(target=send_thread)

48. t.start()

51. def on_start(num):

52. time.sleep(5)

53. # websocket.enableTrace(True)

54. ws = websocket.WebSocketApp(WS_URL + str(num),

55. on_message=on_message,

56. on_error=on_error,

57. on_close=on_close)

58. ws.on_open = on_open

59. ws.run_forever()

62. def thread_web_socket():

63. # 线程池

64. pool_list = ThreadPool(thread_num)

65. num = list()

66. # 设置开启线程的数量

67. for ir in range(thread_num):

68. num.append(ir)

69. requests = makeRequests(on_start, num)

70. [pool_list.putRequest(req) for req in requests]

71. pool_list.wait()

74. if __name__ == "__main__":

75. # 进程池

76. pool = multiprocessing.Pool(processes=processes)

77. # 设置开启进程的数量

78. for i in xrange(processes):

79. pool.apply_async(thread_web_socket)

80. pool.close()

81. pool.join()

以上就是用Python进行websocket接口测试的详细内容。