1 package serial_fixture_test
2
3 import (
4 "flag"
5 "fmt"
6 "io"
7 "net/http"
8 "sync"
9 "testing"
10 "time"
11
12 . "github.com/onsi/ginkgo/v2"
13 . "github.com/onsi/gomega"
14 "github.com/onsi/gomega/ghttp"
15 )
16
17 var noSerial *bool
18
19 func init() {
20 noSerial = flag.CommandLine.Bool("no-serial", false, "set to turn off serial decoration")
21 }
22
23 var SerialDecoration = []interface{}{Serial}
24
25 func TestSerialFixture(t *testing.T) {
26 RegisterFailHandler(Fail)
27 if *noSerial {
28 SerialDecoration = []interface{}{}
29 }
30
31 RunSpecs(t, "SerialFixture Suite")
32 }
33
34 var addr string
35
36 var _ = SynchronizedBeforeSuite(func() []byte {
37 server := ghttp.NewServer()
38 lock := &sync.Mutex{}
39 count := 0
40 server.RouteToHandler("POST", "/counter", func(w http.ResponseWriter, r *http.Request) {
41 lock.Lock()
42 count += 1
43 lock.Unlock()
44 w.WriteHeader(http.StatusOK)
45 })
46 server.RouteToHandler("GET", "/counter", func(w http.ResponseWriter, r *http.Request) {
47 lock.Lock()
48 data := []byte(fmt.Sprintf("%d", count))
49 lock.Unlock()
50 w.Write(data)
51 })
52 return []byte(server.HTTPTestServer.URL)
53 }, func(data []byte) {
54 addr = string(data) + "/counter"
55 })
56
57 var postToCounter = func(g Gomega) {
58 req, err := http.Post(addr, "", nil)
59 g.Ω(err).ShouldNot(HaveOccurred())
60 g.Ω(req.StatusCode).Should(Equal(http.StatusOK))
61 g.Ω(req.Body.Close()).Should(Succeed())
62 }
63
64 var getCounter = func(g Gomega) string {
65 req, err := http.Get(addr)
66 g.Ω(err).ShouldNot(HaveOccurred())
67 content, err := io.ReadAll(req.Body)
68 g.Ω(err).ShouldNot(HaveOccurred())
69 g.Ω(req.Body.Close()).Should(Succeed())
70 return string(content)
71 }
72
73 var _ = SynchronizedAfterSuite(func() {
74 Consistently(postToCounter, 200*time.Millisecond, 10*time.Millisecond).Should(Succeed())
75 }, func() {
76 initialValue := getCounter(Default)
77 Consistently(func(g Gomega) {
78 currentValue := getCounter(g)
79 g.Ω(currentValue).Should(Equal(initialValue))
80 }, 100*time.Millisecond, 10*time.Millisecond).Should(Succeed())
81 })
82
83 var _ = Describe("tests", func() {
84 for i := 0; i < 10; i += 1 {
85 It("runs in parallel", func() {
86 Consistently(postToCounter, 200*time.Millisecond, 10*time.Millisecond).Should(Succeed())
87 })
88 }
89
90 for i := 0; i < 5; i += 1 {
91 It("runs in series", SerialDecoration, func() {
92 Ω(GinkgoParallelProcess()).Should(Equal(1))
93 initialValue := getCounter(Default)
94 Consistently(func(g Gomega) {
95 currentValue := getCounter(g)
96 g.Ω(currentValue).Should(Equal(initialValue))
97 }, 100*time.Millisecond, 10*time.Millisecond).Should(Succeed())
98 })
99 }
100 })
101
View as plain text