From d0202aea41e985999962d76ba334622049ae9271 Mon Sep 17 00:00:00 2001
From: konrad <k@knt.li>
Date: Tue, 16 Jan 2018 11:53:27 +0100
Subject: [PATCH] Added author tests

---
 models/author_test.go | 26 ++++++++++++++++++++++++--
 1 file changed, 24 insertions(+), 2 deletions(-)

diff --git a/models/author_test.go b/models/author_test.go
index d24bba3..254fdf2 100644
--- a/models/author_test.go
+++ b/models/author_test.go
@@ -16,11 +16,24 @@ func TestAddOrUpdateAuthor(t *testing.T) {
 	author1, err := AddOrUpdateAuthor(testauthor)
 	assert.NoError(t, err)
 
+	// And anotherone
+	_, err = AddOrUpdateAuthor(testauthor)
+	assert.NoError(t, err)
+
+	// As of now, we should have 2 author in total. Get the list and check.
+	allauthors, err := ListAuthors("")
+	assert.NoError(t, err)
+
+	for _, author := range allauthors {
+		assert.Equal(t, testauthor.Forename, author.Forename)
+		assert.Equal(t, testauthor.Lastname, author.Lastname)
+	}
+
 	// Get the new author
-	author2, exists, err := GetAuthorByID(author1.ID)
+	gotauthor, exists, err := GetAuthorByID(author1.ID)
 	assert.NoError(t, err)
 	assert.True(t, exists)
-	assert.Equal(t, author2.Forename, testauthor.Forename)
+	assert.Equal(t, gotauthor.Forename, testauthor.Forename)
 
 	// Pass an empty author to see if it fails
 	_, err = AddOrUpdateAuthor(Author{})
@@ -32,4 +45,13 @@ func TestAddOrUpdateAuthor(t *testing.T) {
 	author1updated, err := AddOrUpdateAuthor(testauthor)
 	assert.NoError(t, err)
 	assert.Equal(t, testauthor.Forename, author1updated.Forename)
+
+	// Delete the author
+	err = DeleteAuthorByID(author1.ID)
+	assert.NoError(t, err)
+
+	// Check if it is gone
+	_, exists, err = GetAuthorByID(author1.ID)
+	assert.NoError(t, err)
+	assert.False(t, exists)
 }