如何单元测试调用TensorFlow Lite C API的Go CGO代码并Mock函数

我用C语言的TensorFlow Lite API实现了模型推理逻辑，相关代码分散在多个.c文件中，其中包含创建解释器的get_interpreter函数。目前通过Go的CGO特性调用这个C函数，Go端的调用逻辑伪代码如下：

package main

import "C"

func callCApi() error {
    // 加载数据
    // get_interpreter返回TfLiteInterpreter*类型的解释器
    interpreter := C.get_interpreter(param1, param2)
    // 执行推理
    // 若上述阶段出现错误则返回错误，否则返回nil
    return nil
}

我想对callCApi()函数做单元测试，需要Mockget_interpreter这个C函数，具体该怎么实现？

直接调用CGO的C函数无法直接Mock，核心思路是通过Go接口抽象C函数依赖，再用Mock实现替换真实调用，具体步骤如下：

1. 抽象C函数依赖为Go接口 #

把C函数的调用逻辑封装成Go接口，隔离CGO依赖：

package main

import "C"

// InterpreterCreator 定义创建解释器的接口
type InterpreterCreator interface {
    CreateInterpreter(param1, param2 int) *C.TfLiteInterpreter
}

// RealCreator 真实的C函数调用实现
type RealCreator struct{}

func (r *RealCreator) CreateInterpreter(param1, param2 int) *C.TfLiteInterpreter {
    return C.get_interpreter(C.int(param1), C.int(param2))
}

2. 重构callCApi函数，依赖接口 #

修改原函数，让它接收InterpreterCreator接口作为参数，避免直接调用C函数：

import "fmt"

func callCApi(creator InterpreterCreator) error {
    // 示例：假设param1、param2是加载后的数据
    param1 := 100
    param2 := 200
    interpreter := creator.CreateInterpreter(param1, param2)
    
    // 模拟推理前的错误判断：解释器为nil则返回错误
    if interpreter == nil {
        return fmt.Errorf("failed to create interpreter")
    }
    
    // 执行推理（真实逻辑中调用C的推理函数）
    // ...
    
    return nil
}

真实业务场景中，只需传入&RealCreator{}即可调用真实C函数。

3. 实现Mock的InterpreterCreator #

在测试文件中定义Mock结构体，模拟成功/失败的创建场景：

package main

import "C"

// MockCreator 用于测试的Mock解释器创建器
type MockCreator struct {
    shouldFail bool // 控制是否返回nil解释器（模拟创建失败）
}

func (m *MockCreator) CreateInterpreter(param1, param2 int) *C.TfLiteInterpreter {
    if m.shouldFail {
        return nil // 模拟创建失败
    }
    // 模拟成功创建的解释器（仅需非空指针，无需真实实例）
    return (*C.TfLiteInterpreter)(uintptr(1))
}

4. 编写单元测试 #

用MockCreator验证callCApi的不同逻辑分支：

package main

import "testing"

func TestCallCApi_Success(t *testing.T) {
    mock := &MockCreator{shouldFail: false}
    err := callCApi(mock)
    if err != nil {
        t.Fatalf("预期无错误，实际得到: %v", err)
    }
}

func TestCallCApi_Failure(t *testing.T) {
    mock := &MockCreator{shouldFail: true}
    err := callCApi(mock)
    if err == nil {
        t.Fatal("预期创建解释器失败时返回错误，实际得到nil")
    }
    if err.Error() != "failed to create interpreter" {
        t.Fatalf("错误信息不符合预期: %v", err)
    }
}

额外说明 #

• 如果推理逻辑还依赖其他C函数（比如执行推理的函数），可以用同样的方式抽象为接口，逐步隔离所有CGO依赖。
• 测试中无需构造真实的TfLiteInterpreter实例，只需模拟指针的空/非空状态即可——单元测试的核心是验证Go代码的逻辑，而非C代码的正确性。

内容的提问来源于stack exchange，提问作者xihare