func TestRemotes(t *testing.T) { // Fetch list of remotes remotesUrl := fmt.Sprintf("%s/remotes", server.URL) r := model.Remote{Driver: "ssh", Location: "foo"} rJson := `{"driver": "ssh", "location": "foo"}` request, err := http.NewRequest("POST", remotesUrl, strings.NewReader(rJson)) res, err := http.DefaultClient.Do(request) if err != nil { t.Error(err) } assert.Equal(t, 201, res.StatusCode) rslice, err := store.Remotes() assert.Equal(t, 1, len(rslice)) assert.Equal(t, "ssh", rslice[0].Driver) assert.Equal(t, "foo", rslice[0].Location) r.ID = rslice[0].ID request, err = http.NewRequest("GET", remotesUrl, nil) res, err = http.DefaultClient.Do(request) if err != nil { t.Error(err) } assert.Equal(t, 200, res.StatusCode) defer res.Body.Close() var target model.Remotes json.NewDecoder(res.Body).Decode(&target) assert.Equal(t, 1, len(target)) assert.Equal(t, r, target[0]) rJson = `{"driver": "ssh", "location": "bar"}` request, err = http.NewRequest("PUT", remotesUrl+"/"+strconv.Itoa(int(r.ID)), strings.NewReader(rJson)) res, err = http.DefaultClient.Do(request) if err != nil { t.Error(err) } assert.Equal(t, 200, res.StatusCode) rslice, err = store.Remotes() assert.Equal(t, 1, len(rslice)) assert.Equal(t, "ssh", rslice[0].Driver) assert.Equal(t, "bar", rslice[0].Location) request, err = http.NewRequest("DELETE", remotesUrl+"/"+strconv.Itoa(int(r.ID)), nil) res, err = http.DefaultClient.Do(request) if err != nil { t.Error(err) } assert.Equal(t, 200, res.StatusCode) rslice, err = store.Remotes() assert.Equal(t, 0, len(rslice)) }
func (s *mockStore) SaveRemote(remote *model.Remote) (created bool, err error) { if remote.ID == 0 { s.idCounter++ remote.ID = s.idCounter created = true } s.remotes[remote.ID] = *remote return created, nil }
func (s *mockStore) DeleteRemote(remote *model.Remote) (deleted bool, err error) { jobs, err := s.JobsForRemote(remote) if err != nil { return false, err } for _, j := range jobs { s.DeleteJob(&j) } if _, ok := s.remotes[remote.ID]; ok { delete(s.remotes, remote.ID) deleted = true } remote.ID = 0 return deleted, nil }