From 6d0084274323bc9418bf6ecacb75ce5e673878d5 Mon Sep 17 00:00:00 2001
From: Timothy Lock <lock.k.timothy@gmail.com>
Date: Mon, 22 Jan 2018 15:40:28 -0500
Subject: [PATCH] Bring tests back up to 100%

Make a new method called parseResponse which essentially allows us to feed it our own response  so that we can feed it bad body so that it cannot unmarshal or even read it.
---
 appstore/validator.go      |  5 ++-
 appstore/validator_test.go | 82 +++++++++++++++-----------------------
 2 files changed, 37 insertions(+), 50 deletions(-)

diff --git a/appstore/validator.go b/appstore/validator.go
index 42eeba0..731109c 100644
--- a/appstore/validator.go
+++ b/appstore/validator.go
@@ -115,7 +115,10 @@ func (c *Client) Verify(req IAPRequest, result interface{}) error {
 		return err
 	}
 	defer resp.Body.Close()
+	return c.parseResponse(resp, result, client, req)
+}
 
+func (c *Client) parseResponse(resp *http.Response, result interface{}, client http.Client, req IAPRequest) error {
 	// Read the body now so that we can unmarshal it twice
 	buf, err := ioutil.ReadAll(resp.Body)
 	if err != nil {
@@ -134,7 +137,7 @@ func (c *Client) Verify(req IAPRequest, result interface{}) error {
 		return err
 	}
 	if r.Status == 21007 {
-		b = new(bytes.Buffer)
+		b := new(bytes.Buffer)
 		json.NewEncoder(b).Encode(req)
 		resp, err := client.Post(c.SandboxURL, "application/json; charset=utf-8", b)
 		if err != nil {
diff --git a/appstore/validator_test.go b/appstore/validator_test.go
index 8e21cf0..c0df87f 100644
--- a/appstore/validator_test.go
+++ b/appstore/validator_test.go
@@ -2,10 +2,12 @@ package appstore
 
 import (
 	"errors"
+	"io/ioutil"
 	"net/http"
 	"net/http/httptest"
 	"os"
 	"reflect"
+	"strings"
 	"testing"
 	"time"
 )
@@ -181,7 +183,7 @@ func TestVerifyBadURL(t *testing.T) {
 }
 
 func TestVerifyBadPayload(t *testing.T) {
-	s := httptest.NewServer(badPayload())
+	s := httptest.NewServer(serverWithResponse(`{"status": 21002}`))
 	defer s.Close()
 
 	client := New()
@@ -204,7 +206,7 @@ func TestVerifyBadPayload(t *testing.T) {
 }
 
 func TestVerifyBadResponse(t *testing.T) {
-	s := httptest.NewServer(invalidResponse())
+	s := httptest.NewServer(serverWithResponse(`qwerty!@#$%^`))
 	defer s.Close()
 
 	client := New()
@@ -221,10 +223,10 @@ func TestVerifyBadResponse(t *testing.T) {
 }
 
 func TestVerifySandboxReceipt(t *testing.T) {
-	s := httptest.NewServer(redirectToSandbox())
+	s := httptest.NewServer(serverWithResponse(`{"status": 21007}`))
 	defer s.Close()
 
-	sandboxServ := httptest.NewServer(sandboxSuccess())
+	sandboxServ := httptest.NewServer(serverWithResponse(`{"status": 0}`))
 	defer sandboxServ.Close()
 
 	client := New()
@@ -250,7 +252,7 @@ func TestVerifySandboxReceipt(t *testing.T) {
 }
 
 func TestVerifySandboxReceiptFailure(t *testing.T) {
-	s := httptest.NewServer(redirectToSandbox())
+	s := httptest.NewServer(serverWithResponse(`{"status": 21007}`))
 	defer s.Close()
 
 	client := New()
@@ -269,11 +271,35 @@ func TestVerifySandboxReceiptFailure(t *testing.T) {
 	}
 }
 
-func badPayload() http.Handler {
+func TestCannotReadBody(t *testing.T) {
+	client := New()
+	testResponse := http.Response{Body: ioutil.NopCloser(errReader(0))}
+
+	if client.parseResponse(&testResponse, IAPResponse{}, http.Client{}, IAPRequest{}) == nil {
+		t.Errorf("expected redirectToSandbox to fail to read the body")
+	}
+}
+
+func TestCannotUnmarshalBody(t *testing.T) {
+	client := New()
+	testResponse := http.Response{Body: ioutil.NopCloser(strings.NewReader(`{"status": true}`))}
+
+	if client.parseResponse(&testResponse, StatusResponse{}, http.Client{}, IAPRequest{}) == nil {
+		t.Errorf("expected redirectToSandbox to fail to unmarshal the data")
+	}
+}
+
+type errReader int
+
+func (errReader) Read(p []byte) (n int, err error) {
+	return 0, errors.New("test error")
+}
+
+func serverWithResponse(response string) http.Handler {
 	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		if "POST" == r.Method {
 			w.Header().Set("Content-Type", "application/json")
-			w.Write([]byte(`{"status": 21002}`))
+			w.Write([]byte(response))
 			return
 		} else {
 			w.Write([]byte(`unsupported request`))
@@ -282,45 +308,3 @@ func badPayload() http.Handler {
 		w.WriteHeader(http.StatusBadRequest)
 	})
 }
-
-func redirectToSandbox() http.Handler {
-	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-		if "POST" == r.Method {
-			w.Header().Set("Content-Type", "application/json")
-			w.Write([]byte(`{"status": 21007}`))
-			return
-		} else {
-			w.Write([]byte(`unsupported request`))
-		}
-
-		w.WriteHeader(http.StatusOK)
-	})
-}
-
-func sandboxSuccess() http.Handler {
-	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-		if "POST" == r.Method {
-			w.Header().Set("Content-Type", "application/json")
-			w.Write([]byte(`{"status": 0}`))
-			return
-		} else {
-			w.Write([]byte(`unsupported request`))
-		}
-
-		w.WriteHeader(http.StatusOK)
-	})
-}
-
-func invalidResponse() http.Handler {
-	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
-		if "POST" == r.Method {
-			w.Header().Set("Content-Type", "application/json")
-			w.Write([]byte(`qwerty!@#$%^`))
-			return
-		} else {
-			w.Write([]byte(`unsupported request`))
-		}
-
-		w.WriteHeader(http.StatusOK)
-	})
-}