From 53efb84a58c25a3ca80c4d3af3a0bab107aa92d5 Mon Sep 17 00:00:00 2001
From: Hayden Parker <hayden@hkparker.com>
Date: Sun, 3 Jul 2016 20:57:36 -0700
Subject: [PATCH] lease set lease count tests

---
 lib/common/lease_set_test.go | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/lib/common/lease_set_test.go b/lib/common/lease_set_test.go
index 6b5a464..5328768 100644
--- a/lib/common/lease_set_test.go
+++ b/lib/common/lease_set_test.go
@@ -54,6 +54,27 @@ func TestLeaseCountCorrect(t *testing.T) {
 	}
 }
 
+func TestLeaseCountCorrectWithMultiple(t *testing.T) {
+	assert := assert.New(t)
+
+	lease_set := buildFullLeaseSet(3)
+	count, err := lease_set.LeaseCount()
+	if assert.Nil(err) {
+		assert.Equal(3, count)
+	}
+}
+
+func TestLeaseCountErrorWithTooMany(t *testing.T) {
+	assert := assert.New(t)
+
+	lease_set := buildFullLeaseSet(17)
+	count, err := lease_set.LeaseCount()
+	if assert.NotNil(err) {
+		assert.Equal("invalid lease set: more than 16 leases", err.Error())
+	}
+	assert.Equal(17, count)
+}
+
 // TestLeases
 
 // TestSignature