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