urdt/ussd
5
0
Fork 0
Code Issues 50 Pull Requests 5 Actions 1 Packages Projects Releases Wiki Activity

call finish on engine

Browse Source
This commit is contained in:
Carlosokumu 2024-09-30 18:05:20 +03:00
parent 80ce141b80
commit cf5b4ec744
Signed by: carlos
GPG Key ID: 7BD6BC8160A5C953
1 changed files with 38 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
menu_traversal_test.go
View File

@ -3,10 +3,11 @@ package main
import ( import (
"bytes" "bytes"
"context" "context"
"fmt"
"testing" "testing"
"git.grassecon.net/urdt/ussd/driver" "git.grassecon.net/urdt/ussd/driver"
"git.grassecon.net/urdt/ussd/engine" enginetest "git.grassecon.net/urdt/ussd/engine"
) )
var ( var (
@ -15,34 +16,39 @@ var (
func TestUserRegistration(t *testing.T) { func TestUserRegistration(t *testing.T) {
en, _ := enginetest.TestEngine("session1234112") en, _ := enginetest.TestEngine("session1234112")
var err error 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_successful") groups := driver.FilterGroupsByName(session.Groups, "account_creation_successful")
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)) // for {
if cont { _, err := en.Exec(ctx, []byte(step.Input))
w := bytes.NewBuffer(nil)
_, err = en.Flush(ctx, w)
if err != nil {
t.Fatal(err)
}
b := w.Bytes()
if !bytes.Equal(b, []byte(step.ExpectedContent)) {
t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b)
}
if err != nil {
t.Fail()
} }
w := bytes.NewBuffer(nil)
_, err = en.Flush(ctx, w)
if err != nil {
t.Fatal(err)
}
b := w.Bytes()
if !bytes.Equal(b, []byte(step.ExpectedContent)) {
t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b)
}
// }
} }
} }
} }
} }
func TestAcceptTerms(t *testing.T) { func TestTerms(t *testing.T) {
en, _ := enginetest.TestEngine("session12341123") en, _ := enginetest.TestEngine("session1234112")
var err error defer en.Finish()
ctx := context.Background() ctx := context.Background()
sessions := testData sessions := testData
@ -50,20 +56,24 @@ func TestAcceptTerms(t *testing.T) {
groups := driver.FilterGroupsByName(session.Groups, "account_creation_accept_terms") groups := driver.FilterGroupsByName(session.Groups, "account_creation_accept_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)) // for {
if cont { _, err := en.Exec(ctx, []byte(step.Input))
w := bytes.NewBuffer(nil) if err != nil {
_, err = en.Flush(ctx, w) t.Fail()
if err != nil {
t.Fatal(err)
}
b := w.Bytes()
if !bytes.Equal(b, []byte(step.ExpectedContent)) {
t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b)
}
} }
w := bytes.NewBuffer(nil)
_, err = en.Flush(ctx, w)
if err != nil {
t.Fatal(err)
}
b := w.Bytes()
fmt.Println("valuehere:", string(b))
if !bytes.Equal(b, []byte(step.ExpectedContent)) {
t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b)
}
// }
} }
} }
} }