util_test.go 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283
  1. package util
  2. import (
  3. "github.com/stretchr/testify/require"
  4. "io/ioutil"
  5. "os"
  6. "path/filepath"
  7. "testing"
  8. "time"
  9. )
  10. func TestDurationToHuman_SevenDays(t *testing.T) {
  11. d := 7 * 24 * time.Hour
  12. require.Equal(t, "7d", DurationToHuman(d))
  13. }
  14. func TestDurationToHuman_MoreThanOneDay(t *testing.T) {
  15. d := 49 * time.Hour
  16. require.Equal(t, "2d1h", DurationToHuman(d))
  17. }
  18. func TestDurationToHuman_LessThanOneDay(t *testing.T) {
  19. d := 17*time.Hour + 15*time.Minute
  20. require.Equal(t, "17h15m", DurationToHuman(d))
  21. }
  22. func TestDurationToHuman_TenOfThings(t *testing.T) {
  23. d := 10*time.Hour + 10*time.Minute + 10*time.Second
  24. require.Equal(t, "10h10m10s", DurationToHuman(d))
  25. }
  26. func TestDurationToHuman_Zero(t *testing.T) {
  27. require.Equal(t, "0", DurationToHuman(0))
  28. }
  29. func TestRandomString(t *testing.T) {
  30. s1 := RandomString(10)
  31. s2 := RandomString(10)
  32. s3 := RandomString(12)
  33. require.Equal(t, 10, len(s1))
  34. require.Equal(t, 10, len(s2))
  35. require.Equal(t, 12, len(s3))
  36. require.NotEqual(t, s1, s2)
  37. }
  38. func TestFileExists(t *testing.T) {
  39. filename := filepath.Join(t.TempDir(), "somefile.txt")
  40. require.Nil(t, ioutil.WriteFile(filename, []byte{0x25, 0x86}, 0600))
  41. require.True(t, FileExists(filename))
  42. require.False(t, FileExists(filename+".doesnotexist"))
  43. }
  44. func TestInStringList(t *testing.T) {
  45. s := []string{"one", "two"}
  46. require.True(t, InStringList(s, "two"))
  47. require.False(t, InStringList(s, "three"))
  48. }
  49. func TestExpandHome_WithTilde(t *testing.T) {
  50. require.Equal(t, os.Getenv("HOME")+"/this/is/a/path", ExpandHome("~/this/is/a/path"))
  51. }
  52. func TestExpandHome_NoTilde(t *testing.T) {
  53. require.Equal(t, "/this/is/an/absolute/path", ExpandHome("/this/is/an/absolute/path"))
  54. }
  55. func TestParsePriority(t *testing.T) {
  56. priorities := []string{"", "1", "2", "3", "4", "5", "min", "LOW", " default ", "HIgh", "max", "urgent"}
  57. expected := []int{0, 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, 5}
  58. for i, priority := range priorities {
  59. actual, err := ParsePriority(priority)
  60. require.Nil(t, err)
  61. require.Equal(t, expected[i], actual)
  62. }
  63. }
  64. func TestParsePriority_Invalid(t *testing.T) {
  65. priorities := []string{"-1", "6", "aa", "-"}
  66. for _, priority := range priorities {
  67. _, err := ParsePriority(priority)
  68. require.Equal(t, errInvalidPriority, err)
  69. }
  70. }