mirror of https://github.com/goharbor/harbor.git
122 lines
3.5 KiB
Go
122 lines
3.5 KiB
Go
package dao
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/goharbor/harbor/src/common/dao"
|
|
"github.com/goharbor/harbor/src/pkg/immutabletag/dao/model"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
type immutableRuleDaoTestSuite struct {
|
|
suite.Suite
|
|
require *require.Assertions
|
|
assert *assert.Assertions
|
|
dao ImmutableRuleDao
|
|
id int64
|
|
}
|
|
|
|
func (t *immutableRuleDaoTestSuite) SetupSuite() {
|
|
t.require = require.New(t.T())
|
|
t.assert = assert.New(t.T())
|
|
dao.PrepareTestForPostgresSQL()
|
|
t.dao = New()
|
|
}
|
|
|
|
func (t *immutableRuleDaoTestSuite) TestCreateImmutableRule() {
|
|
ir := &model.ImmutableRule{TagFilter: "**", ProjectID: 1}
|
|
id, err := t.dao.CreateImmutableRule(ir)
|
|
t.require.Nil(err)
|
|
t.require.True(id > 0, "Can not create immutable tag rule")
|
|
|
|
// insert duplicate rows
|
|
ir2 := &model.ImmutableRule{TagFilter: "**", ProjectID: 1}
|
|
id2, err := t.dao.CreateImmutableRule(ir2)
|
|
t.require.True(strings.Contains(err.Error(), "duplicate key"))
|
|
t.require.Equal(int64(0), id2)
|
|
|
|
_, err = t.dao.DeleteImmutableRule(id)
|
|
t.require.Nil(err)
|
|
}
|
|
|
|
func (t *immutableRuleDaoTestSuite) TestUpdateImmutableRule() {
|
|
ir := &model.ImmutableRule{TagFilter: "**", ProjectID: 1}
|
|
id, err := t.dao.CreateImmutableRule(ir)
|
|
t.require.Nil(err)
|
|
t.require.True(id > 0, "Can not create immutable tag rule")
|
|
|
|
updatedIR := &model.ImmutableRule{ID: id, TagFilter: "1.2.0", ProjectID: 1}
|
|
updatedCnt, err := t.dao.UpdateImmutableRule(1, updatedIR)
|
|
t.require.Nil(err)
|
|
t.require.True(updatedCnt > 0, "Failed to update immutable id")
|
|
|
|
newIr, err := t.dao.GetImmutableRule(id)
|
|
t.require.Nil(err)
|
|
t.require.True(newIr.TagFilter == "1.2.0", "Failed to update immutable tag")
|
|
|
|
defer t.dao.DeleteImmutableRule(id)
|
|
|
|
}
|
|
|
|
func (t *immutableRuleDaoTestSuite) TestEnableImmutableRule() {
|
|
ir := &model.ImmutableRule{TagFilter: "**", ProjectID: 1}
|
|
id, err := t.dao.CreateImmutableRule(ir)
|
|
t.require.Nil(err)
|
|
t.require.True(id > 0, "Can not create immutable tag rule")
|
|
|
|
t.dao.ToggleImmutableRule(id, true)
|
|
newIr, err := t.dao.GetImmutableRule(id)
|
|
|
|
t.require.Nil(err)
|
|
t.require.True(newIr.Disabled, "Failed to disable the immutable rule")
|
|
|
|
defer t.dao.DeleteImmutableRule(id)
|
|
}
|
|
|
|
func (t *immutableRuleDaoTestSuite) TestGetImmutableRuleByProject() {
|
|
irs := []*model.ImmutableRule{
|
|
{TagFilter: "version1", ProjectID: 99},
|
|
{TagFilter: "version2", ProjectID: 99},
|
|
{TagFilter: "version3", ProjectID: 99},
|
|
{TagFilter: "version4", ProjectID: 99},
|
|
}
|
|
for _, ir := range irs {
|
|
t.dao.CreateImmutableRule(ir)
|
|
}
|
|
|
|
qrs, err := t.dao.QueryImmutableRuleByProjectID(99)
|
|
t.require.Nil(err)
|
|
t.require.True(len(qrs) == 4, "Failed to query 4 rows!")
|
|
|
|
defer dao.ExecuteBatchSQL([]string{"delete from immutable_tag_rule where project_id = 99 "})
|
|
|
|
}
|
|
func (t *immutableRuleDaoTestSuite) TestGetEnabledImmutableRuleByProject() {
|
|
irs := []*model.ImmutableRule{
|
|
{TagFilter: "version1", ProjectID: 99},
|
|
{TagFilter: "version2", ProjectID: 99},
|
|
{TagFilter: "version3", ProjectID: 99},
|
|
{TagFilter: "version4", ProjectID: 99},
|
|
}
|
|
for i, ir := range irs {
|
|
id, _ := t.dao.CreateImmutableRule(ir)
|
|
if i == 1 {
|
|
t.dao.ToggleImmutableRule(id, true)
|
|
}
|
|
}
|
|
|
|
qrs, err := t.dao.QueryEnabledImmutableRuleByProjectID(99)
|
|
t.require.Nil(err)
|
|
t.require.True(len(qrs) == 3, "Failed to query 3 rows!, got %v", len(qrs))
|
|
|
|
defer dao.ExecuteBatchSQL([]string{"delete from immutable_tag_rule where project_id = 99 "})
|
|
|
|
}
|
|
|
|
func TestImmutableRuleDaoTestSuite(t *testing.T) {
|
|
suite.Run(t, &immutableRuleDaoTestSuite{})
|
|
}
|