ussd/menu_traversal_test.go

140 lines
3.6 KiB
Go
Raw Normal View History

2024-09-27 14:02:29 +02:00
package main
import (
2024-09-30 12:21:17 +02:00
"bytes"
2024-09-27 14:02:29 +02:00
"context"
"testing"
2024-09-30 12:21:17 +02:00
"git.grassecon.net/urdt/ussd/driver"
2024-09-30 17:05:20 +02:00
enginetest "git.grassecon.net/urdt/ussd/engine"
2024-09-27 14:02:29 +02:00
)
2024-09-30 12:21:17 +02:00
var (
testData = driver.ReadData()
)
2024-09-27 14:02:29 +02:00
2024-09-30 12:21:17 +02:00
func TestUserRegistration(t *testing.T) {
en, fn := enginetest.TestEngine("session1234112")
defer fn()
2024-09-30 12:21:17 +02:00
ctx := context.Background()
sessions := testData
for _, session := range sessions {
groups := driver.FilterGroupsByName(session.Groups, "account_creation_successful")
for _, group := range groups {
for _, step := range group.Steps {
2024-09-30 21:27:37 +02:00
cont, err := en.Exec(ctx, []byte(step.Input))
2024-09-27 14:02:29 +02:00
2024-09-30 17:05:20 +02:00
if err != nil {
t.Fatalf("Test case '%s' failed at input '%s': %v", group.Name, step.Input, err)
2024-09-30 21:27:37 +02:00
}
if !cont {
break
2024-09-30 12:21:17 +02:00
}
2024-09-30 17:05:20 +02:00
w := bytes.NewBuffer(nil)
_, err = en.Flush(ctx, w)
if err != nil {
t.Fatalf("Test case '%s' failed during Flush: %v", group.Name, err)
2024-09-30 17:05:20 +02:00
}
b := w.Bytes()
if !bytes.Equal(b, []byte(step.ExpectedContent)) {
t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b)
}
2024-09-30 12:21:17 +02:00
}
}
}
2024-09-27 14:02:29 +02:00
}
2024-09-30 17:05:20 +02:00
func TestTerms(t *testing.T) {
en, fn := enginetest.TestEngine("session1234112_a")
defer fn()
2024-09-30 12:21:17 +02:00
ctx := context.Background()
sessions := testData
2024-09-27 14:02:29 +02:00
2024-09-30 12:21:17 +02:00
for _, session := range sessions {
groups := driver.FilterGroupsByName(session.Groups, "account_creation_accept_terms")
for _, group := range groups {
for _, step := range group.Steps {
2024-09-30 17:05:20 +02:00
_, err := en.Exec(ctx, []byte(step.Input))
if err != nil {
t.Fatalf("Test case '%s' failed during Exec: %v", group.Name, err)
2024-09-30 17:05:20 +02:00
}
2024-09-27 14:02:29 +02:00
2024-09-30 17:05:20 +02:00
w := bytes.NewBuffer(nil)
_, err = en.Flush(ctx, w)
if err != nil {
t.Fatalf("Test case '%s' failed during Flush: %v", group.Name, err)
2024-09-30 17:05:20 +02:00
}
b := w.Bytes()
if !bytes.Equal(b, []byte(step.ExpectedContent)) {
t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b)
2024-09-30 12:21:17 +02:00
}
2024-09-27 14:02:29 +02:00
2024-09-30 12:21:17 +02:00
}
}
}
2024-09-27 14:02:29 +02:00
}
2024-09-30 15:05:14 +02:00
func TestAccountRegistrationRejectTerms(t *testing.T) {
en, fn := enginetest.TestEngine("session1234112_b")
defer fn()
2024-09-30 15:05:14 +02:00
ctx := context.Background()
sessions := testData
for _, session := range sessions {
groups := driver.FilterGroupsByName(session.Groups, "account_creation_reject_terms")
for _, group := range groups {
for _, step := range group.Steps {
2024-09-30 21:12:49 +02:00
cont, err := en.Exec(ctx, []byte(step.Input))
if err != nil {
t.Fatalf("Test case '%s' failed at input '%s': %v", group.Name, step.Input, err)
2024-09-30 21:12:49 +02:00
return
}
if !cont {
break
}
w := bytes.NewBuffer(nil)
if _, err := en.Flush(ctx, w); err != nil {
t.Fatalf("Test case '%s' failed during Flush: %v", group.Name, err)
2024-09-30 21:12:49 +02:00
}
2024-09-30 15:05:14 +02:00
2024-09-30 21:12:49 +02:00
b := w.Bytes()
if !bytes.Equal(b, []byte(step.ExpectedContent)) {
t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b)
2024-09-30 15:05:14 +02:00
}
}
}
}
}
func TestAccountRegistrationInvalidPin(t *testing.T) {
en, fn := enginetest.TestEngine("session1234112")
defer fn()
2024-09-30 15:05:14 +02:00
ctx := context.Background()
sessions := testData
for _, session := range sessions {
groups := driver.FilterGroupsByName(session.Groups, "account_creation_invalid_pin")
for _, group := range groups {
for _, step := range group.Steps {
2024-09-30 21:12:49 +02:00
cont, err := en.Exec(ctx, []byte(step.Input))
if err != nil {
t.Fatalf("Test case '%s' failed at input '%s': %v", group.Name, step.Input, err)
2024-09-30 21:12:49 +02:00
return
}
if !cont {
break
}
w := bytes.NewBuffer(nil)
if _, err := en.Flush(ctx, w); err != nil {
t.Fatalf("Test case '%s' failed during Flush: %v", group.Name, err)
2024-09-30 21:12:49 +02:00
}
2024-09-30 15:05:14 +02:00
2024-09-30 21:12:49 +02:00
b := w.Bytes()
if !bytes.Equal(b, []byte(step.ExpectedContent)) {
t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b)
2024-09-30 15:05:14 +02:00
}
}
}
}
}