) var testParams = map[string]interface{}{ "username": "******", "password": "******", } // go test -test.run="^TestBasicAuth" func TestBasicAuth(t *testing.T) { TestingT(t) } type BasicAuthSuite struct { ba *BasicAuth } var _ = Suite(&BasicAuthSuite{ ba: NewBasicAuth(modules.ToModuleParams(testParams)), }) func (s *BasicAuthSuite) TestHappyPath(c *C) { mc := httputils.NewMockContext() mc.GetHttpRequest().Header = make(map[string][]string) mc.GetHttpRequest().Header.Set("Authorization", "Basic anRlZGlsbGE6cGFzcw==") resp, err := s.ba.ProcessRequest(mc) c.Assert(resp, IsNil) c.Assert(err, IsNil) } func (s *BasicAuthSuite) TestWrongCredentials(c *C) { mc := httputils.NewMockContext() mc.GetHttpRequest().Header = make(map[string][]string) mc.GetHttpRequest().Header.Set("Authorization", "Basic d3Jvbmc6d3Jvbmc=") //wrong:wrong
) var testParams = map[string]interface{}{ "allow": "127.0.0.1, :::1", "deny": "all", } // go test -test.run="^TestAccess$" func TestAccess(t *testing.T) { TestingT(t) } type AccessSuite struct { access *Access } var _ = Suite(&AccessSuite{ access: NewAccess(modules.ToModuleParams(testParams)), }) func (s *AccessSuite) TestHappyPath(c *C) { mc := httputils.NewMockContext() mc.GetHttpRequest().RemoteAddr = "127.0.0.1:80" resp, err := s.access.ProcessRequest(mc) c.Assert(resp, IsNil) c.Assert(err, IsNil) } func (s *AccessSuite) TestNotAllowedIP(c *C) { mc := httputils.NewMockContext() mc.GetHttpRequest().RemoteAddr = "127.0.0.2:80" resp, err := s.access.ProcessRequest(mc) c.Assert(resp, IsNil)