Mocking CGO function call for Unit testing

Question

I am using Tensorflow Lite API in C to infer results from a model file, for which code is written in a .c file across different functions. The functions include get_interpreter which creates a new interpreter, etc.

Now, this is all in C code. I am calling this C function in Go using CGo support in Go.

Go function (pseudocode):

package main

import "C"

func callCApi() error {
    // load data
    // get_interpreter returns an interpreter of type TfLiteInterpreter*
    interpreter = C.get_interpreter(param1, param2)
    // run inference
    // errors are returned if any during the above stages, else nil is returned
    return nil
}

I want to unit test the above Go code callCApi(), how do I achieve this?

I understand that the above C function get_interpreter must be mocked; how can this be mocked?

Chris · Accepted Answer · 2024-03-27 01:09:13Z

-1

I assume you want to mock callCApi.

For mocking functions without boring interface abstraction, you can try xgo.

Here is how xgo would solve your problem:

package main

import (
    "context"
    "fmt"
    "C"

    "github.com/xhd2015/xgo/runtime/core"
    "github.com/xhd2015/xgo/runtime/mock"
)

func callCApi() error {
    // load data
    // get_interpreter returns an interpreter of type TfLiteInterpreter*
    interpreter = C.get_interpreter(param1, param2)
    // run inference
    // errors are returned if any during the above stages, else nil is returned
    return nil
}


func main() {
    mock.Mock(callCApi, func(ctx context.Context, fn *core.FuncInfo, args core.Object, results core.Object) error {
        // do what you like on args and results
        return nil
    })
    // the control flow will be transferred to mock.Mock
    callCApi()   
}

Check https://github.com/xhd2015/xgo for details.

Disclaimer: creator of the xgo project after years of diving into go compiler.

edited Mar 27, 2024 at 1:09

Chris

139k139 gold badges316 silver badges293 bronze badges

answered Mar 25, 2024 at 10:38

xhd2015

871 gold badge1 silver badge6 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

xihare Over a year ago

I want to unit test callCApi for which get_interpreter must be mocked.

xhd2015 Over a year ago

It makes no difference.Wrap the C.get_interpreter in a go function, and mock that go function is enough.

Collectives™ on Stack Overflow

Mocking CGO function call for Unit testing

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related