aboutsummaryrefslogtreecommitdiff
path: root/weed/mq/messages/message_pipeline_test.go
blob: e7ecfa8f65b509a96a42b6b91bf0291f40a3e7a0 (plain)
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
package messages

import (
	"testing"
	"time"
)

func TestAddMessage1(t *testing.T) {
	mp := NewMessagePipeline(0, 3, 10, time.Second, &EmptyMover{})
	go func() {
		for i := 0; i < 100; i++ {
			mr := mp.NextMessageBufferReference()
			println(mr.sequence, mr.fileId)
		}
	}()

	for i := 0; i < 100; i++ {
		message := &Message{
			Key:        []byte("key"),
			Content:    []byte("data"),
			Properties: nil,
			Ts:         time.Now(),
		}
		mp.AddMessage(message)
	}

	mp.ShutdownStart()
	mp.ShutdownWait()

}

func TestAddMessage2(t *testing.T) {
	mp := NewMessagePipeline(0, 3, 10, time.Second, &EmptyMover{})

	for i := 0; i < 100; i++ {
		message := &Message{
			Key:        []byte("key"),
			Content:    []byte("data"),
			Properties: nil,
			Ts:         time.Now(),
		}
		mp.AddMessage(message)
	}

	mp.ShutdownStart()
	mp.ShutdownWait()

	for i := 0; i < 100; i++ {
		mr := mp.NextMessageBufferReference()
		println(mr.sequence, mr.fileId)
	}

}