mirror of https://github.com/statping/statping
150 lines
3.3 KiB
Go
150 lines
3.3 KiB
Go
// Statping
|
|
// Copyright (C) 2018. Hunter Long and the project contributors
|
|
// Written by Hunter Long <info@socialeck.com> and the project contributors
|
|
//
|
|
// https://github.com/hunterlong/statping
|
|
//
|
|
// The licenses for most software and other practical works are designed
|
|
// to take away your freedom to share and change the works. By contrast,
|
|
// the GNU General Public License is intended to guarantee your freedom to
|
|
// share and change all versions of a program--to make sure it remains free
|
|
// software for all its users.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package core
|
|
|
|
import (
|
|
"github.com/hunterlong/statping/source"
|
|
"github.com/hunterlong/statping/utils"
|
|
"github.com/stretchr/testify/assert"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
var (
|
|
dir string
|
|
skipNewDb bool
|
|
)
|
|
|
|
func init() {
|
|
dir = utils.Directory
|
|
utils.InitLogs()
|
|
source.Assets()
|
|
skipNewDb = false
|
|
SampleHits = 480
|
|
}
|
|
|
|
func TestNewCore(t *testing.T) {
|
|
if skipNewDb {
|
|
t.SkipNow()
|
|
}
|
|
utils.DeleteFile(dir + "/config.yml")
|
|
utils.DeleteFile(dir + "/statup.db")
|
|
CoreApp = NewCore()
|
|
assert.NotNil(t, CoreApp)
|
|
CoreApp.Name = "Tester"
|
|
}
|
|
|
|
func TestDbConfig_Save(t *testing.T) {
|
|
if skipNewDb {
|
|
t.SkipNow()
|
|
}
|
|
var err error
|
|
Configs = &DbConfig{
|
|
DbConn: "sqlite",
|
|
Project: "Tester",
|
|
Location: dir,
|
|
}
|
|
Configs, err = Configs.Save()
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "sqlite", Configs.DbConn)
|
|
assert.NotEmpty(t, Configs.ApiKey)
|
|
assert.NotEmpty(t, Configs.ApiSecret)
|
|
}
|
|
|
|
func TestLoadDbConfig(t *testing.T) {
|
|
Configs, err := LoadConfigFile(dir)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "sqlite", Configs.DbConn)
|
|
}
|
|
|
|
func TestDbConnection(t *testing.T) {
|
|
err := Configs.Connect(false, dir)
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestDropDatabase(t *testing.T) {
|
|
if skipNewDb {
|
|
t.SkipNow()
|
|
}
|
|
err := Configs.DropDatabase()
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestSeedSchemaDatabase(t *testing.T) {
|
|
if skipNewDb {
|
|
t.SkipNow()
|
|
}
|
|
err := Configs.CreateDatabase()
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestMigrateDatabase(t *testing.T) {
|
|
err := Configs.MigrateDatabase()
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestSeedDatabase(t *testing.T) {
|
|
if skipNewDb {
|
|
t.SkipNow()
|
|
}
|
|
err := InsertLargeSampleData()
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestReLoadDbConfig(t *testing.T) {
|
|
err := Configs.Connect(false, dir)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "sqlite", Configs.DbConn)
|
|
}
|
|
|
|
func TestSelectCore(t *testing.T) {
|
|
core, err := SelectCore()
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, "Statping Sample Data", core.Name)
|
|
}
|
|
|
|
func TestInsertNotifierDB(t *testing.T) {
|
|
if skipNewDb {
|
|
t.SkipNow()
|
|
}
|
|
err := InsertNotifierDB()
|
|
assert.Nil(t, err)
|
|
}
|
|
|
|
func TestEnvToConfig(t *testing.T) {
|
|
os.Setenv("DB_CONN", "sqlite")
|
|
os.Setenv("DB_USER", "")
|
|
os.Setenv("DB_PASS", "")
|
|
os.Setenv("DB_DATABASE", "")
|
|
os.Setenv("NAME", "Testing")
|
|
os.Setenv("DOMAIN", "http://localhost:8080")
|
|
os.Setenv("DESCRIPTION", "Testing Statping")
|
|
os.Setenv("ADMIN_USER", "admin")
|
|
os.Setenv("ADMIN_PASS", "admin123")
|
|
config, err := EnvToConfig()
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, config.DbConn, "sqlite")
|
|
assert.Equal(t, config.Domain, "http://localhost:8080")
|
|
assert.Equal(t, config.Description, "Testing Statping")
|
|
assert.Equal(t, config.Username, "admin")
|
|
assert.Equal(t, config.Password, "admin123")
|
|
}
|
|
|
|
func TestGetLocalIP(t *testing.T) {
|
|
ip := GetLocalIP()
|
|
assert.Contains(t, ip, "http://")
|
|
}
|