summaryrefslogtreecommitdiff
path: root/common/myurl/urlparse_test.go
blob: b7df7bd12031a03ea1bbfc276e0ad237d4c6c18b (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 myurl

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 = Url("http://www.example.com/index.html").FileExtension()
	if err != nil {
		t.Fatalf("GetFileExtensionFromUrl: %v", err)
	}
	tests[1].Actual, err = Url("https://www.example.com/index.htm").FileExtension()
	if err != nil {
		t.Fatalf("GetFileExtensionFromUrl: %v", err)
	}
	tests[2].Actual, err = Url("https://www.example.com/video.flv?a=1&b=2flv").FileExtension()
	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)
		}
	}
}