store/store_test.go at master · andersnormal/store · 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
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
package store_test

import (
	"math/rand"
	"os"
	"sync"
	"time"

	"github.com/andersnormal/store"
	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
)

const (
	DefaultKeyString = "42"
)

type TestRich struct {
	String string
	Int    int
}

var _ = Describe("Store", func() {
	var (
		s   *store.Store
		err error
	)

	BeforeEach(func() {
		os.Remove(DefaultStore)

		s, err = store.Open(DefaultStore, nil)
		Expect(err).NotTo(HaveOccurred())
	})

	AfterEach(func() {
		err = s.Close()
		Expect(err).NotTo(HaveOccurred())
	})

	Describe("Put(), Delete() and Get() values from the store", func() {
		It("Should allow to Put() values", func() {
			By("Entering 42")

			err = s.Put(store.Byte(DefaultKeyString), "42")
			Expect(err).NotTo(HaveOccurred())
		})

		It("Should allow to Get() values", func() {
			By("Putting and getting 42")

			err = s.Put(store.Byte(DefaultKeyString), "42")
			Expect(err).NotTo(HaveOccurred())

			var val string
			err = s.Get(store.Byte(DefaultKeyString), &val)
			Expect(err).NotTo(HaveOccurred())
			Expect(val).To(Equal("42"))
		})

		It("Should allow to Delete() keys", func() {
			By("Deleting 42")

			err = s.Put(store.Byte(DefaultKeyString), "42")
			Expect(err).NotTo(HaveOccurred())

			err = s.Delete(store.Byte(DefaultKeyString))
			Expect(err).NotTo(HaveOccurred())
		})
	})

	Describe("Put() and Get() rich types", func() {
		It("Should allow to Put() rich types", func() {
			r := &TestRich{"42", 42}

			By("Putting rich type")

			err = s.Put(store.Byte(DefaultKeyString), r)
			Expect(err).NotTo(HaveOccurred())
		})

		It("Should allow to Put() and Get() rich types", func() {
			r := &TestRich{"42", 42}

			By("Putting and getting rich type")

			err = s.Put(store.Byte(DefaultKeyString), r)
			Expect(err).NotTo(HaveOccurred())

			var val TestRich
			err = s.Get(store.Byte(DefaultKeyString), &val)
			Expect(err).NotTo(HaveOccurred())
			Expect(val.Int).To(Equal(42))
			Expect(val.String).To(Equal("42"))
		})
	})

	Describe("Put() and Get() on nil", func() {
		It("Should not allow to Put() nil value", func() {
			err = s.Put(store.Byte(DefaultKeyString), nil)
			Expect(err).To(HaveOccurred())
			Expect(err).To(Equal(store.ErrBadValue))
		})

		It("Should allow to Get() nil value", func() {
			err = s.Put(store.Byte(DefaultKeyString), "42")
			Expect(err).NotTo(HaveOccurred())

			err = s.Get(store.Byte(DefaultKeyString), nil)
			Expect(err).NotTo(HaveOccurred())
		})
	})

	Describe("Put() and Get() on goroutine", func() {
		It("Should be goroutine-safe", func(done Done) {
			var wg sync.WaitGroup

			rand.Seed(time.Now().UnixNano())

			for i := 0; i < 1000; i++ {
				wg.Add(1)
				go func() {
					defer GinkgoRecover()
					defer wg.Done()

					switch rand.Intn(3) {
					case 0:
						err = s.Put(store.Byte(DefaultKeyString), "42")
						Expect(err).NotTo(HaveOccurred())
					case 1:
						var val string
						err = s.Get(store.Byte(DefaultKeyString), &val)
						if err != nil && err != store.ErrKeyNotExist {
							Expect(err).To(HaveOccurred())
							return
						}
						Expect(val).To(Equal("42"))
					case 2:
						err = s.Delete(store.Byte(DefaultKeyString))
						Expect(err).NotTo(HaveOccurred())
					}
				}()
			}
			wg.Wait()
			close(done)
		})
	})

	Measure("it should Put() very efficient", func(b Benchmarker) {
		runtime := b.Time("runtime", func() {
			err = s.Put(store.Byte(DefaultKeyString), "42")
			Expect(err).NotTo(HaveOccurred())
		})

		Ω(runtime.Seconds()).Should(BeNumerically("<", 0.2), "Put() shouldn't take too long.")
	}, 10)
})