Mocking a non-interface function

Question

I have a Go code something like this

func (r *Request) SetRequestMap(ctx *gin.Context, data map[string]interface{}) *Request {
    
    //Some processing code
     
     id, ok := r.map["id"]
    
    if !ok {
        return r
    }

    checkStatus := checkStatusOnline(ctx, id) // checkStatusOnline returns "on" if id is present or "off".
    // It make use of HTTP GET request internally to check if id is present or not. 
    // All json unmarshal is taken care of internally

    if checkStatus == "on" {
        r.map["val"] = "online"
    }

    return r
}

I want to write unit test case for SetRequestMap .

How can I mock checkStatusOnline without implementing any extra functions for mock?

Pass the function in as an argument – Jim Wright Jul 21 '20 at 14:39 — Jim Wright, Jul 21 '20 at 14:39

score 1 · Accepted Answer · answered Jul 21 '20 at 14:41

One way you can mock such functions is using function pointers:

var checkStatusOnline = defaultCheckStatusOnline

func defaultCheckStatusOnline(...) {... }

During a test run, you can set checkStatusOnline to a different implementation to test different scenarios.

func TestAFunc(t *testing.T) {
   checkStatusOnline=func(...) {... }
   defer func() {
      checkStatusOnline=defaultCheckStatusOnline
   }()
   ...
}

score 0 · Answer 2 · answered Jul 21 '20 at 14:42

You can do this to mock the function.

// Code

var checkStatusOnline = func(ctx context.Context, id int) int {
    ...
}

// Test

func TestSetRequestMap(t *testing.T) {
    tempCheckStatusOnline := checkStatusOnline
    checkStatusOnline = func(ctx context.Context, id int) int {
        // mock code
    }
    defer checkStatusOnline = tempCheckStatusOnline

    // Test here
}

Mocking a non-interface function

2 Answers2