Add deleted_branch table fixture (#2832)

* Add deleted_branch table fixture

* Fix deleted branches tests

* Remove unneeded global variable

diff --git a/models/branches_test.go b/models/branches_test.go
index 512104cdd0e32ebbfcfa80b6e94b5b92c068c72b..b7984331edb12bf7ade0f9b322e62140f30041a8 100644 (file)
--- 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 (file)
index 0000000..6a08a78
--- /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