Enhance scan error handling and add tests for scan functionality

Author: hahwul

codecov.yml

@@ -1,2 +1,3 @@
 ignore:  
-  - cmd
+  - cmd
+  - pkg/server/docs/docs.go

pkg/server/scan.go

@@ -34,7 +34,12 @@ func ScanFromAPI(url string, rqOptions model.Options, options model.Options, sid
 	escapedURL := cleanURL(url)
 	vLog.WithField("data1", sid).Debug(escapedURL)
 	vLog.WithField("data1", sid).Debug(newOptions)
-	_, _ = scan.Scan(url, newOptions, sid)
+	_, err := scan.Scan(url, newOptions, sid)
+	if err != nil {
+		vLog.WithField("data1", sid).Error("Scan failed: ", err)
+		return
+	}
+	vLog.WithField("data1", sid).Info("Scan completed successfully")
 }
 
 // GetScan is get scan information

pkg/server/server_test.go

@@ -1,6 +1,8 @@
 package server
 
 import (
+	"bytes"
+	"encoding/json"
 	"net/http"
 	"net/http/httptest"
 	"testing"
@@ -77,3 +79,74 @@ func Test_healthHandler(t *testing.T) {
 		assert.Contains(t, rec.Body.String(), "ok")
 	}
 }
+
+func Test_postScanHandler(t *testing.T) {
+	e := echo.New()
+	rq := Req{
+		URL: "http://example.com",
+		Options: model.Options{
+			Method: "GET",
+		},
+	}
+	body, _ := json.Marshal(rq)
+	req := httptest.NewRequest(http.MethodPost, "/scan", bytes.NewReader(body))
+	req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
+	rec := httptest.NewRecorder()
+	c := e.NewContext(req, rec)
+
+	scans := []string{}
+	options := model.Options{
+		Scan: map[string]model.Scan{},
+	}
+
+	if assert.NoError(t, postScanHandler(c, &scans, options)) {
+		assert.Equal(t, http.StatusOK, rec.Code)
+		assert.Contains(t, rec.Body.String(), "code")
+		assert.Contains(t, rec.Body.String(), "msg")
+		assert.Contains(t, rec.Body.String(), "data")
+	}
+}
+
+func Test_GetScan(t *testing.T) {
+	options := model.Options{
+		Scan: map[string]model.Scan{
+			"test-scan": {URL: "http://example.com", Results: []model.PoC{{Type: "finish"}}},
+		},
+	}
+	scan := GetScan("test-scan", options)
+	assert.Equal(t, "http://example.com", scan.URL)
+	assert.Equal(t, "finish", scan.Results[0].Type)
+}
+
+func Test_GetScans(t *testing.T) {
+	options := model.Options{
+		Scan: map[string]model.Scan{
+			"test-scan1": {URL: "http://example1.com"},
+			"test-scan2": {URL: "http://example2.com"},
+		},
+	}
+	scans := GetScans(options)
+	assert.Contains(t, scans, "test-scan1")
+	assert.Contains(t, scans, "test-scan2")
+}
+
+func Test_ScanFromAPI(t *testing.T) {
+	options := model.Options{
+		Debug: true,
+		Scan:  map[string]model.Scan{},
+	}
+	rqOptions := model.Options{
+		Method: "GET",
+	}
+	sid := "test-scan-id"
+
+	t.Run("Successful Scan", func(t *testing.T) {
+		ScanFromAPI("http://example.com", rqOptions, options, sid)
+		// Add assertions to verify the scan was successful
+	})
+
+	t.Run("Scan with Error", func(t *testing.T) {
+		ScanFromAPI("http://invalid-url", rqOptions, options, sid)
+		// Add assertions to verify error handling
+	})
+}