From 495c915632d6b52e45dae4892d1e981468ad8ef6 Mon Sep 17 00:00:00 2001
From: Keuin <keuinx@gmail.com>
Date: Fri, 16 Sep 2022 21:57:59 +0800
Subject: Add tests.

---
 common/bytesize_test.go | 24 ++++++++++++++++++++++++
 common/urlparse_test.go | 32 ++++++++++++++++++++++++++++++++
 2 files changed, 56 insertions(+)
 create mode 100644 common/bytesize_test.go
 create mode 100644 common/urlparse_test.go

diff --git a/common/bytesize_test.go b/common/bytesize_test.go
new file mode 100644
index 0000000..b68ee4c
--- /dev/null
+++ b/common/bytesize_test.go
@@ -0,0 +1,24 @@
+package common
+
+import (
+	"testing"
+)
+
+func TestPrettyBytes(t *testing.T) {
+	tests := []struct {
+		Expected string
+		Actual   string
+	}{
+		{"128 Byte", PrettyBytes(128)},
+		{"128.00 KiB", PrettyBytes(128 * 1024)},
+		{"128.00 MiB", PrettyBytes(128 * 1024 * 1024)},
+		{"128.00 GiB", PrettyBytes(128 * 1024 * 1024 * 1024)},
+		{"128.00 TiB", PrettyBytes(128 * 1024 * 1024 * 1024 * 1024)},
+		{"131072.00 TiB", PrettyBytes(128 * 1024 * 1024 * 1024 * 1024 * 1024)},
+	}
+	for i, tc := range tests {
+		if tc.Expected != tc.Actual {
+			t.Fatalf("Test %v failed: %v", i, tc)
+		}
+	}
+}
diff --git a/common/urlparse_test.go b/common/urlparse_test.go
new file mode 100644
index 0000000..9ed4449
--- /dev/null
+++ b/common/urlparse_test.go
@@ -0,0 +1,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)
+		}
+	}
+}
-- 
cgit v1.2.3