Merge pull request #33 from vulncheck-oss/pdns-support

Add Support for PDNS endpoint

Author: tcampbPPU

pdns.go

@@ -0,0 +1,27 @@
+package sdk
+
+import (
+	"io"
+	"net/http"
+)
+
+// https://docs.vulncheck.com/api/pdns
+func (c *Client) GetPdns() (string, error) {
+	client := &http.Client{}
+	req, err := http.NewRequest("GET", c.GetUrl()+"/v3/pdns/vulncheck-c2", nil)
+	if err != nil {
+		panic(err)
+	}
+
+	c.SetAuthHeader(req)
+
+	res, err := client.Do(req)
+	if err != nil {
+		panic(err)
+	}
+
+	defer res.Body.Close()
+	body, _ := io.ReadAll(res.Body)
+
+	return string(body), nil
+}

pdns_test.go

@@ -0,0 +1,16 @@
+package sdk
+
+// LIVE TESTS - these tests require a valid token to run
+// func TestGetPdns(t *testing.T) {
+// 	t.Run("pdns is retrieved", func(t *testing.T) {
+// 		client := Connect("URL", "")
+// 		data, err := client.GetPdns()
+// 		if err != nil {
+// 			t.Errorf("Error: %v", err)
+// 		}
+
+// 		if data == "" {
+// 			t.Errorf("Expected a string, got %v", data)
+// 		}
+// 	})
+// }

tags_test.go

@@ -3,7 +3,6 @@ package sdk
 // LIVE TESTS - these tests require a valid token to run
 // func TestGetTag(t *testing.T) {
 // 	t.Run("tag is retrieved", func(t *testing.T) {
-// 		ray.Ray("Running test")
 // 		client := Connect("URL", "")
 // 		data, err := client.GetTag("vulncheck-c2")
 // 		if err != nil {