From bd6f24b4abcd932d55e9d01af3b6992843901d89 Mon Sep 17 00:00:00 2001
From: Lauris BH <lauris@nix.lv>
Date: Sat, 4 Nov 2017 15:31:59 +0200
Subject: [PATCH] Add deleted_branch table fixture (#2832)

* Add deleted_branch table fixture

* Fix deleted branches tests

* Remove unneeded global variable
---
 models/branches_test.go            | 37 +++++++++++-------------------
 models/fixtures/deleted_branch.yml | 15 ++++++++++++
 2 files changed, 29 insertions(+), 23 deletions(-)
 create mode 100644 models/fixtures/deleted_branch.yml

diff --git a/models/branches_test.go b/models/branches_test.go
index 512104cdd..b7984331e 100644
--- a/models/branches_test.go
+++ b/models/branches_test.go
@@ -10,30 +10,17 @@ import (
 	"github.com/stretchr/testify/assert"
 )
 
-var firstBranch = DeletedBranch{
-	ID:          1,
-	Name:        "foo",
-	Commit:      "1213212312313213213132131",
-	DeletedByID: int64(1),
-}
-
-var secondBranch = DeletedBranch{
-	ID:          2,
-	Name:        "bar",
-	Commit:      "5655464564554545466464655",
-	DeletedByID: int64(99),
-}
-
 func TestAddDeletedBranch(t *testing.T) {
 	assert.NoError(t, PrepareTestDatabase())
 	repo := AssertExistsAndLoadBean(t, &Repository{ID: 1}).(*Repository)
-	assert.NoError(t, repo.AddDeletedBranch(firstBranch.Name, firstBranch.Commit, firstBranch.DeletedByID))
+	firstBranch := AssertExistsAndLoadBean(t, &DeletedBranch{ID: 1}).(*DeletedBranch)
+
 	assert.Error(t, repo.AddDeletedBranch(firstBranch.Name, firstBranch.Commit, firstBranch.DeletedByID))
-	assert.NoError(t, repo.AddDeletedBranch(secondBranch.Name, secondBranch.Commit, secondBranch.DeletedByID))
+	assert.NoError(t, repo.AddDeletedBranch("test", "5655464564554545466464656", int64(1)))
 }
+
 func TestGetDeletedBranches(t *testing.T) {
 	assert.NoError(t, PrepareTestDatabase())
-	AssertExistsAndLoadBean(t, &DeletedBranch{ID: 1})
 	repo := AssertExistsAndLoadBean(t, &Repository{ID: 1}).(*Repository)
 
 	branches, err := repo.GetDeletedBranches()
@@ -43,11 +30,17 @@ func TestGetDeletedBranches(t *testing.T) {
 
 func TestGetDeletedBranch(t *testing.T) {
 	assert.NoError(t, PrepareTestDatabase())
+	firstBranch := AssertExistsAndLoadBean(t, &DeletedBranch{ID: 1}).(*DeletedBranch)
+
 	assert.NotNil(t, getDeletedBranch(t, firstBranch))
 }
 
 func TestDeletedBranchLoadUser(t *testing.T) {
 	assert.NoError(t, PrepareTestDatabase())
+
+	firstBranch := AssertExistsAndLoadBean(t, &DeletedBranch{ID: 1}).(*DeletedBranch)
+	secondBranch := AssertExistsAndLoadBean(t, &DeletedBranch{ID: 2}).(*DeletedBranch)
+
 	branch := getDeletedBranch(t, firstBranch)
 	assert.Nil(t, branch.DeletedBy)
 	branch.LoadUser()
@@ -63,19 +56,17 @@ func TestDeletedBranchLoadUser(t *testing.T) {
 
 func TestRemoveDeletedBranch(t *testing.T) {
 	assert.NoError(t, PrepareTestDatabase())
-
-	branch := DeletedBranch{ID: 1}
-	AssertExistsAndLoadBean(t, &branch)
 	repo := AssertExistsAndLoadBean(t, &Repository{ID: 1}).(*Repository)
 
+	firstBranch := AssertExistsAndLoadBean(t, &DeletedBranch{ID: 1}).(*DeletedBranch)
+
 	err := repo.RemoveDeletedBranch(1)
 	assert.NoError(t, err)
-	AssertNotExistsBean(t, &branch)
+	AssertNotExistsBean(t, firstBranch)
 	AssertExistsAndLoadBean(t, &DeletedBranch{ID: 2})
 }
 
-func getDeletedBranch(t *testing.T, branch DeletedBranch) *DeletedBranch {
-	AssertExistsAndLoadBean(t, &DeletedBranch{ID: 1})
+func getDeletedBranch(t *testing.T, branch *DeletedBranch) *DeletedBranch {
 	repo := AssertExistsAndLoadBean(t, &Repository{ID: 1}).(*Repository)
 
 	deletedBranch, err := repo.GetDeletedBranchByID(branch.ID)
diff --git a/models/fixtures/deleted_branch.yml b/models/fixtures/deleted_branch.yml
new file mode 100644
index 000000000..6a08a7834
--- /dev/null
+++ b/models/fixtures/deleted_branch.yml
@@ -0,0 +1,15 @@
+-
+  id: 1
+  repo_id: 1
+  name: foo
+  commit: 1213212312313213213132131
+  deleted_by_id: 1
+  deleted_unix: 978307200
+
+-
+  id: 2
+  repo_id: 1
+  name: bar
+  commit: 5655464564554545466464655
+  deleted_by_id: 99
+  deleted_unix: 978307200