1 files changed, 34 insertions, 0 deletions
diff --git a/client_test.go b/client_test.go
index 1daa196..95f61a6 100644
--- a/client_test.go
+++ b/client_test.go
@@ -10,6 +10,40 @@ import (
 	"github.com/shuLhan/share/lib/test"
 )
 
+func TestClient_DisbursementCheckBalance(t *testing.T) {
+	var (
+		tdata   *test.Data
+		balance *Balance
+		err     error
+		exp     []byte
+		got     []byte
+	)
+
+	tdata, err = test.LoadData(`testdata/disbursement_checkbalance_test.txt`)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	balance, err = testClient.DisbursementCheckBalance()
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	// Set the value to zero, since their value may be different on each
+	// run.
+	balance.Current.Scan(0)
+	balance.Effective.Scan(0)
+
+	got, err = json.MarshalIndent(balance, ``, `  `)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	exp = tdata.Output[`response.json`]
+
+	test.Assert(t, `DisbursementCheckBalance`, string(exp), string(got))
+}
+
 func TestClient_DisbursementListBank(t *testing.T) {
 	var (
 		tdata *test.Data