mirror of
https://github.com/schollz/cowyo.git
synced 2023-08-10 21:13:00 +03:00
68 lines
1.7 KiB
Go
68 lines
1.7 KiB
Go
package scram_test
|
|
|
|
import (
|
|
"crypto/sha1"
|
|
"testing"
|
|
|
|
. "gopkg.in/check.v1"
|
|
"gopkg.in/mgo.v2/internal/scram"
|
|
"strings"
|
|
)
|
|
|
|
var _ = Suite(&S{})
|
|
|
|
func Test(t *testing.T) { TestingT(t) }
|
|
|
|
type S struct{}
|
|
|
|
var tests = [][]string{{
|
|
"U: user pencil",
|
|
"N: fyko+d2lbbFgONRv9qkxdawL",
|
|
"C: n,,n=user,r=fyko+d2lbbFgONRv9qkxdawL",
|
|
"S: r=fyko+d2lbbFgONRv9qkxdawL3rfcNHYJY1ZVvWVs7j,s=QSXCR+Q6sek8bf92,i=4096",
|
|
"C: c=biws,r=fyko+d2lbbFgONRv9qkxdawL3rfcNHYJY1ZVvWVs7j,p=v0X8v3Bz2T0CJGbJQyF0X+HI4Ts=",
|
|
"S: v=rmF9pqV8S7suAoZWja4dJRkFsKQ=",
|
|
}, {
|
|
"U: root fe8c89e308ec08763df36333cbf5d3a2",
|
|
"N: OTcxNDk5NjM2MzE5",
|
|
"C: n,,n=root,r=OTcxNDk5NjM2MzE5",
|
|
"S: r=OTcxNDk5NjM2MzE581Ra3provgG0iDsMkDiIAlrh4532dDLp,s=XRDkVrFC9JuL7/F4tG0acQ==,i=10000",
|
|
"C: c=biws,r=OTcxNDk5NjM2MzE581Ra3provgG0iDsMkDiIAlrh4532dDLp,p=6y1jp9R7ETyouTXS9fW9k5UHdBc=",
|
|
"S: v=LBnd9dUJRxdqZiEq91NKP3z/bHA=",
|
|
}}
|
|
|
|
func (s *S) TestExamples(c *C) {
|
|
for _, steps := range tests {
|
|
if len(steps) < 2 || len(steps[0]) < 3 || !strings.HasPrefix(steps[0], "U: ") {
|
|
c.Fatalf("Invalid test: %#v", steps)
|
|
}
|
|
auth := strings.Fields(steps[0][3:])
|
|
client := scram.NewClient(sha1.New, auth[0], auth[1])
|
|
first, done := true, false
|
|
c.Logf("-----")
|
|
c.Logf("%s", steps[0])
|
|
for _, step := range steps[1:] {
|
|
c.Logf("%s", step)
|
|
switch step[:3] {
|
|
case "N: ":
|
|
client.SetNonce([]byte(step[3:]))
|
|
case "C: ":
|
|
if first {
|
|
first = false
|
|
done = client.Step(nil)
|
|
}
|
|
c.Assert(done, Equals, false)
|
|
c.Assert(client.Err(), IsNil)
|
|
c.Assert(string(client.Out()), Equals, step[3:])
|
|
case "S: ":
|
|
first = false
|
|
done = client.Step([]byte(step[3:]))
|
|
default:
|
|
panic("invalid test line: " + step)
|
|
}
|
|
}
|
|
c.Assert(done, Equals, true)
|
|
c.Assert(client.Err(), IsNil)
|
|
}
|
|
}
|