llcm/benchmark_test.go at main · nekrassov01/llcm · 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
package llcm

import (
	"context"
	"io"
	"testing"
	"time"

	"github.com/aws/aws-sdk-go-v2/aws"
	"github.com/aws/aws-sdk-go-v2/service/cloudwatchlogs"
	"github.com/aws/aws-sdk-go-v2/service/cloudwatchlogs/types"
	"golang.org/x/sync/semaphore"
)

var (
	benchN = 10
	benchR = []string{"us-east-1"}
)

func prepare(n int, regions []string) *Manager {
	var (
		logGroups       = make([]types.LogGroup, n)
		logGroupName    = aws.String("log-group")
		creationTime    = aws.Int64(nowFunc().UnixNano() / int64(time.Millisecond))
		retentionInDays = aws.Int32(365)
		storedBytes     = aws.Int64(1024)
		arn             = aws.String("arn:aws:logs:region:account-id:log-group:log-group")
		logGroupClass   = types.LogGroupClassStandard
	)
	return &Manager{
		client: newMockClient(&mockClient{
			DescribeLogGroupsFunc: func(_ context.Context, _ *cloudwatchlogs.DescribeLogGroupsInput, _ ...func(*cloudwatchlogs.Options)) (*cloudwatchlogs.DescribeLogGroupsOutput, error) {
				for i := range n {
					logGroups[i] = types.LogGroup{
						LogGroupName:    logGroupName,
						CreationTime:    creationTime,
						RetentionInDays: retentionInDays,
						StoredBytes:     storedBytes,
						Arn:             arn,
						LogGroupClass:   logGroupClass,
					}
				}
				out := &cloudwatchlogs.DescribeLogGroupsOutput{
					LogGroups: logGroups,
				}
				return out, nil
			},
			DeleteLogGroupFunc: func(_ context.Context, _ *cloudwatchlogs.DeleteLogGroupInput, _ ...func(*cloudwatchlogs.Options)) (*cloudwatchlogs.DeleteLogGroupOutput, error) {
				return &cloudwatchlogs.DeleteLogGroupOutput{}, nil
			},
			PutRetentionPolicyFunc: func(_ context.Context, _ *cloudwatchlogs.PutRetentionPolicyInput, _ ...func(*cloudwatchlogs.Options)) (*cloudwatchlogs.PutRetentionPolicyOutput, error) {
				return &cloudwatchlogs.PutRetentionPolicyOutput{}, nil
			},
			DeleteRetentionPolicyFunc: func(_ context.Context, _ *cloudwatchlogs.DeleteRetentionPolicyInput, _ ...func(*cloudwatchlogs.Options)) (*cloudwatchlogs.DeleteRetentionPolicyOutput, error) {
				return &cloudwatchlogs.DeleteRetentionPolicyOutput{}, nil
			},
		}),
		regions:      regions,
		desiredState: 365,
		sem:          semaphore.NewWeighted(NumWorker),
	}
}

func BenchmarkList(b *testing.B) {
	man := prepare(benchN, benchR)
	for b.Loop() {
		_, err := man.List(context.Background())
		if err != nil {
			b.Fatal(err)
		}
	}
}

func BenchmarkPreview(b *testing.B) {
	man := prepare(benchN, benchR)
	for b.Loop() {
		_, err := man.Preview(context.Background())
		if err != nil {
			b.Fatal(err)
		}
	}
}

func BenchmarkApply(b *testing.B) {
	man := prepare(benchN, benchR)
	for b.Loop() {
		_, err := man.Apply(context.Background(), io.Discard)
		if err != nil {
			b.Fatal(err)
		}
	}
}