moira

1
package test_helpers
2

3
import (
4
	"go.avito.ru/DO/moira"
5
	"go.avito.ru/DO/moira/logging"
6
	"go.avito.ru/DO/moira/metrics"
7
)
8

9
var (
10
	scheduleDaysAll = []moira.ScheduleDataDay{
11
		{true, "Mon"},
12
		{true, "Tue"},
13
		{true, "Wed"},
14
		{true, "Thu"},
15
		{true, "Fri"},
16
		{true, "Sat"},
17
		{true, "Sun"},
18
	}
19
	scheduleDaysNone = []moira.ScheduleDataDay{
20
		{false, "Mon"},
21
		{false, "Tue"},
22
		{false, "Wed"},
23
		{false, "Thu"},
24
		{false, "Fri"},
25
		{false, "Sat"},
26
		{false, "Sun"},
27
	}
28
)
29

30
func GetSchedule24x7() *moira.ScheduleData {
31
	return &moira.ScheduleData{
32
		Days:           scheduleDaysAll,
33
		TimezoneOffset: 0,
34
		StartOffset:    0,
35
		EndOffset:      1439,
36
	}
37
}
38

39
func GetScheduleNever() *moira.ScheduleData {
40
	return &moira.ScheduleData{
41
		Days:           scheduleDaysNone,
42
		TimezoneOffset: 0,
43
		StartOffset:    0,
44
		EndOffset:      1439,
45
	}
46
}
47

48
// GetTestLogger returns context unaware logger for tests purposes only
49
func GetTestLogger() *logging.Logger {
50
	InitTestLogging()
51
	return logging.GetLogger("")
52
}
53

54
// InitTestLogging initializes logging subsystem for tests purposes only
55
// syslog is disabled, fallback is stdout
56
func InitTestLogging() {
57
	rateLimits := moira.RateLimit{
58
		AcceptRate: 1,
59
		ThreadsQty: 2,
60
	}
61
	_ = metrics.Init(metrics.Config{Enabled: false, IsTest: true}, rateLimits)
62
	_ = logging.Init(
63
		logging.ComponentTests,
64
		logging.Config{
65
			Enabled:  false,
66
			Host:     "",
67
			Port:     0,
68
			Level:    "debug",
69
			Fallback: "stdout",
70
			Debug:    true,
71
		},
72
		rateLimits,
73
	)
74
}
75