Function TestAwsMskIamMechanism

sasl/aws_msk_iam/msk_iam_test.go:24–105

(t *testing.T)

var signTime = time.Date(2021, 10, 14, 13, 5, 0, 0, time.UTC)

func TestAwsMskIamMechanism(t *testing.T) {
	tests := []struct {
		description string
		ctx         func() context.Context
		shouldFail  bool
	}{
		{
			description: "with metadata",
			ctx: func() context.Context {
				return sasl.WithMetadata(context.Background(), &sasl.Metadata{
					Host: "localhost",
					Port: 9092,
				})
			},
		},
		{
			description: "without metadata",
			ctx: func() context.Context {
				return context.Background()
			},
			shouldFail: true,
		},
	}

	for _, tt := range tests {
		t.Run(tt.description, func(t *testing.T) {
			ctx := tt.ctx()

			creds := credentials.NewStaticCredentials(accessKeyId, secretAccessKey, "")
			mskMechanism := &Mechanism{
				Signer:   sigv4.NewSigner(creds),
				Region:   "us-east-1",
				SignTime: signTime,
			}

			sess, auth, err := mskMechanism.Start(ctx)
			if tt.shouldFail { // if error is expected
				if err == nil { // but we don't find one
					t.Fatal("error expected")
				} else { // but we do find one
					return // return early since the remaining assertions are irrelevant
				}
			} else { // if error is not expected (typical)
				if err != nil { // but we do find one
					t.Fatal(err)
				}
			}

			if sess != mskMechanism {
				t.Error(
					"Unexpected session",
					"expected", mskMechanism,
					"got", sess,
				)
			}

			expectedMap := map[string]string{
				"version":             "2020_10_22",