summaryrefslogtreecommitdiffstats
path: root/modules/system/appstate_test.go
diff options
context:
space:
mode:
Diffstat (limited to 'modules/system/appstate_test.go')
-rw-r--r--modules/system/appstate_test.go66
1 files changed, 66 insertions, 0 deletions
diff --git a/modules/system/appstate_test.go b/modules/system/appstate_test.go
new file mode 100644
index 0000000..2f44c7b
--- /dev/null
+++ b/modules/system/appstate_test.go
@@ -0,0 +1,66 @@
+// Copyright 2021 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package system
+
+import (
+ "testing"
+
+ "code.gitea.io/gitea/models/db"
+ "code.gitea.io/gitea/models/unittest"
+
+ "github.com/stretchr/testify/assert"
+ "github.com/stretchr/testify/require"
+)
+
+func TestMain(m *testing.M) {
+ unittest.MainTest(m, &unittest.TestOptions{
+ FixtureFiles: []string{""}, // load nothing
+ })
+}
+
+type testItem1 struct {
+ Val1 string
+ Val2 int
+}
+
+func (*testItem1) Name() string {
+ return "test-item1"
+}
+
+type testItem2 struct {
+ K string
+}
+
+func (*testItem2) Name() string {
+ return "test-item2"
+}
+
+func TestAppStateDB(t *testing.T) {
+ require.NoError(t, unittest.PrepareTestDatabase())
+
+ as := &DBStore{}
+
+ item1 := new(testItem1)
+ require.NoError(t, as.Get(db.DefaultContext, item1))
+ assert.Equal(t, "", item1.Val1)
+ assert.EqualValues(t, 0, item1.Val2)
+
+ item1 = new(testItem1)
+ item1.Val1 = "a"
+ item1.Val2 = 2
+ require.NoError(t, as.Set(db.DefaultContext, item1))
+
+ item2 := new(testItem2)
+ item2.K = "V"
+ require.NoError(t, as.Set(db.DefaultContext, item2))
+
+ item1 = new(testItem1)
+ require.NoError(t, as.Get(db.DefaultContext, item1))
+ assert.Equal(t, "a", item1.Val1)
+ assert.EqualValues(t, 2, item1.Val2)
+
+ item2 = new(testItem2)
+ require.NoError(t, as.Get(db.DefaultContext, item2))
+ assert.Equal(t, "V", item2.K)
+}