34 lines
640 B
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
|
|
}
|