internal/storage/replay_test.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

package storage

import (
	"testing"
	"time"
)

func TestReplayCacheCheckAndMarkLivesOnlyInMemory(t *testing.T) {
	cache := NewReplayCache(40 * time.Millisecond)

	replayed, err := cache.CheckAndMark("token-hash")
	if err != nil {
		t.Fatalf("CheckAndMark returned error: %v", err)
	}
	if replayed {
		t.Fatal("first token use should not be marked as replayed")
	}

	replayed, err = cache.CheckAndMark("token-hash")
	if err != nil {
		t.Fatalf("CheckAndMark returned error on second use: %v", err)
	}
	if !replayed {
		t.Fatal("second token use should be marked as replayed")
	}

	time.Sleep(60 * time.Millisecond)

	replayed, err = cache.CheckAndMark("token-hash")
	if err != nil {
		t.Fatalf("CheckAndMark returned error after ttl expiry: %v", err)
	}
	if replayed {
		t.Fatal("expired token entry should be forgotten from memory")
	}
}