0

我想在 go lang 中模拟 memcache 缓存数据以避免授权我尝试使用 gomock 但无法解决，因为我没有任何接口。

func getAccessTokenFromCache(accessToken string)

func TestSendData(t *testing.T) {
mockCtrl := gomock.NewController(t)
defer mockCtrl.Finish()
mockObj := mock_utils.NewMockCacheInterface(mockCtrl)
mockObj.EXPECT().GetAccessToken("abcd") 
var jsonStr = []byte(`{
    "devices": [
        {"id": "avccc",

        "data":"abcd/"
        }
            ]
}`)
req, err := http.NewRequest("POST", "/send/v1/data", 
bytes.NewBuffer(jsonStr))
req.Header.Set("Content-Type", "application/json")
 req.Header.Set("Authorization", "d958372f5039e28")

rr := httptest.NewRecorder()
handler := http.HandlerFunc(SendData)
handler.ServeHTTP(rr, req)
if status := rr.Code; status != 200 {
    t.Errorf("handler returned wrong status code: got %v want %v",
        status, http.StatusOK)
}
expected := `{"error":"Invalid access token"}`
body, _ := ioutil.ReadAll(rr.Body)

if string(body) != expected {
    t.Errorf("handler returned unexpected body: got %v want %v",
        string(body), expected)
}




func SendData(w http.ResponseWriter, r *http.Request) {

accessToken := r.Header.Get(constants.AUTHORIZATION_HEADER_KEY)

t := utils.CacheType{At1: accessToken}
a := utils.CacheInterface(t)
isAccessTokenValid := utils.CacheInterface.GetAccessToken(a, accessToken)

if !isAccessTokenValid {
    RespondError(w, http.StatusUnauthorized, "Invalid access token")
    return
}
response := make(map[string]string, 1)
response["message"] = "success"
RespondJSON(w, http.StatusOK, response)

}

尝试使用 gomock 模拟

package mock_utils

gen mock for utils for get access controler (1) 定义一个你想模拟的接口。

(2) 使用mockgen从接口生成一个mock。(3) 在测试中使用mock：

4

1 回答 1

0

您需要构建您的代码，以便对服务的每次访问都通过接口实现发生。在您的情况下，理想情况下，您应该创建一个类似的界面

type CacheInterface interface {
      Set(key string, val interface{}) error
      Get(key string) (interface{},error)
}

您的 MemcacheStruct 应该实现这个接口，并且所有与 memcache 相关的调用都应该从那里发生。就像在您的情况下GetAccessToken应该调用cacheInterface.get(key)的那样，您的 cacheInterface 应该引用此接口的 memcache 实现。这是一种更好的方式来设计你的 go 程序，这不仅可以帮助你编写测试，而且如果你想使用不同的内存数据库来帮助缓存，它也会有所帮助。例如，假设将来您想使用 redis 作为缓存存储，那么您需要更改的只是创建此接口的新实现。

于 2018-04-06T08:53:27.053 回答