queue/module_test.go at master · tinh-tinh/queue · 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
package queue_test

import (
	"testing"

	"github.com/redis/go-redis/v9"
	"github.com/stretchr/testify/require"
	"github.com/tinh-tinh/queue/v2"
	"github.com/tinh-tinh/tinhtinh/v2/core"
)

func Test_Module(t *testing.T) {
	addr := "localhost:6379"
	module := core.NewModule(core.NewModuleOptions{
		Imports: []core.Modules{
			queue.Register("jobs", &queue.Options{
				Connect: &redis.Options{
					Addr:     addr,
					DB:       0,
					Password: "",
				},
				Workers:       6,
				RetryFailures: 3,
			}),
		},
	})

	queue := queue.Inject(module, "jobs")
	require.NotNil(t, queue)
}

func Test_Panic(t *testing.T) {
	require.Panics(t, func() {
		module := core.NewModule(core.NewModuleOptions{
			Imports: []core.Modules{
				queue.Register("jobs"),
			},
		})

		require.NotNil(t, module)
	})
}

func Test_DefaultOptions(t *testing.T) {
	module := core.NewModule(core.NewModuleOptions{
		Imports: []core.Modules{
			queue.ForRoot(&queue.Options{
				Connect: &redis.Options{
					Addr:     "localhost:6379",
					DB:       0,
					Password: "",
				},
				Workers:       6,
				RetryFailures: 3,
			}),
			queue.Register("video"),
			queue.Register("media", &queue.Options{
				Workers: 3,
			}),
		},
	})

	videoQueue := queue.Inject(module, "video")
	require.NotNil(t, videoQueue)

	mediaQueue := queue.Inject(module, "media")
	require.NotNil(t, mediaQueue)
}

func Test_Factory(t *testing.T) {
	module := core.NewModule(core.NewModuleOptions{
		Imports: []core.Modules{
			queue.ForRootFactory(func(ref core.RefProvider) *queue.Options {
				return &queue.Options{
					Connect: &redis.Options{
						Addr:     "localhost:6379",
						DB:       0,
						Password: "",
					},
					Workers:       6,
					RetryFailures: 3,
				}
			}),
			queue.Register("livestream"),
		},
	})

	videoQueue := queue.Inject(module, "livestream")
	require.NotNil(t, videoQueue)
}