mirror of
https://github.com/cloudflare/redoctober.git
synced 2026-01-06 21:47:37 +00:00
82 lines
1.4 KiB
Go
82 lines
1.4 KiB
Go
// keycache_test.go: tests for keycache.go
|
|
//
|
|
// Copyright (c) 2013 CloudFlare, Inc.
|
|
package keycache
|
|
|
|
import (
|
|
"github.com/cloudflare/redoctober/passvault"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
var now = time.Now()
|
|
var nextYear = now.AddDate(1, 0, 0)
|
|
var emptyKey = make([]byte, 16)
|
|
var dummy = make([]byte, 16)
|
|
|
|
func TestUsesFlush(t *testing.T) {
|
|
singleUse := ActiveUser{
|
|
Admin: true,
|
|
Type: passvault.AESRecord,
|
|
Expiry: nextYear,
|
|
Uses: 2,
|
|
aesKey: emptyKey,
|
|
}
|
|
|
|
UserKeys["first"] = singleUse
|
|
|
|
Refresh()
|
|
if len(UserKeys) != 1 {
|
|
t.Fatalf("Error in number of live keys")
|
|
}
|
|
|
|
EncryptKey(dummy, "first", nil)
|
|
|
|
Refresh()
|
|
if len(UserKeys) != 1 {
|
|
t.Fatalf("Error in number of live keys %v", UserKeys)
|
|
}
|
|
|
|
DecryptKey(dummy, "first", nil)
|
|
|
|
Refresh()
|
|
if len(UserKeys) != 0 {
|
|
t.Fatalf("Error in number of live keys %v", UserKeys)
|
|
}
|
|
}
|
|
|
|
func TestTimeFlush(t *testing.T) {
|
|
oneSec, _ := time.ParseDuration("1s")
|
|
one := now.Add(oneSec)
|
|
|
|
singleUse := ActiveUser{
|
|
Admin: true,
|
|
Type: passvault.AESRecord,
|
|
Expiry: one,
|
|
Uses: 10,
|
|
aesKey: emptyKey,
|
|
}
|
|
|
|
UserKeys["first"] = singleUse
|
|
|
|
Refresh()
|
|
if len(UserKeys) != 1 {
|
|
t.Fatalf("Error in number of live keys")
|
|
}
|
|
|
|
EncryptKey(dummy, "first", nil)
|
|
|
|
Refresh()
|
|
if len(UserKeys) != 1 {
|
|
t.Fatalf("Error in number of live keys")
|
|
}
|
|
|
|
time.Sleep(oneSec)
|
|
|
|
_, err := DecryptKey(dummy, "first", nil)
|
|
|
|
if err == nil {
|
|
t.Fatalf("Error in pruning expired key")
|
|
}
|
|
}
|