1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283 |
- // Copyright (c) 2018 Shivaram Lingamneni <slingamn@cs.stanford.edu>
- // released under the MIT license
-
- package utils
-
- import (
- "testing"
- )
-
- const (
- storedToken = "1e82d113a59a874cccf82063ec603221"
- badToken = "1e82d113a59a874cccf82063ec603222"
- shortToken = "1e82d113a59a874cccf82063ec60322"
- longToken = "1e82d113a59a874cccf82063ec6032211"
- )
-
- func TestGenerateSecretToken(t *testing.T) {
- token := GenerateSecretToken()
- if len(token) != SecretTokenLength {
- t.Errorf("bad token: %v", token)
- }
- }
-
- func TestTokenCompare(t *testing.T) {
- if !SecretTokensMatch(storedToken, storedToken) {
- t.Error("matching tokens must match")
- }
-
- if SecretTokensMatch(storedToken, badToken) {
- t.Error("non-matching tokens must not match")
- }
-
- if SecretTokensMatch(storedToken, shortToken) {
- t.Error("non-matching tokens must not match")
- }
-
- if SecretTokensMatch(storedToken, longToken) {
- t.Error("non-matching tokens must not match")
- }
-
- if SecretTokensMatch("", "") {
- t.Error("the empty token should not match anything")
- }
-
- if SecretTokensMatch("", storedToken) {
- t.Error("the empty token should not match anything")
- }
- }
-
- func TestMunging(t *testing.T) {
- count := 131072
- set := make(map[string]bool)
- var token string
- for i := 0; i < count; i++ {
- token = GenerateSecretToken()
- set[token] = true
- }
- // all tokens generated thus far should be unique
- assertEqual(len(set), count, t)
-
- // iteratively munge the last generated token an additional `count` times
- mungedToken := token
- for i := 0; i < count; i++ {
- mungedToken = MungeSecretToken(mungedToken)
- assertEqual(len(mungedToken), len(token), t)
- set[mungedToken] = true
- }
- // munged tokens should not collide with generated tokens, or each other
- assertEqual(len(set), count*2, t)
- }
-
- func BenchmarkGenerateSecretToken(b *testing.B) {
- for i := 0; i < b.N; i++ {
- GenerateSecretToken()
- }
- }
-
- func BenchmarkMungeSecretToken(b *testing.B) {
- t := GenerateSecretToken()
- for i := 0; i < b.N; i++ {
- t = MungeSecretToken(t)
- }
- }
|