modified auth test case

This commit is contained in:
john 2020-06-14 01:14:09 +08:00
parent 072c71589f
commit cdeb482eab

View File

@ -14,25 +14,30 @@ import (
) )
func TestBasicAuth(t *testing.T) { func TestBasicAuth(t *testing.T) {
pairs := processAccounts(Accounts{ authMap := processAccounts(Accounts{
"admin": "password", "admin": "password",
"foo": "bar", "foo": "bar",
"bar": "foo", "bar": "foo",
}) })
assert.Len(t, pairs, 3)
assert.Contains(t, pairs, authPair{ assert.Len(t, authMap, 3)
user: "bar", assert.Contains(t,authMap,"Basic YmFyOmZvbw==")
value: "Basic YmFyOmZvbw==", assert.Contains(t,authMap,"Basic Zm9vOmJhcg==")
}) assert.Contains(t,authMap,"Basic YWRtaW46cGFzc3dvcmQ=")
assert.Contains(t, pairs, authPair{
user: "foo", //assert.Contains(t, pairs, authPair{
value: "Basic Zm9vOmJhcg==", // user: "bar",
}) // value: "Basic YmFyOmZvbw==",
assert.Contains(t, pairs, authPair{ //})
user: "admin", //assert.Contains(t, pairs, authPair{
value: "Basic YWRtaW46cGFzc3dvcmQ=", // user: "foo",
}) // value: "Basic Zm9vOmJhcg==",
//})
//assert.Contains(t, pairs, authPair{
// user: "admin",
// value: "Basic YWRtaW46cGFzc3dvcmQ=",
//})
} }
func TestBasicAuthFails(t *testing.T) { func TestBasicAuthFails(t *testing.T) {
@ -46,33 +51,33 @@ func TestBasicAuthFails(t *testing.T) {
} }
func TestBasicAuthSearchCredential(t *testing.T) { func TestBasicAuthSearchCredential(t *testing.T) {
pairs := processAccounts(Accounts{ authMap := processAccounts(Accounts{
"admin": "password", "admin": "password",
"foo": "bar", "foo": "bar",
"bar": "foo", "bar": "foo",
}) })
user, found := pairs.searchCredential(authorizationHeader("admin", "password")) user, found := authMap.searchCredential(authorizationHeader("admin", "password"))
assert.Equal(t, "admin", user) assert.Equal(t, "admin", user)
assert.True(t, found) assert.True(t, found)
user, found = pairs.searchCredential(authorizationHeader("foo", "bar")) user, found = authMap.searchCredential(authorizationHeader("foo", "bar"))
assert.Equal(t, "foo", user) assert.Equal(t, "foo", user)
assert.True(t, found) assert.True(t, found)
user, found = pairs.searchCredential(authorizationHeader("bar", "foo")) user, found = authMap.searchCredential(authorizationHeader("bar", "foo"))
assert.Equal(t, "bar", user) assert.Equal(t, "bar", user)
assert.True(t, found) assert.True(t, found)
user, found = pairs.searchCredential(authorizationHeader("admins", "password")) user, found = authMap.searchCredential(authorizationHeader("admins", "password"))
assert.Empty(t, user) assert.Empty(t, user)
assert.False(t, found) assert.False(t, found)
user, found = pairs.searchCredential(authorizationHeader("foo", "bar ")) user, found = authMap.searchCredential(authorizationHeader("foo", "bar "))
assert.Empty(t, user) assert.Empty(t, user)
assert.False(t, found) assert.False(t, found)
user, found = pairs.searchCredential("") user, found = authMap.searchCredential("")
assert.Empty(t, user) assert.Empty(t, user)
assert.False(t, found) assert.False(t, found)
} }