forked from quic-go/quic-go
36 lines
740 B
Go
36 lines
740 B
Go
package handshake
|
|
|
|
import (
|
|
"net"
|
|
"time"
|
|
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
)
|
|
|
|
var _ = Describe("STK Generator", func() {
|
|
var stkGen *STKGenerator
|
|
|
|
BeforeEach(func() {
|
|
var err error
|
|
stkGen, err = NewSTKGenerator()
|
|
Expect(err).ToNot(HaveOccurred())
|
|
})
|
|
|
|
It("generates an STK", func() {
|
|
ip := net.IPv4(127, 0, 0, 1)
|
|
token, err := stkGen.NewToken(ip)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(token).ToNot(BeEmpty())
|
|
})
|
|
|
|
It("verifies an STK", func() {
|
|
ip := net.IPv4(192, 168, 0, 1)
|
|
token, err := stkGen.NewToken(ip)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
t, err := stkGen.VerifyToken(ip, token)
|
|
Expect(err).ToNot(HaveOccurred())
|
|
Expect(t).To(BeTemporally("~", time.Now(), time.Second))
|
|
})
|
|
})
|