check the status of cont

This commit is contained in:
Alfred Kamanda 2024-09-30 22:12:49 +03:00
parent 9eb35ec52d
commit f3f276c549
Signed by untrusted user: Alfred-mk
GPG Key ID: 7EA3D01708908703

View File

@ -82,26 +82,29 @@ func TestTerms(t *testing.T) {
func TestAccountRegistrationRejectTerms(t *testing.T) { func TestAccountRegistrationRejectTerms(t *testing.T) {
en, _ := enginetest.TestEngine("session1234112") en, _ := enginetest.TestEngine("session1234112")
defer en.Finish() defer en.Finish()
var err error
ctx := context.Background() ctx := context.Background()
sessions := testData sessions := testData
for _, session := range sessions { for _, session := range sessions {
groups := driver.FilterGroupsByName(session.Groups, "account_creation_reject_terms") groups := driver.FilterGroupsByName(session.Groups, "account_creation_reject_terms")
for _, group := range groups { for _, group := range groups {
for _, step := range group.Steps { for _, step := range group.Steps {
cont, _ := en.Exec(ctx, []byte(step.Input)) cont, err := en.Exec(ctx, []byte(step.Input))
if cont {
w := bytes.NewBuffer(nil)
_, err = en.Flush(ctx, w)
if err != nil { if err != nil {
t.Fatal(err) t.Errorf("Test case '%s' failed at input '%s': %v", group.Name, step.Input, err)
return
} }
if !cont {
break
}
w := bytes.NewBuffer(nil)
if _, err := en.Flush(ctx, w); err != nil {
t.Errorf("Test case '%s' failed during Flush: %v", group.Name, err)
}
b := w.Bytes() b := w.Bytes()
if !bytes.Equal(b, []byte(step.ExpectedContent)) { if !bytes.Equal(b, []byte(step.ExpectedContent)) {
t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b) t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b)
} }
}
} }
} }
} }
@ -110,26 +113,29 @@ func TestAccountRegistrationRejectTerms(t *testing.T) {
func TestAccountRegistrationInvalidPin(t *testing.T) { func TestAccountRegistrationInvalidPin(t *testing.T) {
en, _ := enginetest.TestEngine("session1234112") en, _ := enginetest.TestEngine("session1234112")
defer en.Finish() defer en.Finish()
var err error
ctx := context.Background() ctx := context.Background()
sessions := testData sessions := testData
for _, session := range sessions { for _, session := range sessions {
groups := driver.FilterGroupsByName(session.Groups, "account_creation_invalid_pin") groups := driver.FilterGroupsByName(session.Groups, "account_creation_invalid_pin")
for _, group := range groups { for _, group := range groups {
for _, step := range group.Steps { for _, step := range group.Steps {
cont, _ := en.Exec(ctx, []byte(step.Input)) cont, err := en.Exec(ctx, []byte(step.Input))
if cont {
w := bytes.NewBuffer(nil)
_, err = en.Flush(ctx, w)
if err != nil { if err != nil {
t.Fatal(err) t.Errorf("Test case '%s' failed at input '%s': %v", group.Name, step.Input, err)
return
} }
if !cont {
break
}
w := bytes.NewBuffer(nil)
if _, err := en.Flush(ctx, w); err != nil {
t.Errorf("Test case '%s' failed during Flush: %v", group.Name, err)
}
b := w.Bytes() b := w.Bytes()
if !bytes.Equal(b, []byte(step.ExpectedContent)) { if !bytes.Equal(b, []byte(step.ExpectedContent)) {
t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b) t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b)
} }
}
} }
} }
} }