This repository has been archived by the owner on Aug 31, 2022. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_helper.go
97 lines (77 loc) · 2.23 KB
/
test_helper.go
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
93
94
95
96
97
package main
import (
"bytes"
"github.com/aws/aws-lambda-go/events"
"html/template"
"os"
)
// テスト用の期待値を作成する
func createExpectedSignUpMessage(m SignUpMessage) (*bytes.Buffer, error) {
t := template.New("signup-template.html")
currentDir, _ := os.Getwd()
templatePath := currentDir + "/signup-template.html"
templates := template.Must(t.ParseFiles(templatePath))
var bodyBuffer bytes.Buffer
err := templates.Execute(&bodyBuffer, m)
if err != nil {
return nil, err
}
return &bodyBuffer, nil
}
// ForgotPasswordMessageカスタムメッセージのテスト用期待値を作成する
func createExpectedForgotPasswordMessageMessage(m ForgotPasswordMessage) (*bytes.Buffer, error) {
t := template.New("forgot-password-template.html")
currentDir, _ := os.Getwd()
templatePath := currentDir + "/forgot-password-template.html"
templates := template.Must(t.ParseFiles(templatePath))
var bodyBuffer bytes.Buffer
err := templates.Execute(&bodyBuffer, m)
if err != nil {
return nil, err
}
return &bodyBuffer, nil
}
type createUserPoolsCustomMessageEventParams struct {
TriggerSource string
UserPoolID string
UserName string
Sub string
CodeParameter string
}
func createUserPoolsCustomMessageEvent(p *createUserPoolsCustomMessageEventParams) *events.CognitoEventUserPoolsCustomMessage {
cc := &events.CognitoEventUserPoolsCallerContext{
AWSSDKVersion: "",
ClientID: "",
}
ch := &events.CognitoEventUserPoolsHeader{
Version: "",
TriggerSource: p.TriggerSource,
Region: "",
UserPoolID: p.UserPoolID,
CallerContext: *cc,
UserName: p.UserName,
}
ua := map[string]interface{}{
"sub": p.Sub,
}
cm := map[string]string{
"key": "",
}
req := &events.CognitoEventUserPoolsCustomMessageRequest{
UserAttributes: ua,
CodeParameter: p.CodeParameter,
UsernameParameter: p.UserName,
ClientMetadata: cm,
}
res := &events.CognitoEventUserPoolsCustomMessageResponse{
SMSMessage: "SMSMessage",
EmailMessage: "EmailMessage",
EmailSubject: "EmailSubject",
}
ev := &events.CognitoEventUserPoolsCustomMessage{
CognitoEventUserPoolsHeader: *ch,
Request: *req,
Response: *res,
}
return ev
}