From 6e7326e7b208c5c93c9a393c094997c56569b836 Mon Sep 17 00:00:00 2001
From: Ben Schumacher <benschumacher@gmail.com>
Date: Tue, 19 May 2015 13:46:19 -0600
Subject: [PATCH] Adding tests for DelToken in pkg/server/token.go

---
 pkg/server/token_test.go | 54 +++++++++++++++++++++++++++++++++++++---
 1 file changed, 51 insertions(+), 3 deletions(-)

diff --git a/pkg/server/token_test.go b/pkg/server/token_test.go
index 854920506..66eff4bf6 100644
--- a/pkg/server/token_test.go
+++ b/pkg/server/token_test.go
@@ -4,6 +4,7 @@ import (
 	"bytes"
 	"database/sql"
 	"encoding/json"
+	"fmt"
 	"net/http"
 	"testing"
 
@@ -18,7 +19,7 @@ import (
 	"github.com/stretchr/testify/mock"
 )
 
-var tokenTests = []struct {
+var createTests = []struct {
 	inLabel  string
 	inBody   string
 	storeErr error
@@ -30,13 +31,27 @@ var tokenTests = []struct {
 	{"app2", `{"label": "app2"}`, nil, 200, types.TokenUser},
 }
 
+var deleteTests = []struct {
+	inLabel       string
+	errTokenLabel error
+	errDelToken   error
+	outCode       int
+	outToken      *types.Token
+}{
+	{"app1", sql.ErrNoRows, nil, 404, &types.Token{}},
+	{"app2", nil, sql.ErrNoRows, 400, &types.Token{Label: "app2"}},
+	{"app3", nil, nil, 200, &types.Token{Label: "app2"}},
+}
+
 func TestToken(t *testing.T) {
 	store := new(mocks.Store)
 
 	g := Goblin(t)
 	g.Describe("Token", func() {
+
+		// POST /api/user/tokens
 		g.It("should create tokens", func() {
-			for _, test := range tokenTests {
+			for _, test := range createTests {
 				rw := recorder.New()
 				ctx := gin.Context{Engine: gin.Default(), Writer: rw}
 				body := bytes.NewBufferString(test.inBody)
@@ -73,6 +88,39 @@ func TestToken(t *testing.T) {
 			}
 		})
 
-		g.It("should delete tokens")
+		// DELETE /api/user/tokens/:label
+		g.It("should delete tokens", func() {
+			for _, test := range deleteTests {
+				rw := recorder.New()
+				ctx := gin.Context{Engine: gin.Default(), Writer: rw}
+				ctx.Params = append(ctx.Params, gin.Param{Key: "label", Value: test.inLabel})
+
+				ctx.Set("datastore", store)
+				ctx.Set("user", &types.User{Login: "Freya"})
+
+				config := settings.Settings{Session: &settings.Session{Secret: "Otto"}}
+				ctx.Set("settings", &config)
+				ctx.Set("session", session.New(config.Session))
+
+				// prepare the mock
+				store.On("TokenLabel", mock.AnythingOfType("*types.User"), test.inLabel).Return(test.outToken, test.errTokenLabel).Once()
+
+				if test.errTokenLabel == nil {
+					// we don't need this expectation if we error on our first
+					store.On("DelToken", mock.AnythingOfType("*types.Token")).Return(test.errDelToken).Once()
+				}
+				fmt.Println(test)
+				DelToken(&ctx)
+
+				g.Assert(rw.Code).Equal(test.outCode)
+				if test.outCode != 200 {
+					continue
+				}
+
+				var respjson map[string]interface{}
+				json.Unmarshal(rw.Body.Bytes(), &respjson)
+				fmt.Println(rw.Code, respjson)
+			}
+		})
 	})
 }