[feat] get openvpn args/gateways from the eip-service.json

1 files changed, 19 insertions, 3 deletions

diff --git a/bitmask_go/bonafide_test.go b/bitmask_go/bonafide_test.go
index c40c98d..152b108 100644
--- a/bitmask_go/bonafide_test.go
+++ b/bitmask_go/bonafide_test.go
@@ -11,16 +11,32 @@ var (
 )
 
 func TestGetCert(t *testing.T) {
-	cert, err := getCertPem()
+	b := newBonafide()
+	cert, err := b.getCertPem()
 	if err != nil {
-		t.Fatal("get_cert returned an error: ", err)
+		t.Fatal("getCert returned an error: ", err)
 	}
 
 	if !bytes.Contains(cert, privateKeyHeader) {
 		t.Errorf("No private key present: \n%q", cert)
 	}
 
-	if !bytes.Equal(cert, certHeader) {
+	if !bytes.Contains(cert, certHeader) {
 		t.Errorf("No cert present: \n%q", cert)
 	}
 }
+
+func TestGetGateways(t *testing.T) {
+	b := newBonafide()
+	gateways, err := b.getGateways()
+	if err != nil {
+		t.Fatal("getGateways returned an error: ", err)
+	}
+
+	for _, gw := range gateways {
+		if gw.IPAddress == "5.79.86.180" {
+			return
+		}
+	}
+	t.Errorf("5.79.86.180 not in the list")
+}