我。嘲讽pool.cacheLock.Lock()
覆盖该分支的一种方法是模拟pool.cacheLock.Lock()，模拟版本可以将url 插入到地图中。所以在这个调用之后再次检查，会发现它并执行将进入第二个if 语句的主体。
使用接口模拟
模拟pool.cacheLock.Lock() 的一种方法是使pool.cacheLock 成为一个接口，并且在测试中您可以设置一个模拟值，其Lock() 方法将“脏插入”到地图中。
这是使用pool.cacheLock 接口的代码的简化版本：
type rwmutex interface {
Lock()
RLock()
RUnlock()
Unlock()
}
type fPool struct {
cache map[string]string
cacheLock rwmutex
}
func (pool *fPool) fetch(url string) string {
pool.cacheLock.RLock()
if rp, pres := pool.cache[url]; pres {
pool.cacheLock.RUnlock()
return rp
}
pool.cacheLock.RUnlock()
pool.cacheLock.Lock()
if rp, pres := pool.cache[url]; pres {
pool.cacheLock.Unlock()
// Skip adding url if someone snuck it in between RUnlock an Lock
return rp
}
rp := url + "~data"
pool.cache[url] = rp
pool.cacheLock.Unlock()
return rp
}
它的正常用法是：
pool := fPool{
cache: map[string]string{},
cacheLock: &sync.RWMutex{},
}
fmt.Println(pool.fetch("http://google.com"))
还有一个会触发第二个if正文的测试用例：
type testRwmutex struct {
sync.RWMutex // Embed RWMutex so we don't have to implement everything
customLock func()
}
func (trw *testRwmutex) Lock() {
trw.RWMutex.Lock()
if trw.customLock != nil {
trw.customLock()
}
}
func TestFPoolFetch(t *testing.T) {
trw := &testRwmutex{RWMutex: sync.RWMutex{}}
pool := &fPool{
cache: map[string]string{},
cacheLock: trw,
}
exp := "http://google.com~test"
trw.customLock = func() {
pool.cache["http://google.com"] = exp
}
if got := pool.fetch("http://google.com"); got != exp {
t.Errorf("Expected: %s, got: %s", exp, got)
}
}
使用函数字段进行模拟
模拟pool.cacheLock.Lock() 的另一种方法是将此功能“外包”给函数类型的字段，测试可以替换为一个函数，该函数除了调用它之外，还执行“脏插入”。
再次简化示例：
func NewFPool() *fPool {
mu := &sync.RWMutex{}
return &fPool{
cache: map[string]string{},
cacheLock: mu,
lock: mu.Lock,
}
}
type fPool struct {
cache map[string]string
cacheLock *sync.RWMutex
lock func()
}
func (pool *fPool) fetch(url string) string {
pool.cacheLock.RLock()
if rp, pres := pool.cache[url]; pres {
pool.cacheLock.RUnlock()
return rp
}
pool.cacheLock.RUnlock()
pool.lock()
if rp, pres := pool.cache[url]; pres {
pool.cacheLock.Unlock()
// Skip adding url if someone snuck it in between RUnlock an Lock
return rp
}
rp := url + "~data"
pool.cache[url] = rp
pool.cacheLock.Unlock()
return rp
}
正常用法是：
pool := NewFPool()
fmt.Println(pool.fetch("http://google.com"))
还有一个会触发第二个if正文的测试用例：
func TestFPoolFetch(t *testing.T) {
pool := NewFPool()
oldLock := pool.lock
exp := "http://google.com~test"
pool.lock = func() {
oldLock()
pool.cache["http://google.com"] = exp
}
if got := pool.fetch("http://google.com"); got != exp {
t.Errorf("Expected: %s, got: %s", exp, got)
}
}
二。使用简单的test 标志
这里的想法是，为了支持简单的测试，您可以在 fPool 的实现中构建一个简单的 test 标志（例如，它可以是 fPool 的字段），并且您要测试的代码故意检查这个标志：
type fPool struct {
cache map[string]string
cacheLock *sync.RWMutex
test bool
}
func (pool *fPool) fetch(url string) string {
pool.cacheLock.RLock()
if rp, pres := pool.cache[url]; pres {
pool.cacheLock.RUnlock()
return rp
}
pool.cacheLock.RUnlock()
pool.cacheLock.Lock()
if rp, pres := pool.cache[url]; pres || pool.test {
pool.cacheLock.Unlock()
// Skip adding url if someone snuck it in between RUnlock an Lock
return rp
}
rp := url + "~data"
pool.cache[url] = rp
pool.cacheLock.Unlock()
return rp
}
现在如果你想测试第二个if的身体，你要做的就是：
func TestFPoolFetch(t *testing.T) {
pool := NewFPool()
pool.test = true
exp := ""
if got := pool.fetch("http://google.com"); got != exp {
t.Errorf("Expected: %s, got: %s", exp, got)
}
}