call finish on engine

This commit is contained in:
Carlosokumu 2024-09-30 18:05:20 +03:00
parent 80ce141b80
commit cf5b4ec744
Signed by untrusted user: carlos
GPG Key ID: 7BD6BC8160A5C953

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,15 +16,20 @@ 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))
if err != nil {
t.Fail()
}
w := bytes.NewBuffer(nil) w := bytes.NewBuffer(nil)
_, err = en.Flush(ctx, w) _, err = en.Flush(ctx, w)
if err != nil { if err != nil {
@ -33,16 +39,16 @@ func TestUserRegistration(t *testing.T) {
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)
} }
// }
} }
} }
} }
}
} }
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,19 +56,23 @@ 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))
if err != nil {
t.Fail()
}
w := bytes.NewBuffer(nil) w := bytes.NewBuffer(nil)
_, err = en.Flush(ctx, w) _, err = en.Flush(ctx, w)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
b := w.Bytes() b := w.Bytes()
fmt.Println("valuehere:", string(b))
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)
} }
// }
}
} }
} }