我正在使用labstack/echo网络服务器和gofight进行单元测试。在学习中,go想要知道是否存在用于访问(嵌入式)回波结构之外的状态的go习惯用法。例如:
type WebPlusDB struct {
web *echo.Echo
db *databaseInterface
}
func NewEngine() *echo.Echo {
e := echo.New()
e.GET("/hello", route_hello)
return e
}
func NewWebPlusDb() {
e := NewEngine()
db := database.New()
return WebPlusDB{e,db}
}
// for use in unit tests
func NewFakeEngine() *WebPlusDB {
e := NewEngine()
db := fakeDatabase.New()
return WebPlusDB{e,db}
}
func route_hello(c echo.Context) error {
log.Printf("Hello world\n")
// how do I access WebPlusDB.db from here?
return c.String(http.StatusOK, "hello world")
}
然后在我使用的测试代码中:
import (
"github.com/labstack/echo"
"github.com/appleboy/gofight"
"github.com/stretchr/testify/assert"
"testing"
)
func TestHelloWorld(t *testing.T) {
r := gofight.New()
r.GET("/hello").
Run(NewFakeEngine(), func(r gofight.HTTPResponse, rq gofight.HTTPRequest) {
assert.Equal(t, http.StatusOK, r.Code)
assert.Equal(t, "hello world", r.Body.String())
// test database access
})
}
最简单的解决方案是必须使用全局变量,而不是将echo嵌入“ WebPlusDB”中并在其中添加状态。我想要更好的封装。我想我应该使用类似WebPlusDB结构的东西,而不是echo.Echo加上全局状态。对于单元测试而言,这也许无关紧要,但是我想知道,在一个更完善的行事方式中(在这种情况下,避免使用全局变量)。
有没有解决方案,或者这是回声设计的弱点? 它具有中间件的扩展点,但数据库后端并不是真正的middleware as defined here。
注意:我在这里使用数据库来说明常见情况,但是可以是任何东西(我实际上在使用amqp)
您似乎可以扩展context接口,但是它在哪里创建?看起来它使用了一种 downcast :
e.GET("/", func(c echo.Context) error {
cc := c.(*CustomContext)
}
我认为(也许是错误地)只允许在接口和echo上使用。Context.Echo()返回的类型不是接口。
答案 0 :(得分:3)
您可以将实例方法作为函数值传递,这可能是处理此问题的最直接方法:
type WebPlusDB struct {
web *echo.Echo
db *databaseInterface
}
func (w WebPlusDB) route_hello(c echo.Context) error {
log.Printf("Hello world\n")
// do whatever with w
return c.String(http.StatusOK, "hello world")
}
func NewEngine() *echo.Echo {
e := echo.New()
w := NewWebPlusDb()
e.GET("/hello", w.route_hello)
return e
}