List strains.
This commit is contained in:
parent
6bba9058aa
commit
b8bda910cd
9 changed files with 151 additions and 0 deletions
|
@ -32,6 +32,9 @@ type StrainsService interface {
|
|||
// Get a strain
|
||||
Get(id int64) (*Strain, error)
|
||||
|
||||
// List all strains
|
||||
List(opt *StrainListOptions) ([]*Strain, error)
|
||||
|
||||
// Create a strain record
|
||||
Create(strain *Strain) (bool, error)
|
||||
}
|
||||
|
@ -85,8 +88,33 @@ func (s *strainsService) Create(strain *Strain) (bool, error) {
|
|||
return resp.StatusCode == http.StatusCreated, nil
|
||||
}
|
||||
|
||||
type StrainListOptions struct {
|
||||
ListOptions
|
||||
}
|
||||
|
||||
func (s *strainsService) List(opt *StrainListOptions) ([]*Strain, error) {
|
||||
url, err := s.client.url(router.Strains, nil, opt)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
req, err := s.client.NewRequest("GET", url.String(), nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var strains []*Strain
|
||||
_, err = s.client.Do(req, &strains)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return strains, nil
|
||||
}
|
||||
|
||||
type MockStrainsService struct {
|
||||
Get_ func(id int64) (*Strain, error)
|
||||
List_ func(opt *StrainListOptions) ([]*Strain, error)
|
||||
Create_ func(strain *Strain) (bool, error)
|
||||
}
|
||||
|
||||
|
@ -105,3 +133,10 @@ func (s *MockStrainsService) Create(strain *Strain) (bool, error) {
|
|||
}
|
||||
return s.Create_(strain)
|
||||
}
|
||||
|
||||
func (s *MockStrainsService) List(opt *StrainListOptions) ([]*Strain, error) {
|
||||
if s.List_ == nil {
|
||||
return nil, nil
|
||||
}
|
||||
return s.List_(opt)
|
||||
}
|
||||
|
|
|
@ -80,3 +80,36 @@ func TestStrainService_Create(t *testing.T) {
|
|||
t.Errorf("Strains.Create returned %+v, want %+v", strain, want)
|
||||
}
|
||||
}
|
||||
|
||||
func TestStrainService_List(t *testing.T) {
|
||||
setup()
|
||||
defer teardown()
|
||||
|
||||
want := []*Strain{newStrain()}
|
||||
|
||||
var called bool
|
||||
mux.HandleFunc(urlPath(t, router.Strains, nil), func(w http.ResponseWriter, r *http.Request) {
|
||||
called = true
|
||||
testMethod(t, r, "GET")
|
||||
testFormValues(t, r, values{})
|
||||
|
||||
writeJSON(w, want)
|
||||
})
|
||||
|
||||
strains, err := client.Strains.List(nil)
|
||||
if err != nil {
|
||||
t.Errorf("Strains.List returned error: %v", err)
|
||||
}
|
||||
|
||||
if !called {
|
||||
t.Fatal("!called")
|
||||
}
|
||||
|
||||
for _, u := range want {
|
||||
normalizeTime(&u.CreatedAt, &u.UpdatedAt, &u.DeletedAt)
|
||||
}
|
||||
|
||||
if !reflect.DeepEqual(strains, want) {
|
||||
t.Errorf("Strains.List return %+v, want %+v", strains, want)
|
||||
}
|
||||
}
|
||||
|
|
Reference in a new issue