add testcase

tubone24 · Feb 6, 2021 · 3e2438e · 3e2438e
1 parent 7e9749d
commit 3e2438e
Show file tree

Hide file tree

Showing 3 changed files with 89 additions and 1 deletion.
diff --git a/pkg/utils/file.go b/pkg/utils/file.go
@@ -26,7 +26,7 @@ func ReadFile(file string) ([]byte, error) {
 	var content []byte
 	fp, err := os.Open(file)
 	if err != nil {
-		panic(err)
+		return nil, err
 	}
 	defer fp.Close()
 	buffer := make([]byte, bufferSize)

diff --git a/pkg/utils/file_test.go b/pkg/utils/file_test.go
@@ -117,6 +117,9 @@ func TestGetContentType(t *testing.T) {
 		{"test.wmv", "video/x-ms-wmv"},
 		{"test.flv", "video/x-flv"},
 		{"test.swf", "application/x-shockwave-flash"},
+		{"test.tar.gz", "application/x-tar"},
+		{"test.jar", "application/java-archive"},
+		{"test.sh", "application/x-sh"},
 		{"test.mse", "application/x-www-form-urlencoded"},
 	}
 
@@ -155,6 +158,59 @@ func TestReadWriteFile(t *testing.T) {
 	}
 }
 
+func TestWriteFileExistDir(t *testing.T) {
+	fileDir, err :=  ioutil.TempDir("", "redump_test")
+	if err != nil {
+		t.Errorf("Error occured: %s", err)
+	}
+	defer os.RemoveAll(fileDir)
+	fileDir2 := filepath.Join(fileDir, "/dummy")
+	filename := filepath.Join(fileDir, "/dummy")
+	err = utils.MakeDir(fileDir2)
+	if err != nil {
+		t.Errorf("Error occured: %s", err)
+	}
+	err = utils.WriteFile(filename, []byte("aaaa"))
+	if err == nil {
+		t.Errorf("Error not occured")
+	}
+}
+
+func TestReadFileInvalidFileName(t *testing.T) {
+	fileDir, err :=  ioutil.TempDir("", "redump_test")
+	if err != nil {
+		t.Errorf("Error occured: %s", err)
+	}
+	defer os.RemoveAll(fileDir)
+	filename := filepath.Join(fileDir, "/dummy")
+	err = utils.MakeDir(fileDir)
+	if err != nil {
+		t.Errorf("Error occured: %s", err)
+	}
+	_, err = utils.ReadFile(filename)
+	if err == nil {
+		t.Errorf("Error not occured")
+	}
+}
+
+func TestReadFileLookUpDir(t *testing.T) {
+	fileDir, err :=  ioutil.TempDir("", "redump_test")
+	if err != nil {
+		t.Errorf("Error occured: %s", err)
+	}
+	defer os.RemoveAll(fileDir)
+	fileDir2 := filepath.Join(fileDir, "/dummy")
+	filename := filepath.Join(fileDir, "/dummy")
+	err = utils.MakeDir(fileDir2)
+	if err != nil {
+		t.Errorf("Error occured: %s", err)
+	}
+	_, err = utils.ReadFile(filename)
+	if err == nil {
+		t.Errorf("Error not occured")
+	}
+}
+
 func TestCheckDir(t *testing.T) {
 	fileDir, err :=  ioutil.TempDir("", "redump_test")
 	if err != nil {

diff --git a/pkg/utils/http_test.go b/pkg/utils/http_test.go
@@ -131,6 +131,38 @@ func TestApiGet(t *testing.T) {
 	}
 }
 
+func TestApiGetInvalidUrl(t *testing.T) {
+	e := utils.NewHttpClient(1000)
+	_, err := e.Get("https://jdooudnpoowudqw.jdpofufoqf9-sjhfq-wdwhqwajasjpdjwdjqwoddjqwopdjdqdudl.com.kjpwf.coaa.coomwef.com")
+	if err == nil {
+		t.Errorf("expected error but no errors ouccured")
+	}
+}
+
+func TestApiPostInvalidUrl(t *testing.T) {
+	e := utils.NewHttpClient(1000)
+	_, err := e.Post("https://jdooudnpoowudqw.jdpofufoqf9-sjhfq-wdwhqwajasjpdjwdjqwoddjqwopdjdqdudl.com.kjpwf.coaa.coomwef.com", "application/json", []byte("{\"aaa\": \"aaa\"}"))
+	if err == nil {
+		t.Errorf("expected error but no errors ouccured")
+	}
+}
+
+func TestApiPutInvalidUrl(t *testing.T) {
+	e := utils.NewHttpClient(1000)
+	err := e.Put("https://jdooudnpoowudqw.jdpofufoqf9-sjhfq-wdwhqwajasjpdjwdjqwoddjqwopdjdqdudl.com.kjpwf.coaa.coomwef.com", "application/json", []byte("{\"aaa\": \"aaa\"}"))
+	if err == nil {
+		t.Errorf("expected error but no errors ouccured")
+	}
+}
+
+func TestApiDeleteInvalidUrl(t *testing.T) {
+	e := utils.NewHttpClient(1000)
+	err := e.Delete("https://jdooudnpoowudqw.jdpofufoqf9-sjhfq-wdwhqwajasjpdjwdjqwoddjqwopdjdqdudl.com.kjpwf.coaa.coomwef.com")
+	if err == nil {
+		t.Errorf("expected error but no errors ouccured")
+	}
+}
+
 func TestApiPost(t *testing.T) {
 	cases := map[string]struct {
 		client               *http.Client