Indicate duplicate upload slots to the client

test/main_test.go

@@ -4,6 +4,7 @@ import (
 	"github.com/jinzhu/gorm"
 	_ "github.com/jinzhu/gorm/dialects/sqlite"
 	"github.com/simon987/ws_bucket/api"
+	"os"
 	"testing"
 	"time"
 )
@@ -22,4 +23,5 @@ func TestMain(m *testing.M) {
 	time.Sleep(time.Millisecond * 100)
 
 	m.Run()
+	_ = os.Remove("test.db")
 }

test/slot_test.go

@@ -46,6 +46,25 @@ func TestAllocateUploadSlotUnsafePath(t *testing.T) {
 	}
 }
 
+func TestDuplicateUploadSlot(t *testing.T) {
+
+	if allocateUploadSlot(api.AllocateUploadSlotRequest{
+		FileName: "test.png",
+		Token:    "testdupe",
+		MaxSize:  100,
+	}).Ok != true {
+		t.Error()
+	}
+
+	if allocateUploadSlot(api.AllocateUploadSlotRequest{
+		FileName: "test.png",
+		Token:    "testdupe",
+		MaxSize:  100,
+	}).Ok != false {
+		t.Error()
+	}
+}
+
 func allocateUploadSlot(request api.AllocateUploadSlotRequest) (ar *api.GenericResponse) {
 	resp := Post("/slot", request)
 	UnmarshalResponse(resp, &ar)