如何动态的测试Thrift服务

​RPC协议与Thrift框架

RPC（Remote Procedure Call，远程过程调用）是一种通信协议，用于不同的进程或计算机之间进行通信和交互。它允许应用程序在本地或远程计算机上调用另一个应用程序中的函数或方法，就像调用本地函数一样。

RPC协议通过网络传输协议来实现不同计算机之间的通信，常见的网络传输协议有TCP、UDP等。当一个客户端需要调用远程服务器上的某个方法时，客户端会向服务器发送一个请求（包括请求参数），服务器接收到请求后执行相应的方法，并返回结果（包括返回值）。客户端接收到结果后继续执行自己的逻辑。

RPC协议的优点在于它可以隐藏底层通信细节，使得开发人员可以像调用本地函数一样调用远程函数，从而简化了分布式系统的开发。此外，RPC协议还具有高效、灵活、可扩展等优点，因此广泛应用于分布式系统、微服务架构等领域。

RPC（Remote Procedure Call）是一种通用的远程过程调用协议，而Thrift则是基于RPC协议开发的高效、跨语言和跨平台的服务框架。

Thrift是由Facebook开源的一个分布式服务框架，它使用IDL（Interface Definition Language）定义服务接口，然后通过自动生成代码的方式实现各种编程语言之间的互操作性。Thrift支持多种数据序列化和传输协议，包括二进制编码、JSON、XML等，还提供了多种服务器的实现方式和客户端库。

在Thrift的内部实现中，它使用了RPC协议来实现远程过程调用，并利用不同的编程语言的反射机制来对消息进行序列化和反序列化。通过使用RPC协议，Thrift可以在不同的计算机上运行服务，并允许客户端以本地服务的方式调用远程服务。此外，Thrift还提供了诸如负载均衡、故障转移、安全认证等方面的特性，使得分布式服务开发更加方便和高效。

实现一个Thrift服务

以下是一个使用Java和Thrift框架实现的简单订单查询服务示例。该服务包括一个接口，对应着查询订单信息的操作：

1. 定义Thrift服务的IDL文件（例如example.thrift），其中定义了服务名称、接口、方法、参数和返回值等信息：

namespace java example

struct Order {
    1: i32 orderId,
    2: string customerName,
    3: double amount
}

service OrderService {
    Order getOrder(1: i32 orderId)
}

上面的IDL文件中定义了一个结构体Order，表示订单信息，以及一个接口OrderService，包含了一个获取订单信息的方法getOrder。

2. 使用Thrift提供的工具生成Java代码（例如在命令行中执行thrift --gen java example.thrift）。这将生成与IDL文件中定义的服务接口和方法相对应的Java类和接口。

3. 实现服务接口的具体逻辑。例如，在服务器端实现一个订单查询服务：

