From dad9562dfc5599bc8418df830757195ab4c98ad5 Mon Sep 17 00:00:00 2001
From: Brandur <brandur@mutelight.org>
Date: Wed, 26 Jul 2017 12:27:15 -0700
Subject: [PATCH] Set special `Stripelocal-Version` response header

Sets a special `Stripelocal-Version` response header that will allow
library test suites to judge whether they need a newer version of
stripelocal than the one that's currently running.
---
 server.go      |  2 ++
 server_test.go | 15 ++++++++++++++-
 2 files changed, 16 insertions(+), 1 deletion(-)

diff --git a/server.go b/server.go
index f49856cd..d6499c75 100644
--- a/server.go
+++ b/server.go
@@ -305,6 +305,8 @@ func writeResponse(w http.ResponseWriter, start time.Time, status int, data inte
 		return
 	}
 
+	w.Header().Set("Stripelocal-Version", version)
+
 	w.WriteHeader(status)
 	_, err = w.Write(encodedData)
 	if err != nil {
diff --git a/server_test.go b/server_test.go
index 834b0b0e..1611434e 100644
--- a/server_test.go
+++ b/server_test.go
@@ -11,6 +11,19 @@ import (
 	assert "github.com/stretchr/testify/require"
 )
 
+func TestStubServer_SetsSpecialHeaders(t *testing.T) {
+	server := getStubServer(t)
+
+	// Does this regardless of endpoint
+	req := httptest.NewRequest("GET", "https://stripe.com/", nil)
+	w := httptest.NewRecorder()
+	server.HandleRequest(w, req)
+
+	resp := w.Result()
+	assert.Equal(t, http.StatusNotFound, resp.StatusCode)
+	assert.Equal(t, version, resp.Header.Get("Stripelocal-Version"))
+}
+
 func TestStubServer_ParameterValidation(t *testing.T) {
 	server := getStubServer(t)
 
@@ -23,7 +36,7 @@ func TestStubServer_ParameterValidation(t *testing.T) {
 	assert.NoError(t, err)
 
 	assert.Contains(t, string(body), "property 'amount' is required")
-	assert.Equal(t, 400, resp.StatusCode)
+	assert.Equal(t, http.StatusBadRequest, resp.StatusCode)
 }
 
 func TestStubServer_RoutesRequest(t *testing.T) {