blob: 9ed4449838e991a0775d7e6f50fdcaa44b0a79f2 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
|
package common
import "testing"
func TestGetFileExtensionFromUrl(t *testing.T) {
tests := []struct {
Expected string
Actual string
}{
{Expected: "html"},
{Expected: "htm"},
{Expected: "flv"},
}
var err error
tests[0].Actual, err = GetFileExtensionFromUrl("http://www.example.com/index.html")
if err != nil {
t.Fatalf("GetFileExtensionFromUrl: %v", err)
}
tests[1].Actual, err = GetFileExtensionFromUrl("https://www.example.com/index.htm")
if err != nil {
t.Fatalf("GetFileExtensionFromUrl: %v", err)
}
tests[2].Actual, err = GetFileExtensionFromUrl("https://www.example.com/video.flv?a=1&b=2flv")
if err != nil {
t.Fatalf("GetFileExtensionFromUrl: %v", err)
}
for i, tc := range tests {
if tc.Expected != tc.Actual {
t.Fatalf("Test %v failed: %v", i, tc)
}
}
}
|