public class OrderServiceImpl implements OrderService.Iface {    private Map orderMap = new HashMap<>();​    public OrderServiceImpl() {        // 初始化订单数据        for (int i = 1; i <= 10; i++) {            Order order = new Order(i, "Customer" + i, i * 100.0);            orderMap.put(order.orderId, order);        }    }​    @Override    public Order getOrder(int orderId) throws TException {        if (orderMap.containsKey(orderId)) {            return orderMap.get(orderId);        } else {            throw new TException("Order not found");        }    }​    public static void main(String[] args) throws Exception {        TServerTransport serverTransport = new TServerSocket(9090);        TProcessor processor = new OrderService.Processor(new OrderServiceImpl());        TServer server = new TSimpleServer(new TServer.Args(serverTransport).processor(processor));        System.out.println("Starting the server...");        server.serve();    }}

在上述代码中，使用Thrift提供的TProcessor将服务实现类与IDL文件中定义的接口绑定起来，并创建了一个TSimpleServer对象来启动服务器。在具体的业务逻辑中，查询订单信息时会遍历已有订单数据并查找对应订单信息。

4. 在客户端编写代码调用服务。例如：

public class OrderServiceClient {    public static void main(String[] args) throws Exception {        TTransport transport = new TSocket("localhost", 9090);        transport.open();        TProtocol protocol = new TBinaryProtocol(transport);        OrderService.Client client = new OrderService.Client(protocol);​        int orderId = 5;        try {            Order order = client.getOrder(orderId);            System.out.printf("Order #%d: %s, %.2f\n", order.orderId, order.customerName, order.amount);        } catch (TException e) {            System.out.println("Failed to get order: " + e.getMessage());        }​        transport.close();    }}

在客户端代码中，创建一个与服务器通信的TSocket对象，然后使用该对象创建TProtocol对象，并利用其创建OrderService.Client客户端代理对象。通过客户端代理对象调用服务器提供的服务接口和方法，并获取服务结果。

如何测试thrift服务

下面是一个基于TestNG的Thrift自动化测试框架设计思路，并给出对应的Java代码实现。

1. 模块划分

该框架可以划分为以下几个模块：

- 测试套件模块：定义测试用例，包括测试方法、输入参数和预期结果等。

- 测试数据模块：提供测试所需的数据，例如订单信息、客户信息等。这些数据可以从文件、数据库或其他来源中读取。

- 测试环境模块：创建并初始化测试环境，包括启动测试服务器和创建Thrift客户端代理对象等。

- 测试执行模块：执行测试用例，并进行结果断言和记录。可以使用Mock对象模拟测试环境，从而避免对真实环境造成影响。

- 测试报告模块：生成测试报告，展示测试的结果和统计信息等。

2. 实现方式

在Java中，可以通过以下方式来实现上述模块：

- 使用TestNG注解定义测试用例（例如@Test）和测试套件（例如@Suite）。

- 创建POJO类，表示订单信息、客户信息等测试数据。

- 使用@BeforeSuite、@AfterSuite、@BeforeTest、@AfterTest等注解，在适当的时机创建和清除测试环境。

- 在测试方法中，通过Thrift客户端代理对象调用服务接口和方法，并对返回结果进行断言和记录。

- 使用TestNG提供的Reporter API来生成测试报告。

下面是一个基于TestNG的Thrift自动化测试框架示例代码，假设服务接口为example.OrderService：

import example.Order;import example.OrderService;import org.apache.thrift.protocol.TBinaryProtocol;import org.apache.thrift.transport.TSocket;import org.testng.annotations.*;​import java.util.Random;​public class OrderServiceTest {    private static final int PORT = 9090;    private static OrderService.Client client;​    @BeforeSuite    public void setUp() throws Exception {        // 创建Thrift客户端代理对象        TSocket socket = new TSocket("localhost", PORT);        socket.open();        client = new OrderService.Client(new TBinaryProtocol(socket));    }​    @AfterSuite    public void tearDown() throws Exception {        // 关闭Thrift客户端        if (client != null) {            client.getInputProtocol().getTransport().close();        }    }​    @DataProvider(name = "orderData")    public Object[][] createOrderData() {        // 创建订单数据        return new Object[][]{                {new Order(1, "Customer1", 100)},                {new Order(2, "Customer2", 200)},                {new Order(3, "Customer3", 300)}        };    }​    @Test(dataProvider = "orderData")    public void testGetOrder(Order order) throws Exception {        // 测试用例1：查询已有订单信息        Order result = client.getOrder(order.orderId);        assertOrderEquals(order, result);​        // 测试用例2：查询不存在的订单信息        try {            client.getOrder(0);        } catch (Exception e) {            assertErrorMessageEquals("Order not found", e.getMessage());        }    }​    // 添加更多测试用例...​    private void assertOrderEquals(Order expected, Order actual) {        // 比较订单信息是否相等    }​    private void assertErrorMessageEquals(String expected, String actual) {        // 比较错误消息是否相等    }}

在上述代码中，使用TestNG提供的注解@BeforeSuite和@AfterSuite分别创建和关闭Thrift客户端代理对象，并使用@DataProvider注解定义测试数据。在@Test注解的测试方法中，通过Thrift客户端代理对象调用服务方法，并对返回结果进行断言。此外，还可以结合持续集成工具（例如Jenkins）和自动化测试框架（例如Selenium或Appium）来实现部署、测试和反馈的自动化流程。

往期推荐

聊聊工作中的自我管理和向上管理

经验分享｜测试工程师转型测试开发历程

聊聊UI自动化的PageObject设计模式

细读《阿里测试之道》

我在阿里做测开