Python 实现自动化测试 dubbo 协议接口

前言

在工作或学习过程中，可能会遇到后端服务里有使用 dubbo 协议实现的接口，dubbo 协议接口的测试方法不同于 http/https 类型的接口，不能简单使用request.post的方法来完成自动化测试。

如果需要对 dubbo 协议的接口进行自动化测试，你会想到哪种解决方案呢？

今天的分享将提供一种方案由 Python 来实现 dubbo 协议的接口自动化测试。

准备工作：

安装Python3+的运行环境
使用python -m pip install telnetlib安装依赖包

PS：使用依赖包telnetlib是想通过 Python 来模拟 Linux 下对 dubbo 接口调用的命令行操作

Python 实现源码

import json
import telnetlib


class Dubbo(telnetlib.Telnet):
    prompt = 'dubbo>'
    
    def __init__(self, host=None, port=0):
      super().__init__(host, port)
      self.write(b'\n')
      
    def command(self, flag, str_=''):
      data = self.read_until(flag.encode())
      self.write(str_.encode() + b'\n')
      return data
      
    def invoke(self, service_name, method_name):
      command_str = "invoke {0}.{1}()".format(service_name, method_name)
      self.command(Dubbo.prompt, command_str)
      data = self.command(Dubbo.prompt, "")
      data = json.loads(data.decode(Dubbo.coding, errors='ignore').split('\n')[0].strip())
      

if __name__ == '__main__':
  conn = Dubbo('serviceIp', port)
  result = conn.invoke(
    "serviceName",
    "serviceName.method"
  )
  print(result)

源码重点详解

1、代码段：

command_str = “invoke {0}.{1}()”.format(service_name, method_name)

(1)、invoke 为调用 dubbo 接口的命令名称；

(2)、{0} 为 invoke 命令后需要接的第一个参数，它的值将由传入进来的 service_name 取代，一般是传入服务名称，这里的写法只是为了占位；

(3)、{1} 为 invoke 命令后需要接的第二个参数，它的值将由传入进来的 method_name 取代，一般是传入被调用的方法名称，这里的写法只是为了占位；

(4)、() 为 invoke 命令后调用 method_name 方法时需要的格式，类似 Python 中如果想要调用一个函数是需要加上 () 才能调用成功的，如果有参数还需要在 () 里传入对应的参数；

(5)、问题：如果被调用的dubbo服务的方法需要传入对应的参数，怎么处理呢？

答：

将代码改写为：“invoke {0}{1}({2})”.format(service_name,method_name,param_data)，需要在 () 里添加占位符，并在调用该方法时传入对应的参数，一般是 key:value 格式的参数，即 Python 中的字典格式的数据；

2、代码段：

conn = Dubbo(‘serviceIp’, port)