From b69e836d41c89f155e86f883e3c4e00ec5d9a674 Mon Sep 17 00:00:00 2001
From: atbore-phx <atbore@gmail.com>
Date: Tue, 5 Mar 2024 19:37:59 +0100
Subject: [PATCH] feat(tests): improved storage tests

---
 pkg/storage/storage_test.go | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/pkg/storage/storage_test.go b/pkg/storage/storage_test.go
index 8e3200f..895e208 100644
--- a/pkg/storage/storage_test.go
+++ b/pkg/storage/storage_test.go
@@ -35,6 +35,27 @@ var resp string = `{
 	}
 }`
 
+var respBD string = `{
+	"Body" : {
+	   "Data" : {
+		  "0" : {
+			 "Controller" : {
+				"DesignedCapacity" : 11059.0,
+				"Enable" : 0,
+				"StateOfCharge_Relative" : 82.0
+			 }
+			},
+			 "1" : {
+				"Controller" : {
+				   "DesignedCapacity" : 13809.0,
+				   "Enable" : 0,
+				   "StateOfCharge_Relative" : 70.0
+			 }
+		  }
+	   }
+	}
+}`
+
 var respJsonErr string = `{
 	"Body" : {
 	   "Data" : {
@@ -207,3 +228,17 @@ func TestHandlerError2(t *testing.T) {
 
 	teardown()
 }
+
+func TestHandlerError3(t *testing.T) {
+	setup(respBD)
+
+	st := storage.New()
+	ip := strings.TrimPrefix(mockServer.URL, "http://")
+
+	charge, charge_max, err := st.Handler(ip)
+	assert.Equal(t, float64(0), charge)
+	assert.Equal(t, float64(0), charge_max)
+	assert.Error(t, err)
+
+	teardown()
+}