call finish on engine
This commit is contained in:
parent
80ce141b80
commit
cf5b4ec744
@ -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)
|
||||||
}
|
}
|
||||||
|
// }
|
||||||
}
|
|
||||||
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user