package utils import ( "fmt" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" "net/http" "net/http/httptest" "os" "testing" "time" ) func TestCreateLog(t *testing.T) { Directory = os.Getenv("STATPING_DIR") err := createLog(Directory) assert.Nil(t, err) } func TestReplaceValue(t *testing.T) { assert.Equal(t, true, replaceVal(true)) assert.Equal(t, 42, replaceVal(42)) assert.Equal(t, "hello world", replaceVal("hello world")) assert.Equal(t, "5s", replaceVal(5*time.Second)) } func TestInitLogs(t *testing.T) { assert.Nil(t, InitLogs()) require.NotEmpty(t, Params.GetString("STATPING_DIR")) require.False(t, Params.GetBool("DISABLE_LOGS")) Log.Infoln("this is a test") assert.FileExists(t, Directory+"/logs/statping.log") } func TestDir(t *testing.T) { assert.Contains(t, Directory, "statping/statping") } func TestCommand(t *testing.T) { t.SkipNow() _, out, err := Command("/bin/echo", "\"statping testing\"") assert.Nil(t, err) assert.Contains(t, out, "statping") } func TestToInt(t *testing.T) { assert.Equal(t, int64(55), ToInt("55")) assert.Equal(t, int64(55), ToInt(55)) assert.Equal(t, int64(55), ToInt(55.0)) assert.Equal(t, int64(55), ToInt([]byte("55"))) } func TestToString(t *testing.T) { assert.Equal(t, "55", ToString(55)) assert.Equal(t, "55.000000", ToString(55.0)) assert.Equal(t, "55", ToString([]byte("55"))) dir, _ := time.ParseDuration("55s") assert.Equal(t, "55s", ToString(dir)) assert.Equal(t, "true", ToString(true)) assert.Equal(t, Now().Format("Monday January _2, 2006 at 03:04PM"), ToString(Now())) } func ExampleToString() { amount := 42 fmt.Print(ToString(amount)) // Output: 42 } func TestSaveFile(t *testing.T) { assert.Nil(t, SaveFile(Directory+"/test.txt", []byte("testing saving a file"))) } func TestOpenFile(t *testing.T) { f, err := OpenFile(Directory + "/test.txt") require.Nil(t, err) assert.Equal(t, "testing saving a file", f) } func TestFileExists(t *testing.T) { assert.True(t, FileExists(Directory+"/test.txt")) assert.False(t, FileExists(Directory+"fake.txt")) } func TestDeleteFile(t *testing.T) { assert.Nil(t, DeleteFile(Directory+"/test.txt")) assert.Error(t, DeleteFile(Directory+"/missingfilehere.txt")) } func TestFormatDuration(t *testing.T) { dur, _ := time.ParseDuration("158s") assert.Equal(t, "2 minutes 38 seconds", FormatDuration(dur)) dur, _ = time.ParseDuration("-65s") assert.Equal(t, "-1 minute 5 seconds", FormatDuration(dur)) dur, _ = time.ParseDuration("3s") assert.Equal(t, "3 seconds", FormatDuration(dur)) dur, _ = time.ParseDuration("48h") assert.Equal(t, "2 days", FormatDuration(dur)) dur, _ = time.ParseDuration("12h") assert.Equal(t, "12 hours", FormatDuration(dur)) } func ExampleDurationReadable() { dur, _ := time.ParseDuration("25m") readable := DurationReadable(dur) fmt.Print(readable) // Output: 25 minutes } func TestLogHTTP(t *testing.T) { req, err := http.NewRequest("GET", "/", nil) assert.Nil(t, err) assert.NotNil(t, req) } func TestStringInt(t *testing.T) { assert.Equal(t, "1", ToString("1")) } func ExampleStringInt() { amount := "42" fmt.Print(ToString(amount)) // Output: 42 } func TestHashPassword(t *testing.T) { pass := HashPassword("password123") assert.Equal(t, 60, len(pass)) assert.True(t, CheckHash("password123", pass)) assert.False(t, CheckHash("wrongpasswd", pass)) } func TestHuman(t *testing.T) { assert.Equal(t, "10 seconds", Duration{10 * time.Second}.Human()) assert.Equal(t, "1 day 12 hours", Duration{36 * time.Hour}.Human()) assert.Equal(t, "45 minutes", Duration{45 * time.Minute}.Human()) } func TestSha256Hash(t *testing.T) { assert.Equal(t, "ef92b778bafe771e89245b89ecbc08a44a4e166c06659911881f383d4473e94f", Sha256Hash("password123")) } func TestNotNumbber(t *testing.T) { assert.True(t, NotNumber("notint")) assert.True(t, NotNumber("1293notanint922")) assert.False(t, NotNumber("0")) assert.False(t, NotNumber("5")) } func TestNewSHA1Hash(t *testing.T) { hash := NewSHA256Hash() assert.NotEmpty(t, hash) assert.Len(t, hash, 64) assert.Len(t, NewSHA256Hash(), 64) assert.NotEqual(t, hash, NewSHA256Hash()) } func TestRandomString(t *testing.T) { assert.NotEmpty(t, RandomString(5)) } func TestDeleteDirectory(t *testing.T) { assert.Nil(t, DeleteDirectory(Directory+"/logs")) } func TestRenameDirectory(t *testing.T) { assert.Nil(t, CreateDirectory(Directory+"/example")) require.DirExists(t, Directory+"/example") assert.Nil(t, RenameDirectory(Directory+"/example", Directory+"/renamed_example")) require.DirExists(t, Directory+"/renamed_example") assert.Nil(t, os.RemoveAll(Directory+"/renamed_example")) } func TestHttpRequest(t *testing.T) { // Start a local HTTP server server := httptest.NewServer(http.HandlerFunc(func(rw http.ResponseWriter, req *http.Request) { // Test request parameters assert.Equal(t, req.URL.String(), "/") assert.Equal(t, req.Header["Aaa"], []string{"bbbb="}) assert.Equal(t, req.Header["Ccc"], []string{"ddd"}) // Send response to be tested rw.Write([]byte(`OK`)) })) // Close the server when test finishes defer server.Close() body, resp, err := HttpRequest(server.URL, "GET", "application/json", []string{"aaa=bbbb=", "ccc=ddd"}, nil, 2*time.Second, false, nil) assert.Nil(t, err) assert.Equal(t, []byte("OK"), body) assert.Equal(t, resp.StatusCode, 200) } func TestConfigLoad(t *testing.T) { InitLogs() InitEnvs() s := Params.GetString b := Params.GetBool Params.Set("DB_CONN", "sqlite") assert.Equal(t, "sqlite", s("DB_CONN")) assert.Equal(t, Directory, s("STATPING_DIR")) assert.True(t, b("SAMPLE_DATA")) assert.True(t, b("ALLOW_REPORTS")) } func TestPerlin(t *testing.T) { p := NewPerlin(2, 2, 5, Now().UnixNano()) require.NotNil(t, p) for hi := 1.; hi <= 100.; hi++ { assert.NotZero(t, p.Noise1D(hi/500)) } }