aboutsummaryrefslogtreecommitdiffstats
path: root/models/db/engine_test.go
diff options
context:
space:
mode:
authorLunny Xiao <xiaolunwen@gmail.com>2022-06-15 15:02:00 +0800
committerGitHub <noreply@github.com>2022-06-15 09:02:00 +0200
commitd91d4db344792760a18265c6902d4ac909d0ee84 (patch)
treee32232b041671df97164c4398e48316744c32ceb /models/db/engine_test.go
parentb4d420d8658feced78d897de4fbc9cc6ca76ab4e (diff)
downloadgitea-d91d4db344792760a18265c6902d4ac909d0ee84.tar.gz
gitea-d91d4db344792760a18265c6902d4ac909d0ee84.zip
Move tests as seperate sub packages to reduce duplicated file names (#19951)
Diffstat (limited to 'models/db/engine_test.go')
-rw-r--r--models/db/engine_test.go56
1 files changed, 56 insertions, 0 deletions
diff --git a/models/db/engine_test.go b/models/db/engine_test.go
new file mode 100644
index 0000000000..41279c5005
--- /dev/null
+++ b/models/db/engine_test.go
@@ -0,0 +1,56 @@
+// Copyright 2019 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 db_test
+
+import (
+ "os"
+ "path/filepath"
+ "testing"
+
+ "code.gitea.io/gitea/models/db"
+ issues_model "code.gitea.io/gitea/models/issues"
+ "code.gitea.io/gitea/models/unittest"
+ "code.gitea.io/gitea/modules/setting"
+
+ "github.com/stretchr/testify/assert"
+)
+
+func TestDumpDatabase(t *testing.T) {
+ assert.NoError(t, unittest.PrepareTestDatabase())
+
+ dir, err := os.MkdirTemp(os.TempDir(), "dump")
+ assert.NoError(t, err)
+
+ type Version struct {
+ ID int64 `xorm:"pk autoincr"`
+ Version int64
+ }
+ assert.NoError(t, db.GetEngine(db.DefaultContext).Sync2(new(Version)))
+
+ for _, dbType := range setting.SupportedDatabaseTypes {
+ assert.NoError(t, db.DumpDatabase(filepath.Join(dir, dbType+".sql"), dbType))
+ }
+}
+
+func TestDeleteOrphanedObjects(t *testing.T) {
+ assert.NoError(t, unittest.PrepareTestDatabase())
+
+ countBefore, err := db.GetEngine(db.DefaultContext).Count(&issues_model.PullRequest{})
+ assert.NoError(t, err)
+
+ _, err = db.GetEngine(db.DefaultContext).Insert(&issues_model.PullRequest{IssueID: 1000}, &issues_model.PullRequest{IssueID: 1001}, &issues_model.PullRequest{IssueID: 1003})
+ assert.NoError(t, err)
+
+ orphaned, err := db.CountOrphanedObjects("pull_request", "issue", "pull_request.issue_id=issue.id")
+ assert.NoError(t, err)
+ assert.EqualValues(t, 3, orphaned)
+
+ err = db.DeleteOrphanedObjects("pull_request", "issue", "pull_request.issue_id=issue.id")
+ assert.NoError(t, err)
+
+ countAfter, err := db.GetEngine(db.DefaultContext).Count(&issues_model.PullRequest{})
+ assert.NoError(t, err)
+ assert.EqualValues(t, countBefore, countAfter)
+}