mongo-lock/purge_test.go at master · dmichel1/mongo-lock · GitHub

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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
// Copyright 2018, Square, Inc.

package lock_test

import (
	"context"
	"sort"

	"testing"
	"time"

	lock "github.com/square/mongo-lock"
)

func TestPurge(t *testing.T) {
	// setup and teardown are defined in lock_test.go
	collection := setup(t)
	defer teardown(t, collection)

	ctx, cancel := context.WithTimeout(context.Background(), time.Second*60)
	defer cancel()

	client := lock.NewClient(collection)

	// Create some locks.
	err := client.XLock(ctx, "resource1", "aaaa", lock.LockDetails{})
	if err != nil {
		t.Error(err)
	}
	err = client.XLock(ctx, "resource2", "bbbb", lock.LockDetails{TTL: 1})
	if err != nil {
		t.Error(err)
	}
	err = client.SLock(ctx, "resource3", "cccc", lock.LockDetails{TTL: 1}, -1)
	if err != nil {
		t.Error(err)
	}

	// Sleep for a second to let TTLs expire
	time.Sleep(time.Duration(1500) * time.Millisecond)

	// Purge the locks.
	purger := lock.NewPurger(client)
	purged, err := purger.Purge(ctx)
	if err != nil {
		t.Error(err)
	}

	if len(purged) != 2 {
		t.Errorf("%d locks purged, expected %d", len(purged), 2)
	}

	var purgedSorted lock.LockStatusesByCreatedAtDesc
	purgedSorted = purged
	sort.Sort(purgedSorted)
	if purged[0].Resource != "resource3" {
		t.Errorf("purged[0].Resource = %s, expected %s", purged[0].Resource, "resource3")
	}
	if purged[1].Resource != "resource2" {
		t.Errorf("purged[1].Resource = %s, expected %s", purged[1].Resource, "resource2")
	}
}

func TestPurgeSameLockIdDiffTTLs(t *testing.T) {
	// setup and teardown are defined in lock_test.go
	collection := setup(t)
	defer teardown(t, collection)

	ctx, cancel := context.WithTimeout(context.Background(), time.Second*60)
	defer cancel()

	client := lock.NewClient(collection)

	// Create some locks with different TTLs, all owned by the same lockId.
	err := client.XLock(ctx, "resource1", "aaaa", lock.LockDetails{}) // no TTL
	if err != nil {
		t.Error(err)
	}
	err = client.XLock(ctx, "resource2", "aaaa", lock.LockDetails{TTL: 30})
	if err != nil {
		t.Error(err)
	}
	err = client.SLock(ctx, "resource3", "aaaa", lock.LockDetails{TTL: 1}, -1)
	if err != nil {
		t.Error(err)
	}

	// Sleep for a second to let some TTLs expire
	time.Sleep(time.Duration(1500) * time.Millisecond)

	// Purge the locks.
	purger := lock.NewPurger(client)
	purged, err := purger.Purge(ctx)
	if err != nil {
		t.Error(err)
	}

	if len(purged) != 3 {
		t.Errorf("%d locks purged, expected %d", len(purged), 3)
	}

	var purgedSorted lock.LockStatusesByCreatedAtDesc
	purgedSorted = purged
	sort.Sort(purgedSorted)
	if purged[0].Resource != "resource3" {
		t.Errorf("purged[0].Resource = %s, expected %s", purged[0].Resource, "resource3")
	}
	if purged[1].Resource != "resource2" {
		t.Errorf("purged[1].Resource = %s, expected %s", purged[1].Resource, "resource2")
	}
	if purged[2].Resource != "resource1" {
		t.Errorf("purged[2].Resource = %s, expected %s", purged[2].Resource, "resource1")
	}
}