From 74fa2c3d65e2236aab4aa56ead1ca59eff4f8019 Mon Sep 17 00:00:00 2001
From: Carlosokumu <carlosokumu254@gmail.com>
Date: Wed, 2 Oct 2024 11:23:34 +0300
Subject: [PATCH] update tests

---
 menu_traversal_test.go | 67 ++++++++++++++++++++++++++++++++++++++++++
 test_data.json         | 54 ++++++++++++++++++++++++++++++++++
 2 files changed, 121 insertions(+)

diff --git a/menu_traversal_test.go b/menu_traversal_test.go
index 02782c0..2dc7bd8 100644
--- a/menu_traversal_test.go
+++ b/menu_traversal_test.go
@@ -536,3 +536,70 @@ func TestMyAccount_View_Profile(t *testing.T) {
 		}
 	}
 }
+
+
+func TestMyAccount_Check_Balance(t *testing.T){
+	en, fn := enginetest.TestEngine("session1234112")
+	defer fn()
+	ctx := context.Background()
+	sessions := testData
+	for _, session := range sessions {
+		groups := driver.FilterGroupsByName(session.Groups, "menu_my_account_check_balance")
+		for _, group := range groups {
+			for index, step := range group.Steps {
+				t.Logf("step %v with input %v", index, step.Input)
+				cont, err := en.Exec(ctx, []byte(step.Input))
+				if err != nil {
+					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()
+				if !bytes.Equal(b, []byte(step.ExpectedContent)) {
+					t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b)
+				}
+
+			}
+		}
+	}
+}
+
+
+func TestMyAccount_Check_Balance_IncorrectPin(t *testing.T){
+	en, fn := enginetest.TestEngine("session1234112")
+	defer fn()
+	ctx := context.Background()
+	sessions := testData
+	for _, session := range sessions {
+		groups := driver.FilterGroupsByName(session.Groups, "menu_my_account_check_balance_incorrect_pin")
+		for _, group := range groups {
+			for index, step := range group.Steps {
+				t.Logf("step %v with input %v", index, step.Input)
+				cont, err := en.Exec(ctx, []byte(step.Input))
+				if err != nil {
+					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()
+				if !bytes.Equal(b, []byte(step.ExpectedContent)) {
+					t.Fatalf("expected:\n\t%s\ngot:\n\t%s\n", step.ExpectedContent, b)
+				}
+
+			}
+		}
+	}
+}
+
diff --git a/test_data.json b/test_data.json
index 4388b5a..651fa80 100644
--- a/test_data.json
+++ b/test_data.json
@@ -481,6 +481,60 @@
                         "expectedContent":""
                     }                 
                 ]
+            },
+            {
+                "name": "menu_my_account_check_balance",
+                "steps": [
+                    {
+                        "input":"",
+                       "expectedContent": "Balance: 0.003 CELO\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
+
+                    },
+                    {
+                        "input": "3",
+                       "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
+                    },
+                    {
+                        "input": "3",
+                       "expectedContent": "Balances:\n1:My balance\n2:Community balance\n0:Back"
+                    },
+                    {
+                        "input": "1",
+                       "expectedContent": "Please enter your PIN:"
+                    },
+                    {
+                        "input": "1234",
+                       "expectedContent": "Please enter your PIN:"
+                    }
+                                 
+                ]
+            },
+            {
+                "name": "menu_my_account_check_balance_incorrect_pin",
+                "steps": [
+                    {
+                        "input":"",
+                       "expectedContent": "Balance: 0.003 CELO\n\n1:Send\n2:My Vouchers\n3:My Account\n4:Help\n9:Quit"
+
+                    },
+                    {
+                        "input": "3",
+                       "expectedContent": "My Account\n1:Profile\n2:Change language\n3:Check balances\n4:Check statement\n5:PIN options\n6:My Address\n0:Back"
+                    },
+                    {
+                        "input": "3",
+                       "expectedContent": "Balances:\n1:My balance\n2:Community balance\n0:Back"
+                    },
+                    {
+                        "input": "1",
+                       "expectedContent": "Please enter your PIN:"
+                    },
+                    {
+                        "input": "1235",
+                       "expectedContent": "Incorrect pin\n1:retry\n9:Quit"
+                    }
+                                 
+                ]
             }
 
         ]