fix: gen code generator; some unit tests

This commit is contained in:
Jacky 2024-10-22 17:27:40 +08:00
parent 33a996e777
commit 918f920d57
No known key found for this signature in database
GPG key ID: 215C21B10DF38B4D
8 changed files with 86 additions and 178 deletions

View file

@ -3,45 +3,48 @@ package cluster
import (
"github.com/0xJacky/Nginx-UI/settings"
"github.com/stretchr/testify/assert"
"github.com/uozi-tech/cosy/sandbox"
"testing"
)
func Test_parseNodeUrl(t *testing.T) {
settings.Init("../../app.example.ini")
t.Log(settings.ClusterSettings.Node)
node := settings.ClusterSettings.Node[0]
sandbox.NewInstance("../../app.example.ini", "sqlite").
Run(func(instance *sandbox.Instance) {
t.Log(settings.ClusterSettings.Node)
node := settings.ClusterSettings.Node[0]
env, err := parseNodeUrl(node)
if err != nil {
t.Fatal(err)
}
env, err := parseNodeUrl(node)
if err != nil {
t.Fatal(err)
}
assert.Equal(t, "node1", env.Name)
assert.Equal(t, "http://10.0.0.1:9000", env.URL)
assert.Equal(t, "my-node-secret", env.Token)
assert.Equal(t, true, env.Enabled)
assert.Equal(t, "node1", env.Name)
assert.Equal(t, "http://10.0.0.1:9000", env.URL)
assert.Equal(t, "my-node-secret", env.Token)
assert.Equal(t, true, env.Enabled)
node = settings.ClusterSettings.Node[1]
node = settings.ClusterSettings.Node[1]
env, err = parseNodeUrl(node)
if err != nil {
t.Fatal(err)
}
env, err = parseNodeUrl(node)
if err != nil {
t.Fatal(err)
}
assert.Equal(t, "node2", env.Name)
assert.Equal(t, "http://10.0.0.2:9000", env.URL)
assert.Equal(t, "my-node-secret", env.Token)
assert.Equal(t, true, env.Enabled)
assert.Equal(t, "node2", env.Name)
assert.Equal(t, "http://10.0.0.2:9000", env.URL)
assert.Equal(t, "my-node-secret", env.Token)
assert.Equal(t, true, env.Enabled)
node = settings.ClusterSettings.Node[2]
node = settings.ClusterSettings.Node[2]
env, err = parseNodeUrl(node)
if err != nil {
t.Fatal(err)
}
env, err = parseNodeUrl(node)
if err != nil {
t.Fatal(err)
}
assert.Equal(t, "node3", env.Name)
assert.Equal(t, "http://10.0.0.3", env.URL)
assert.Equal(t, "my-node-secret", env.Token)
assert.Equal(t, true, env.Enabled)
assert.Equal(t, "node3", env.Name)
assert.Equal(t, "http://10.0.0.3", env.URL)
assert.Equal(t, "my-node-secret", env.Token)
assert.Equal(t, true, env.Enabled)
})
}

View file

@ -1,22 +0,0 @@
package cron
import (
"github.com/0xJacky/Nginx-UI/internal/kernal"
"github.com/0xJacky/Nginx-UI/settings"
"testing"
"time"
)
func TestRestartLogrotate(t *testing.T) {
settings.Init("../../app.ini")
kernal.InitDatabase()
InitCronJobs()
time.Sleep(5 * time.Second)
RestartLogrotate()
time.Sleep(2 * time.Second)
}