Exemplo n.º 1
0
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
}
Exemplo n.º 2
0
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))
}
Exemplo n.º 3
0
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
}