nurse/protocols/http/status_validation.go

34 lines
640 B
Go

package http
import (
"fmt"
"net/http"
"github.com/baez90/nurse/grammar"
"github.com/baez90/nurse/validation"
)
var _ validation.FromCall[*http.Response] = (*StatusValidator)(nil)
type StatusValidator struct {
Want int
}
func (s *StatusValidator) Validate(resp *http.Response) error {
if resp.StatusCode != s.Want {
return fmt.Errorf("want HTTP status %d but got %d", s.Want, resp.StatusCode)
}
return nil
}
func (s *StatusValidator) UnmarshalCall(c grammar.Call) error {
if err := grammar.ValidateParameterCount(c.Params, 1); err != nil {
return err
}
var err error
s.Want, err = c.Params[0].AsInt()
return err
}