From 8f07f6f4d7e91a9e4b7164a39759907fac5fb8a1 Mon Sep 17 00:00:00 2001 From: Keuin Date: Sat, 1 Jul 2023 22:06:01 +0800 Subject: Refactor: encapsulate custom url manipulation function into object method. --- common/myurl/urlparse_test.go | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 common/myurl/urlparse_test.go (limited to 'common/myurl/urlparse_test.go') diff --git a/common/myurl/urlparse_test.go b/common/myurl/urlparse_test.go new file mode 100644 index 0000000..b7df7bd --- /dev/null +++ b/common/myurl/urlparse_test.go @@ -0,0 +1,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) + } + } +} -- cgit v1.2.3