OLD | NEW |
1 package util | 1 package util |
2 | 2 |
3 import ( | 3 import ( |
4 "fmt" | 4 "fmt" |
5 "io/ioutil" | 5 "io/ioutil" |
6 "os" | 6 "os" |
7 "path/filepath" | 7 "path/filepath" |
8 "strconv" | 8 "strconv" |
| 9 "strings" |
9 "testing" | 10 "testing" |
10 | 11 |
11 "go.skia.org/infra/go/util" | 12 "go.skia.org/infra/go/util" |
12 ) | 13 ) |
13 | 14 |
14 import ( | 15 import ( |
15 assert "github.com/stretchr/testify/require" | 16 assert "github.com/stretchr/testify/require" |
16 ) | 17 ) |
17 | 18 |
18 const ( | 19 const ( |
(...skipping 59 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
78 | 79 |
79 // Assert error returned when specified dir does not exist. | 80 // Assert error returned when specified dir does not exist. |
80 nonexistantDir := filepath.Join(os.TempDir(), "util_test_nonexistant") | 81 nonexistantDir := filepath.Join(os.TempDir(), "util_test_nonexistant") |
81 util.RemoveAll(nonexistantDir) | 82 util.RemoveAll(nonexistantDir) |
82 if err := CreateTimestampFile(nonexistantDir); err != nil { | 83 if err := CreateTimestampFile(nonexistantDir); err != nil { |
83 // Expected error | 84 // Expected error |
84 } else { | 85 } else { |
85 t.Error("Unexpected lack of error") | 86 t.Error("Unexpected lack of error") |
86 } | 87 } |
87 } | 88 } |
| 89 |
| 90 func TestGetStartRange(t *testing.T) { |
| 91 assert.Equal(t, 1, GetStartRange(1, 1000)) |
| 92 assert.Equal(t, 2001, GetStartRange(3, 1000)) |
| 93 assert.Equal(t, 41, GetStartRange(3, 20)) |
| 94 } |
| 95 |
| 96 func TestGetPathToPyFiles(t *testing.T) { |
| 97 swarmingPath := GetPathToPyFiles(true) |
| 98 assert.True(t, strings.HasSuffix(swarmingPath, filepath.Join("src", "go.
skia.org", "infra", "ct", "py"))) |
| 99 nonSwarmingPath := GetPathToPyFiles(false) |
| 100 assert.True(t, strings.HasSuffix(nonSwarmingPath, filepath.Join("src", "
go.skia.org", "infra", "ct", "py"))) |
| 101 } |
OLD | NEW |