From 5944259f94c0debe74e7f4f1ee9350f6c1a68e54 Mon Sep 17 00:00:00 2001
From: Christian Muehlhaeuser <muesli@gmail.com>
Date: Thu, 25 May 2017 12:02:11 +0200
Subject: [PATCH] Added tests for error handling

---
 toktok_test.go | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/toktok_test.go b/toktok_test.go
index 8a36f9f..eba7359 100644
--- a/toktok_test.go
+++ b/toktok_test.go
@@ -24,9 +24,20 @@ func TestCodeGen(t *testing.T) {
 	}
 }
 
+func TestCodeError(t *testing.T) {
+	_, err := NewBucket(1)
+	if err != ErrTokenLengthTooSmall {
+		t.Errorf("Expected error %v, got %v", ErrTokenLengthTooSmall, err)
+	}
+
+	_, err = NewBucketWithRunes(8, "foo")
+	if err != ErrTooFewRunes {
+		t.Errorf("Expected error %v, got %v", ErrTooFewRunes, err)
+	}
+}
+
 func TestCodeResolve(t *testing.T) {
-	length := uint(8)
-	bucket, err := NewBucket(length)
+	bucket, err := NewBucket(8)
 	if err != nil {
 		t.Error("Error creating new token bucket:", err)
 	}
@@ -49,8 +60,7 @@ func TestCodeResolve(t *testing.T) {
 }
 
 func TestCodeFaultyResolve(t *testing.T) {
-	length := uint(8)
-	bucket, err := NewBucket(length)
+	bucket, err := NewBucket(8)
 	if err != nil {
 		t.Error("Error creating new token bucket:", err)
 	}