Function TestRequirePluginEnabled

plugin/pluginenabled_test.go:13–42

(t *testing.T)

)

func TestRequirePluginEnabled(t *testing.T) {
	db := testdb.NewDBWithDefaultUser(t)
	conf := &model.PluginConf{
		ID:      1,
		UserID:  1,
		Enabled: true,
	}
	db.CreatePluginConf(conf)

	g := gin.New()

	mux := g.Group("/", requirePluginEnabled(1, db))

	mux.GET("/", func(c *gin.Context) {
		c.Status(200)
	})

	getCode := func() int {
		r := httptest.NewRequest("GET", "/", nil)
		w := httptest.NewRecorder()
		g.ServeHTTP(w, r)
		return w.Code
	}

	assert.Equal(t, 200, getCode())

	conf.Enabled = false
	db.UpdatePluginConf(conf)
	assert.Equal(t, 400, getCode())
}