system.error: define/add severals custom error function

Author: muharihar

system/error/definitions.go

@@ -5,12 +5,74 @@ import (
 	"net/http"
 )
 
-// ForbiddenAccess error
-func ForbiddenAccess() *SystemError {
-	return CustomForbiddenAccess("Forbidden Access")
+// CustomSystemError custom SystemError
+func CustomSystemError(code int, message string) *SystemError {
+	return &SystemError{StatusCode: code, Err: fmt.Errorf(message)}
+}
+
+// CustomBadRequest custom bad request error
+func CustomBadRequest(message string) *SystemError {
+	return CustomSystemError(http.StatusBadRequest, message)
+}
+
+// BadRequest error
+//	Similar with: 400 Http Status Code
+func BadRequest() *SystemError {
+	return CustomBadRequest("Bad Request")
+}
+
+// CustomUnauthorizedAccess error
+func CustomUnauthorizedAccess(message string) *SystemError {
+	return CustomSystemError(http.StatusUnauthorized, message)
+}
+
+// UnauthorizedAccess error
+//	Similar with: 401 Http Status Code
+func UnauthorizedAccess() *SystemError {
+	return CustomUnauthorizedAccess("Unauthorized Access")
 }
 
 // CustomForbiddenAccess error
 func CustomForbiddenAccess(message string) *SystemError {
 	return &SystemError{StatusCode: http.StatusForbidden, Err: fmt.Errorf(message)}
 }
+
+// ForbiddenAccess error
+//	Similar with: 403 Http Status Code
+func ForbiddenAccess() *SystemError {
+	return CustomForbiddenAccess("Forbidden Access")
+}
+
+// CustomNotFound error
+func CustomNotFound(message string) *SystemError {
+	return CustomSystemError(http.StatusNotFound, message)
+}
+
+// NotFound error
+//	Similar withL 404 Http Status Code
+func NotFound() *SystemError {
+	return CustomNotFound("Resource Not Found")
+}
+
+// CustomConflict error
+//	Similar withL 409 Http Status Code
+func CustomConflict(message string) *SystemError {
+	return CustomSystemError(http.StatusConflict, message)
+}
+
+// Conflict error
+//	Similar withL 409 Http Status Code
+func Conflict() *SystemError {
+	return CustomConflict("Resource Conflict")
+}
+
+// CustomInternalSystemError error
+func CustomInternalSystemError(message string) *SystemError {
+	return CustomSystemError(http.StatusInternalServerError, message)
+}
+
+// InternalSystemError error
+//	Similar with: 500 Http Status Code
+func InternalSystemError() *SystemError {
+	return CustomInternalSystemError("Internal System Error")
+}

system/error/error.go

@@ -9,6 +9,14 @@ type SystemError struct {
 }
 
 // Error string
-func (r *SystemError) Error() string {
-	return fmt.Sprintf("Status %d: Error => %v", r.StatusCode, r.Err)
+func (e *SystemError) Error() string {
+	return fmt.Sprintf("error [%d]: %s", e.StatusCode, e.Err)
+}
+
+// Extensions error Extensions
+func (e *SystemError) Extensions() map[string]interface{} {
+	return map[string]interface{}{
+		"code":    e.StatusCode,
+		"message": e.Err,
+	}
 }

system/error/error_test.go

@@ -1,16 +1,90 @@
 package error
 
 import (
+	"fmt"
 	"net/http"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
 )
 
 func TestError_Methods(t *testing.T) {
-	err := ForbiddenAccess()
+	// System Error
+	err := &SystemError{StatusCode: 0, Err: fmt.Errorf("Zero")}
+	assert.Equal(t, "Zero", err.Err.Error())
+	assert.Equal(t, 0, err.StatusCode)
+	assert.Equal(t, "error [0]: Zero", err.Error())
 
+	// Custom System Error
+	err = CustomSystemError(1, "One")
+	assert.Equal(t, "One", err.Err.Error())
+	assert.Equal(t, 1, err.StatusCode)
+	assert.Equal(t, "error [1]: One", err.Error())
+
+	// Bad Request
+	err = CustomBadRequest("Bad Request")
+	assert.Equal(t, "Bad Request", err.Err.Error())
+	assert.Equal(t, http.StatusBadRequest, err.StatusCode)
+	assert.Equal(t, "error [400]: Bad Request", err.Error())
+
+	err = BadRequest()
+	assert.Equal(t, "Bad Request", err.Err.Error())
+	assert.Equal(t, http.StatusBadRequest, err.StatusCode)
+	assert.Equal(t, "error [400]: Bad Request", err.Error())
+
+	// Unauthorized Access
+	err = CustomUnauthorizedAccess("Unauthorized Access")
+	assert.Equal(t, "Unauthorized Access", err.Err.Error())
+	assert.Equal(t, http.StatusUnauthorized, err.StatusCode)
+	assert.Equal(t, "error [401]: Unauthorized Access", err.Error())
+
+	err = UnauthorizedAccess()
+	assert.Equal(t, "Unauthorized Access", err.Err.Error())
+	assert.Equal(t, http.StatusUnauthorized, err.StatusCode)
+	assert.Equal(t, "error [401]: Unauthorized Access", err.Error())
+
+	// Forbidden Access
+	err = CustomForbiddenAccess("Forbidden Access")
+	assert.Equal(t, "Forbidden Access", err.Err.Error())
+	assert.Equal(t, http.StatusForbidden, err.StatusCode)
+	assert.Equal(t, "error [403]: Forbidden Access", err.Error())
+
+	err = ForbiddenAccess()
 	assert.Equal(t, "Forbidden Access", err.Err.Error())
 	assert.Equal(t, http.StatusForbidden, err.StatusCode)
-	assert.Equal(t, "Status 403: Error => Forbidden Access", err.Error())
+	assert.Equal(t, "error [403]: Forbidden Access", err.Error())
+
+	// Not Found
+	err = CustomNotFound("Resource Not Found")
+	assert.Equal(t, "Resource Not Found", err.Err.Error())
+	assert.Equal(t, http.StatusNotFound, err.StatusCode)
+	assert.Equal(t, "error [404]: Resource Not Found", err.Error())
+
+	err = NotFound()
+	assert.Equal(t, "Resource Not Found", err.Err.Error())
+	assert.Equal(t, http.StatusNotFound, err.StatusCode)
+	assert.Equal(t, "error [404]: Resource Not Found", err.Error())
+
+	// Conflict
+	err = CustomConflict("Resource Conflict")
+	assert.Equal(t, "Resource Conflict", err.Err.Error())
+	assert.Equal(t, http.StatusConflict, err.StatusCode)
+	assert.Equal(t, "error [409]: Resource Conflict", err.Error())
+
+	err = Conflict()
+	assert.Equal(t, "Resource Conflict", err.Err.Error())
+	assert.Equal(t, http.StatusConflict, err.StatusCode)
+	assert.Equal(t, "error [409]: Resource Conflict", err.Error())
+
+	// Internal System Error
+	err = CustomInternalSystemError("Internal System Error")
+	assert.Equal(t, "Internal System Error", err.Err.Error())
+	assert.Equal(t, http.StatusInternalServerError, err.StatusCode)
+	assert.Equal(t, "error [500]: Internal System Error", err.Error())
+
+	err = InternalSystemError()
+	assert.Equal(t, "Internal System Error", err.Err.Error())
+	assert.Equal(t, http.StatusInternalServerError, err.StatusCode)
+	assert.Equal(t, "error [500]: Internal System Error", err.Error())
+
 }