Handle testing for infura using mock server

Author: waybackarchiver

pkg/infura/infura.go

@@ -22,6 +22,8 @@ const api = "https://ipfs.infura.io:5001"
 type Infura struct {
 	ProjectID     string
 	ProjectSecret string
+
+	client *http.Client
 }
 
 // PinFile alias to *Infura.PinFile, the purpose is to be backwards
@@ -107,7 +109,7 @@ func (inf *Infura) PinWithBytes(buf []byte) (string, error) {
 
 func (inf *Infura) pinFile(r *io.PipeReader, m *multipart.Writer) (string, error) {
 	endpoint := api + "/api/v0/add"
-	client := httpretry.NewClient(nil)
+	client := httpretry.NewClient(inf.client)
 	// client.Timeout = 30 * time.Second
 
 	req, err := http.NewRequest(http.MethodPost, endpoint, r)
@@ -170,7 +172,7 @@ func (inf *Infura) PinHash(hash string) (bool, error) {
 	if inf.ProjectID != "" && inf.ProjectSecret != "" {
 		req.Header.Add("Authorization", "Basic "+basicAuth(inf.ProjectID, inf.ProjectSecret))
 	}
-	client := httpretry.NewClient(nil)
+	client := httpretry.NewClient(inf.client)
 	resp, err := client.Do(req)
 	if err != nil {
 		return false, err
@@ -196,7 +198,7 @@ func (inf *Infura) PinHash(hash string) (bool, error) {
 		return false, err
 	}
 
-	if h, ok := dat["Pins"].([]interface{}); ok {
+	if h, ok := dat["Pins"].([]interface{}); ok && len(h) > 0 {
 		return h[0] == hash, nil
 	}
 

pkg/infura/infura_test.go

@@ -4,14 +4,57 @@ import (
 	"bytes"
 	"io"
 	"io/ioutil"
+	"net/http"
 	"os"
 	"strings"
+	"sync/atomic"
 	"testing"
 
 	"github.com/wabarc/helper"
 )
 
+var (
+	projectID     = "fake-project-id"
+	projectSecret = "fake-project-secret"
+	addJSON       = `{
+  "Bytes": 0,
+  "Hash": "Qmaisz6NMhDB51cCvNWa1GMS7LU1pAxdF4Ld6Ft9kZEP2a",
+  "Name": "name",
+  "Size": "string"
+}`
+	pinHashJSON = `{
+  "Pins": [
+    "Qmaisz6NMhDB51cCvNWa1GMS7LU1pAxdF4Ld6Ft9kZEP2a"
+  ],
+  "Progress": 0
+}`
+	badRequestJSON      = `{}`
+	unauthorizedJSON    = `{}`
+	tooManyRequestsJSON = `{}`
+)
+
+func handleResponse(w http.ResponseWriter, r *http.Request) {
+	authorization := r.Header.Get("Authorization")
+	if len(authorization) < 10 {
+		w.WriteHeader(http.StatusUnauthorized)
+		_, _ = w.Write([]byte(unauthorizedJSON))
+		return
+	}
+	switch r.URL.Path {
+	case "/api/v0/add":
+		_, _ = w.Write([]byte(addJSON))
+	case "/api/v0/pin/add":
+		_, _ = w.Write([]byte(pinHashJSON))
+	default:
+		_, _ = w.Write([]byte(badRequestJSON))
+	}
+}
+
 func TestPinFile(t *testing.T) {
+	httpClient, mux, server := helper.MockServer()
+	mux.HandleFunc("/", handleResponse)
+	defer server.Close()
+
 	content := []byte(helper.RandString(6, "lower"))
 	tmpfile, err := ioutil.TempFile("", "ipfs-pinner-")
 	if err != nil {
@@ -23,13 +66,17 @@ func TestPinFile(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	inf := &Infura{}
+	inf := &Infura{projectID, projectSecret, httpClient}
 	if _, err := inf.PinFile(tmpfile.Name()); err != nil {
 		t.Error(err)
 	}
 }
 
 func TestPinWithReader(t *testing.T) {
+	httpClient, mux, server := helper.MockServer()
+	mux.HandleFunc("/", handleResponse)
+	defer server.Close()
+
 	content := []byte(helper.RandString(6, "lower"))
 	tmpfile, err := ioutil.TempFile("", "ipfs-pinner-")
 	if err != nil {
@@ -49,7 +96,7 @@ func TestPinWithReader(t *testing.T) {
 		{"bytes.Buffer", bytes.NewBufferString(helper.RandString(6, "lower"))},
 	}
 
-	inf := &Infura{}
+	inf := &Infura{projectID, projectSecret, httpClient}
 	for _, test := range tests {
 		t.Run(test.name, func(t *testing.T) {
 			file := test.file.(io.Reader)
@@ -61,18 +108,62 @@ func TestPinWithReader(t *testing.T) {
 }
 
 func TestPinWithBytes(t *testing.T) {
-	inf := &Infura{}
+	httpClient, mux, server := helper.MockServer()
+	mux.HandleFunc("/", handleResponse)
+	defer server.Close()
+
+	inf := &Infura{projectID, projectSecret, httpClient}
 	buf := []byte(helper.RandString(6, "lower"))
 	if _, err := inf.PinWithBytes(buf); err != nil {
 		t.Error(err)
 	}
 }
 
 func TestPinHash(t *testing.T) {
+	httpClient, mux, server := helper.MockServer()
+	mux.HandleFunc("/", handleResponse)
+	defer server.Close()
+
 	hash := "Qmaisz6NMhDB51cCvNWa1GMS7LU1pAxdF4Ld6Ft9kZEP2a"
 
-	inf := &Infura{}
+	inf := &Infura{projectID, projectSecret, httpClient}
 	if ok, err := inf.PinHash(hash); !ok || err != nil {
 		t.Error(err)
 	}
 }
+
+func TestRateLimit(t *testing.T) {
+	if testing.Short() {
+		t.Skip("skip in short mode")
+	}
+
+	httpClient, mux, server := helper.MockServer()
+	var retries int32
+	mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
+		// Retry one times
+		if retries < 1 {
+			atomic.AddInt32(&retries, 1)
+			w.WriteHeader(http.StatusTooManyRequests)
+			_, _ = w.Write([]byte(tooManyRequestsJSON))
+		} else {
+			_, _ = w.Write([]byte(addJSON))
+		}
+	})
+	defer server.Close()
+
+	content := []byte(helper.RandString(6, "lower"))
+	tmpfile, err := ioutil.TempFile("", "ipfs-pinner-")
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer os.Remove(tmpfile.Name())
+
+	if _, err := tmpfile.Write(content); err != nil {
+		t.Fatal(err)
+	}
+
+	inf := &Infura{projectID, projectSecret, httpClient}
+	if _, err := inf.PinFile(tmpfile.Name()); err != nil {
+		t.Error(err)
+	}
+}