Платформа ЦРНП "Мирокод" для разработки проектов
https://git.mirocod.ru
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 lines
2.6 KiB
80 lines
2.6 KiB
// Copyright 2017 The Gitea Authors. All rights reserved. |
|
// Use of this source code is governed by a MIT-style |
|
// license that can be found in the LICENSE file. |
|
|
|
package models |
|
|
|
import ( |
|
"testing" |
|
|
|
"github.com/stretchr/testify/assert" |
|
) |
|
|
|
func TestAddDeletedBranch(t *testing.T) { |
|
assert.NoError(t, PrepareTestDatabase()) |
|
repo := AssertExistsAndLoadBean(t, &Repository{ID: 1}).(*Repository) |
|
firstBranch := AssertExistsAndLoadBean(t, &DeletedBranch{ID: 1}).(*DeletedBranch) |
|
|
|
assert.Error(t, repo.AddDeletedBranch(firstBranch.Name, firstBranch.Commit, firstBranch.DeletedByID)) |
|
assert.NoError(t, repo.AddDeletedBranch("test", "5655464564554545466464656", int64(1))) |
|
} |
|
|
|
func TestGetDeletedBranches(t *testing.T) { |
|
assert.NoError(t, PrepareTestDatabase()) |
|
repo := AssertExistsAndLoadBean(t, &Repository{ID: 1}).(*Repository) |
|
|
|
branches, err := repo.GetDeletedBranches() |
|
assert.NoError(t, err) |
|
assert.Len(t, branches, 2) |
|
} |
|
|
|
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() |
|
assert.NotNil(t, branch.DeletedBy) |
|
assert.Equal(t, "user1", branch.DeletedBy.Name) |
|
|
|
branch = getDeletedBranch(t, secondBranch) |
|
assert.Nil(t, branch.DeletedBy) |
|
branch.LoadUser() |
|
assert.NotNil(t, branch.DeletedBy) |
|
assert.Equal(t, "Ghost", branch.DeletedBy.Name) |
|
} |
|
|
|
func TestRemoveDeletedBranch(t *testing.T) { |
|
assert.NoError(t, PrepareTestDatabase()) |
|
repo := AssertExistsAndLoadBean(t, &Repository{ID: 1}).(*Repository) |
|
|
|
firstBranch := AssertExistsAndLoadBean(t, &DeletedBranch{ID: 1}).(*DeletedBranch) |
|
|
|
err := repo.RemoveDeletedBranch(1) |
|
assert.NoError(t, err) |
|
AssertNotExistsBean(t, firstBranch) |
|
AssertExistsAndLoadBean(t, &DeletedBranch{ID: 2}) |
|
} |
|
|
|
func getDeletedBranch(t *testing.T, branch *DeletedBranch) *DeletedBranch { |
|
repo := AssertExistsAndLoadBean(t, &Repository{ID: 1}).(*Repository) |
|
|
|
deletedBranch, err := repo.GetDeletedBranchByID(branch.ID) |
|
assert.NoError(t, err) |
|
assert.Equal(t, branch.ID, deletedBranch.ID) |
|
assert.Equal(t, branch.Name, deletedBranch.Name) |
|
assert.Equal(t, branch.Commit, deletedBranch.Commit) |
|
assert.Equal(t, branch.DeletedByID, deletedBranch.DeletedByID) |
|
|
|
return deletedBranch |
|
}
|
|
|