142 lines
3.3 KiB
Go
142 lines
3.3 KiB
Go
package check
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/LazyBachelor/LazyPM/internal/models"
|
|
)
|
|
|
|
type Check = models.Check
|
|
type ValidationFeedback = models.ValidationFeedback
|
|
|
|
func NewCheck(message string, valid bool) Check {
|
|
return Check{
|
|
Message: message,
|
|
Valid: valid,
|
|
}
|
|
}
|
|
|
|
type Expector struct {
|
|
ValidationFeedback
|
|
}
|
|
|
|
func NewExpector() *Expector {
|
|
return &Expector{
|
|
ValidationFeedback: ValidationFeedback{
|
|
Success: false,
|
|
Checks: []Check{},
|
|
},
|
|
}
|
|
}
|
|
|
|
func (e *Expector) Valid() bool {
|
|
return len(e.Errors()) == 0
|
|
}
|
|
|
|
func (e *Expector) Complete() ValidationFeedback {
|
|
e.Success = len(e.Errors()) == 0
|
|
return e.ValidationFeedback
|
|
}
|
|
|
|
func (e *Expector) Fatal(message string) ValidationFeedback {
|
|
e.Success = false
|
|
e.Message = message
|
|
return e.ValidationFeedback
|
|
}
|
|
|
|
func (e *Expector) CompleteWithMessage(message string) ValidationFeedback {
|
|
e.Success = len(e.Errors()) == 0
|
|
if !e.Success {
|
|
e.Message = message
|
|
}
|
|
return e.ValidationFeedback
|
|
}
|
|
|
|
func (e *Expector) Errors() []error {
|
|
var errors []error
|
|
for _, check := range e.Checks {
|
|
if !check.Valid {
|
|
errors = append(errors, fmt.Errorf("%s", check.Message))
|
|
}
|
|
}
|
|
return errors
|
|
}
|
|
|
|
func (e *Expector) Pass(message string) *Expector {
|
|
e.Checks = append(e.Checks, NewCheck(message, true))
|
|
return e
|
|
}
|
|
|
|
func (e *Expector) Fail(message string) *Expector {
|
|
e.Checks = append(e.Checks, NewCheck(message, false))
|
|
return e
|
|
}
|
|
|
|
func (e *Expector) NotEmptyAndEqual(val, expected string, message string) *Expector {
|
|
if val == "" {
|
|
return e.Fail(fmt.Sprintf("%s is empty", message))
|
|
} else if val != expected {
|
|
return e.Fail(fmt.Sprintf(`%s expected "%v", got "%v"`, message, expected, val))
|
|
}
|
|
return e.Pass(message + " is correct")
|
|
}
|
|
|
|
func (e *Expector) Equal(val, expected any, message string) *Expector {
|
|
if val != expected {
|
|
return e.Fail(fmt.Sprintf(`%s expected "%v", got "%v"`, message, expected, val))
|
|
}
|
|
return e.Pass(message + " is correct")
|
|
}
|
|
|
|
func (e *Expector) Assert(condition bool, message string) *Expector {
|
|
if !condition {
|
|
return e.Fail(message)
|
|
}
|
|
return e.Pass(message + " is correct")
|
|
}
|
|
|
|
func (e *Expector) Nil(value any, message string) *Expector {
|
|
check := NewCheck(message, value == nil)
|
|
e.Checks = append(e.Checks, check)
|
|
return e
|
|
}
|
|
|
|
func (e *Expector) NotNil(value any, message string) *Expector {
|
|
check := NewCheck(message, value != nil)
|
|
e.Checks = append(e.Checks, check)
|
|
return e
|
|
}
|
|
|
|
func (e *Expector) Contains(s, substr, message string) *Expector {
|
|
if !strings.Contains(s, substr) {
|
|
return e.Fail(fmt.Sprintf(`%s expected to contain "%v", but it does not.`, message, substr))
|
|
} else {
|
|
return e.Pass(fmt.Sprintf(`%s contains "%v"`, message, substr))
|
|
}
|
|
}
|
|
|
|
func (e *Expector) NotContains(s, substr, message string) *Expector {
|
|
check := NewCheck(message, !strings.Contains(s, substr))
|
|
e.Checks = append(e.Checks, check)
|
|
return e
|
|
}
|
|
|
|
func (e *Expector) NotEmptyString(s string, message string) *Expector {
|
|
check := NewCheck(message, s != "")
|
|
e.Checks = append(e.Checks, check)
|
|
return e
|
|
}
|
|
|
|
func (e *Expector) EmptySlice(s []any, message string) *Expector {
|
|
check := NewCheck(message, len(s) == 0)
|
|
e.Checks = append(e.Checks, check)
|
|
return e
|
|
}
|
|
|
|
func (e *Expector) NotEmptySlice(s []any, message string) *Expector {
|
|
check := NewCheck(message, len(s) > 0)
|
|
e.Checks = append(e.Checks, check)
|
|
return e
|
|
}
|