func (s *ConsumptionSuite) TestGetServiceOrErrorShoudNotReturnErrorWhenServiceIsNotRestricted(c *gocheck.C) {
srv := service.Service{Name: "foo"}
err := srv.Create()
c.Assert(err, gocheck.IsNil)
_, err = getServiceOrError("foo", s.user)
c.Assert(err, gocheck.IsNil)
}
func (s *S) TestRemoveUnits(c *gocheck.C) {
var calls int32
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
v := atomic.LoadInt32(&calls)
atomic.StoreInt32(&calls, v+1)
w.WriteHeader(http.StatusNoContent)
}))
srvc := service.Service{Name: "mysql", Endpoint: map[string]string{"production": ts.URL}}
err := srvc.Create()
c.Assert(err, gocheck.IsNil)
defer s.conn.Services().Remove(bson.M{"_id": "mysql"})
app := App{
Name: "chemistry",
Framework: "python",
}
instance := service.ServiceInstance{
Name: "my-inst",
ServiceName: "mysql",
Teams: []string{s.team.Name},
Apps: []string{app.Name},
}
instance.Create()
defer s.conn.ServiceInstances().Remove(bson.M{"name": "my-inst"})
err = s.conn.Apps().Insert(app)
c.Assert(err, gocheck.IsNil)
defer s.conn.Apps().Remove(bson.M{"name": app.Name})
c.Assert(err, gocheck.IsNil)
s.provisioner.Provision(&app)
defer s.provisioner.Destroy(&app)
app.AddUnits(4)
otherApp := App{Name: app.Name, Units: app.Units}
err = otherApp.RemoveUnits(2)
c.Assert(err, gocheck.IsNil)
ts.Close()
units := s.provisioner.GetUnits(&app)
c.Assert(units, gocheck.HasLen, 3) // when you provision you already have one, so it's 4+1-2 (in provisioner, in app struct we have 2)
c.Assert(units[0].Name, gocheck.Equals, "chemistry/0")
c.Assert(units[1].Name, gocheck.Equals, "chemistry/3")
c.Assert(units[2].Name, gocheck.Equals, "chemistry/4")
err = app.Get()
c.Assert(err, gocheck.IsNil)
c.Assert(app.Framework, gocheck.Equals, "python")
c.Assert(app.Units, gocheck.HasLen, 2)
c.Assert(app.Units[0].Name, gocheck.Equals, "chemistry/3")
c.Assert(app.Units[1].Name, gocheck.Equals, "chemistry/4")
ok := make(chan int8)
go func() {
for _ = range time.Tick(1e3) {
if atomic.LoadInt32(&calls) == 2 {
ok <- 1
return
}
}
}()
select {
case <-ok:
case <-time.After(2e9):
c.Fatal("Did not call service endpoint twice.")
}
}
func (s *S) TestUnbindUnit(c *gocheck.C) {
called := false
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
called = true
w.WriteHeader(http.StatusNoContent)
}))
defer ts.Close()
srvc := service.Service{Name: "mysql", Endpoint: map[string]string{"production": ts.URL}}
err := srvc.Create()
c.Assert(err, gocheck.IsNil)
defer s.conn.Services().Remove(bson.M{"_id": "mysql"})
instance := service.ServiceInstance{
Name: "my-mysql",
ServiceName: "mysql",
Teams: []string{s.team.Name},
Apps: []string{"painkiller"},
}
instance.Create()
defer s.conn.ServiceInstances().Remove(bson.M{"_id": "my-mysql"})
a, err := createTestApp(s.conn, "painkiller", "", []string{s.team.Name}, []app.Unit{{Ip: "10.10.10.10"}})
c.Assert(err, gocheck.IsNil)
defer s.conn.Apps().Remove(bson.M{"name": a.Name})
err = instance.UnbindUnit(a.GetUnits()[0])
c.Assert(err, gocheck.IsNil)
c.Assert(called, gocheck.Equals, true)
}
func (s *ConsumptionSuite) TestCreateInstanceHandlerReturnsErrorWhenUserCannotUseService(c *gocheck.C) {
service := service.Service{Name: "mysql", IsRestricted: true}
service.Create()
recorder, request := makeRequestToCreateInstanceHandler(c)
err := CreateInstanceHandler(recorder, request, s.user)
c.Assert(err, gocheck.ErrorMatches, "^This user does not have access to this service$")
}
func (s *S) TestHandleBindServiceMessage(c *gocheck.C) {
called := false
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
called = true
w.Write([]byte(`{"DATABASE_USER":"******","DATABASE_PASSWORD":"******"}`))
}))
defer ts.Close()
srvc := service.Service{Name: "mysql", Endpoint: map[string]string{"production": ts.URL}}
err := srvc.Create()
c.Assert(err, gocheck.IsNil)
defer s.conn.Services().Remove(bson.M{"_id": "mysql"})
instance := service.ServiceInstance{Name: "my-mysql", ServiceName: "mysql", Teams: []string{s.team.Name}}
instance.Create()
defer s.conn.ServiceInstances().Remove(bson.M{"_id": "my-mysql"})
a := App{
Name: "nemesis",
Units: []Unit{
{
Name: "i-00800",
State: "started",
Machine: 19,
},
},
}
err = s.conn.Apps().Insert(a)
c.Assert(err, gocheck.IsNil)
defer s.conn.Apps().Remove(bson.M{"name": a.Name})
err = instance.AddApp(a.Name)
c.Assert(err, gocheck.IsNil)
err = s.conn.ServiceInstances().Update(bson.M{"name": instance.Name}, instance)
c.Assert(err, gocheck.IsNil)
message := queue.Message{Action: BindService, Args: []string{a.Name, a.Units[0].Name}}
handle(&message)
c.Assert(called, gocheck.Equals, true)
}
func (s *ConsumptionSuite) TestCreateInstanceHandlerSavesServiceInstanceInDb(c *gocheck.C) {
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.Write([]byte(`{"DATABASE_HOST":"localhost"}`))
}))
defer ts.Close()
se := service.Service{
Name: "mysql",
Teams: []string{s.team.Name},
Endpoint: map[string]string{"production": ts.URL},
}
se.Create()
defer s.conn.Services().Remove(bson.M{"_id": se.Name})
recorder, request := makeRequestToCreateInstanceHandler(c)
err := createServiceInstance(recorder, request, s.token)
c.Assert(err, gocheck.IsNil)
var si service.ServiceInstance
err = s.conn.ServiceInstances().Find(bson.M{"name": "brainSQL", "service_name": "mysql"}).One(&si)
c.Assert(err, gocheck.IsNil)
s.conn.ServiceInstances().Update(bson.M{"name": si.Name}, si)
c.Assert(si.Name, gocheck.Equals, "brainSQL")
c.Assert(si.ServiceName, gocheck.Equals, "mysql")
action := testing.Action{
Action: "create-service-instance",
User: s.user.Email,
Extra: []interface{}{`{"name":"brainSQL","service_name":"mysql"}`},
}
c.Assert(action, testing.IsRecorded)
}
func (s *ProvisionSuite) TestUpdateHandlerShouldUpdateTheServiceWithDataFromManifest(c *gocheck.C) {
service := service.Service{
Name: "mysqlapi",
Endpoint: map[string]string{"production": "sqlapi.com"},
OwnerTeams: []string{s.team.Name},
}
err := service.Create()
c.Assert(err, gocheck.IsNil)
defer s.conn.Services().Remove(bson.M{"_id": service.Name})
p, err := filepath.Abs("testdata/manifest.yml")
manifest, err := ioutil.ReadFile(p)
c.Assert(err, gocheck.IsNil)
request, err := http.NewRequest("PUT", "/services", bytes.NewBuffer(manifest))
c.Assert(err, gocheck.IsNil)
recorder := httptest.NewRecorder()
err = serviceUpdate(recorder, request, s.token)
c.Assert(err, gocheck.IsNil)
c.Assert(recorder.Code, gocheck.Equals, http.StatusNoContent)
err = s.conn.Services().Find(bson.M{"_id": service.Name}).One(&service)
c.Assert(err, gocheck.IsNil)
c.Assert(service.Endpoint["production"], gocheck.Equals, "mysqlapi.com")
endpoints := map[string]string{"production": "mysqlapi.com", "test": "localhost:8000"}
action := testing.Action{
Action: "update-service",
User: s.user.Email,
Extra: []interface{}{service.Name, endpoints},
}
c.Assert(action, testing.IsRecorded)
}
func (s *ConsumptionSuite) TestServiceInstanceStatusHandler(c *gocheck.C) {
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusNoContent)
w.Write([]byte(`Service instance "my_nosql" is up`))
}))
defer ts.Close()
srv := service.Service{Name: "mongodb", OwnerTeams: []string{s.team.Name}, Endpoint: map[string]string{"production": ts.URL}}
err := srv.Create()
c.Assert(err, gocheck.IsNil)
defer srv.Delete()
si := service.ServiceInstance{Name: "my_nosql", ServiceName: srv.Name, Teams: []string{s.team.Name}}
err = si.Create()
c.Assert(err, gocheck.IsNil)
defer service.DeleteInstance(&si)
recorder, request := makeRequestToStatusHandler("my_nosql", c)
err = serviceInstanceStatus(recorder, request, s.token)
c.Assert(err, gocheck.IsNil)
b, err := ioutil.ReadAll(recorder.Body)
c.Assert(string(b), gocheck.Equals, "Service instance \"my_nosql\" is up")
action := testing.Action{
Action: "service-instance-status",
User: s.user.Email,
Extra: []interface{}{"my_nosql"},
}
c.Assert(action, testing.IsRecorded)
}
func (s *ConsumptionSuite) TestServicesInstancesHandler(c *gocheck.C) {
srv := service.Service{Name: "redis", Teams: []string{s.team.Name}}
err := srv.Create()
c.Assert(err, gocheck.IsNil)
instance := service.ServiceInstance{
Name: "redis-globo",
ServiceName: "redis",
Apps: []string{"globo"},
Teams: []string{s.team.Name},
}
err = instance.Create()
c.Assert(err, gocheck.IsNil)
request, err := http.NewRequest("GET", "/services/instances", nil)
c.Assert(err, gocheck.IsNil)
recorder := httptest.NewRecorder()
err = ServicesInstancesHandler(recorder, request, s.user)
c.Assert(err, gocheck.IsNil)
body, err := ioutil.ReadAll(recorder.Body)
c.Assert(err, gocheck.IsNil)
var instances []service.ServiceModel
err = json.Unmarshal(body, &instances)
c.Assert(err, gocheck.IsNil)
expected := []service.ServiceModel{
{Service: "redis", Instances: []string{"redis-globo"}},
}
c.Assert(instances, gocheck.DeepEquals, expected)
}
func (s *ConsumptionSuite) TestRemoveServiceInstanceHandler(c *gocheck.C) {
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusOK)
}))
defer ts.Close()
se := service.Service{Name: "foo", Endpoint: map[string]string{"production": ts.URL}}
err := se.Create()
defer s.conn.Services().Remove(bson.M{"_id": se.Name})
c.Assert(err, gocheck.IsNil)
si := service.ServiceInstance{Name: "foo-instance", ServiceName: "foo", Teams: []string{s.team.Name}}
err = si.Create()
c.Assert(err, gocheck.IsNil)
recorder, request := makeRequestToRemoveInstanceHandler("foo-instance", c)
err = removeServiceInstance(recorder, request, s.token)
c.Assert(err, gocheck.IsNil)
b, err := ioutil.ReadAll(recorder.Body)
c.Assert(err, gocheck.IsNil)
c.Assert(string(b), gocheck.Equals, "service instance successfuly removed")
n, err := s.conn.ServiceInstances().Find(bson.M{"name": "foo-instance"}).Count()
c.Assert(err, gocheck.IsNil)
c.Assert(n, gocheck.Equals, 0)
action := testing.Action{
Action: "remove-service-instance",
User: s.user.Email,
Extra: []interface{}{"foo-instance"},
}
c.Assert(action, testing.IsRecorded)
}
func (s *S) TestUnbindRemovesAppFromServiceInstance(c *gocheck.C) {
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusNoContent)
}))
defer ts.Close()
srvc := service.Service{Name: "mysql", Endpoint: map[string]string{"production": ts.URL}}
err := srvc.Create()
c.Assert(err, gocheck.IsNil)
defer s.conn.Services().Remove(bson.M{"_id": "mysql"})
instance := service.ServiceInstance{
Name: "my-mysql",
ServiceName: "mysql",
Teams: []string{s.team.Name},
Apps: []string{"painkiller"},
}
instance.Create()
defer s.conn.ServiceInstances().Remove(bson.M{"_id": "my-mysql"})
a, err := createTestApp(s.conn, "painkiller", "", []string{s.team.Name}, []app.Unit{{Ip: "10.10.10.10"}})
c.Assert(err, gocheck.IsNil)
defer s.conn.Apps().Remove(bson.M{"name": a.Name})
err = instance.UnbindApp(&a)
c.Assert(err, gocheck.IsNil)
s.conn.ServiceInstances().Find(bson.M{"_id": instance.Name}).One(&instance)
c.Assert(instance.Apps, gocheck.DeepEquals, []string{})
}
func (s *ConsumptionSuite) TestGetServiceOrErrorShouldReturnErrorWhenUserHaveNoAccessToService(c *gocheck.C) {
srv := service.Service{Name: "foo", IsRestricted: true}
err := srv.Create()
c.Assert(err, gocheck.IsNil)
_, err = getServiceOrError("foo", s.user)
c.Assert(err, gocheck.ErrorMatches, "^This user does not have access to this service$")
}
func (s *S) TestBindUnit(c *gocheck.C) {
called := false
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
called = true
w.Write([]byte(`{"DATABASE_USER":"******","DATABASE_PASSWORD":"******"}`))
}))
defer ts.Close()
srvc := service.Service{Name: "mysql", Endpoint: map[string]string{"production": ts.URL}}
err := srvc.Create()
c.Assert(err, gocheck.IsNil)
defer s.conn.Services().Remove(bson.M{"_id": "mysql"})
instance := service.ServiceInstance{Name: "my-mysql", ServiceName: "mysql", Teams: []string{s.team.Name}}
instance.Create()
defer s.conn.ServiceInstances().Remove(bson.M{"_id": "my-mysql"})
a, err := createTestApp(s.conn, "painkiller", "", []string{s.team.Name}, []app.Unit{{Ip: "10.10.10.10"}})
c.Assert(err, gocheck.IsNil)
defer s.conn.Apps().Remove(bson.M{"name": a.Name})
envs, err := instance.BindUnit(&a, a.GetUnits()[0])
c.Assert(err, gocheck.IsNil)
c.Assert(called, gocheck.Equals, true)
expectedEnvs := map[string]string{
"DATABASE_USER": "******",
"DATABASE_PASSWORD": "******",
}
c.Assert(envs, gocheck.DeepEquals, expectedEnvs)
}
func (s *ConsumptionSuite) TestServicePlansHandler(c *gocheck.C) {
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
content := `[{"name": "ignite", "description": "some value"}, {"name": "small", "description": "not space left for you"}]`
w.Write([]byte(content))
}))
defer ts.Close()
srvc := service.Service{Name: "mysql", Endpoint: map[string]string{"production": ts.URL}}
err := srvc.Create()
c.Assert(err, gocheck.IsNil)
defer srvc.Delete()
request, err := http.NewRequest("GET", "/services/mysql/plans?:name=mysql", nil)
c.Assert(err, gocheck.IsNil)
recorder := httptest.NewRecorder()
err = servicePlans(recorder, request, s.token)
c.Assert(err, gocheck.IsNil)
body, err := ioutil.ReadAll(recorder.Body)
c.Assert(err, gocheck.IsNil)
var plans []service.Plan
err = json.Unmarshal(body, &plans)
c.Assert(err, gocheck.IsNil)
expected := []service.Plan{
{Name: "ignite", Description: "some value"},
{Name: "small", Description: "not space left for you"},
}
c.Assert(plans, gocheck.DeepEquals, expected)
action := testing.Action{
Action: "service-plans",
User: s.user.Email,
Extra: []interface{}{"mysql"},
}
c.Assert(action, testing.IsRecorded)
}
func (s *ConsumptionSuite) TestGetServiceOrError(c *gocheck.C) {
srv := service.Service{Name: "foo", Teams: []string{s.team.Name}, IsRestricted: true}
err := srv.Create()
c.Assert(err, gocheck.IsNil)
rSrv, err := getServiceOrError("foo", s.user)
c.Assert(err, gocheck.IsNil)
c.Assert(rSrv.Name, gocheck.Equals, srv.Name)
}
func (s *ProvisionSuite) TestgetServiceByOwner(c *gocheck.C) {
srv := service.Service{Name: "foo", OwnerTeams: []string{s.team.Name}}
err := srv.Create()
c.Assert(err, gocheck.IsNil)
defer srv.Delete()
rSrv, err := getServiceByOwner("foo", s.user)
c.Assert(err, gocheck.IsNil)
c.Assert(rSrv.Name, gocheck.Equals, srv.Name)
}
func (s *S) TestUnbindRemovesEnvironmentVariableFromApp(c *gocheck.C) {
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusNoContent)
}))
defer ts.Close()
srvc := service.Service{Name: "mysql", Endpoint: map[string]string{"production": ts.URL}}
err := srvc.Create()
c.Assert(err, gocheck.IsNil)
defer s.conn.Services().Remove(bson.M{"_id": "mysql"})
instance := service.ServiceInstance{
Name: "my-mysql",
ServiceName: "mysql",
Teams: []string{s.team.Name},
Apps: []string{"painkiller"},
}
err = instance.Create()
c.Assert(err, gocheck.IsNil)
defer s.conn.ServiceInstances().Remove(bson.M{"_id": "my-mysql"})
a := app.App{
Name: "painkiller",
Teams: []string{s.team.Name},
Env: map[string]bind.EnvVar{
"DATABASE_HOST": {
Name: "DATABASE_HOST",
Value: "arrea",
Public: false,
InstanceName: instance.Name,
},
"MY_VAR": {
Name: "MY_VAR",
Value: "123",
},
},
Units: []app.Unit{
{
Ip: "10.10.10.10",
},
},
}
err = s.conn.Apps().Insert(&a)
c.Assert(err, gocheck.IsNil)
defer s.conn.Apps().Remove(bson.M{"name": a.Name})
err = instance.UnbindApp(&a)
c.Assert(err, gocheck.IsNil)
newApp := app.App{Name: a.Name}
err = newApp.Get()
c.Assert(err, gocheck.IsNil)
expected := map[string]bind.EnvVar{
"MY_VAR": {
Name: "MY_VAR",
Value: "123",
},
}
c.Assert(a.Env, gocheck.DeepEquals, expected)
}
func (s *ProvisionSuite) TestGetDocHandler(c *gocheck.C) {
se := service.Service{Name: "some_service", Doc: "some doc", OwnerTeams: []string{s.team.Name}}
se.Create()
defer s.conn.Services().Remove(bson.M{"_id": se.Name})
request, err := http.NewRequest("GET", fmt.Sprintf("/services/%s/doc?:name=%s", se.Name, se.Name), nil)
c.Assert(err, gocheck.IsNil)
recorder := httptest.NewRecorder()
err = GetDocHandler(recorder, request, s.token)
c.Assert(err, gocheck.IsNil)
c.Assert(recorder.Body.String(), gocheck.Equals, "some doc")
}
func getServiceByOwner(name string, u *auth.User) (service.Service, error) {
s := service.Service{Name: name}
err := s.Get()
if err != nil {
return s, &errors.HTTP{Code: http.StatusNotFound, Message: "Service not found"}
}
if !auth.CheckUserAccess(s.OwnerTeams, u) {
msg := "This user does not have access to this service"
return s, &errors.HTTP{Code: http.StatusForbidden, Message: msg}
}
return s, err
}
func (s *ConsumptionSuite) TestDocHandlerReturns401WhenUserHasNoAccessToService(c *gocheck.C) {
srv := service.Service{
Name: "coolnosql",
Doc: "some doc...",
IsRestricted: true,
}
err := srv.Create()
c.Assert(err, gocheck.IsNil)
recorder, request := s.makeRequestToGetDocHandler("coolnosql", c)
err = Doc(recorder, request, s.user)
c.Assert(err, gocheck.ErrorMatches, "^This user does not have access to this service$")
}
func serviceCreate(w http.ResponseWriter, r *http.Request, t *auth.Token) error {
defer r.Body.Close()
body, err := ioutil.ReadAll(r.Body)
if err != nil {
return err
}
var sy serviceYaml
err = goyaml.Unmarshal(body, &sy)
if err != nil {
return err
}
if _, ok := sy.Endpoint["production"]; !ok {
return &errors.HTTP{Code: http.StatusBadRequest, Message: "You must provide a production endpoint in the manifest file."}
}
u, err := t.User()
if err != nil {
return err
}
rec.Log(u.Email, "create-service", sy.Id, sy.Endpoint)
conn, err := db.Conn()
if err != nil {
return err
}
defer conn.Close()
var teams []auth.Team
err = conn.Teams().Find(bson.M{"users": u.Email}).All(&teams)
if err != nil {
return err
}
if len(teams) == 0 {
msg := "In order to create a service, you should be member of at least one team"
return &errors.HTTP{Code: http.StatusForbidden, Message: msg}
}
n, err := conn.Services().Find(bson.M{"_id": sy.Id}).Count()
if err != nil {
return &errors.HTTP{Code: http.StatusInternalServerError, Message: err.Error()}
}
if n != 0 {
msg := fmt.Sprintf("Service with name %s already exists.", sy.Id)
return &errors.HTTP{Code: http.StatusInternalServerError, Message: msg}
}
s := service.Service{
Name: sy.Id,
Endpoint: sy.Endpoint,
OwnerTeams: auth.GetTeamsNames(teams),
}
err = s.Create()
if err != nil {
return err
}
fmt.Fprint(w, "success")
return nil
}
func (s *ConsumptionSuite) TestRemoveServiceHandlerWithoutPermissionShouldReturn401(c *gocheck.C) {
se := service.Service{Name: "foo"}
err := se.Create()
defer s.conn.Services().Remove(bson.M{"_id": se.Name})
c.Assert(err, gocheck.IsNil)
si := service.ServiceInstance{Name: "foo-instance", ServiceName: "foo"}
err = si.Create()
defer s.conn.ServiceInstances().Remove(bson.M{"name": si.Name})
c.Assert(err, gocheck.IsNil)
recorder, request := makeRequestToRemoveInstanceHandler("foo-instance", c)
err = removeServiceInstance(recorder, request, s.token)
c.Assert(err.Error(), gocheck.Equals, service.ErrAccessNotAllowed.Error())
}
func (s *ConsumptionSuite) TestRemoveServiceHandlerWIthAssociatedAppsShouldFailAndReturnError(c *gocheck.C) {
se := service.Service{Name: "foo"}
err := se.Create()
defer s.conn.Services().Remove(bson.M{"_id": se.Name})
c.Assert(err, gocheck.IsNil)
si := service.ServiceInstance{Name: "foo-instance", ServiceName: "foo", Apps: []string{"foo-bar"}, Teams: []string{s.team.Name}}
err = si.Create()
defer s.conn.ServiceInstances().Remove(bson.M{"name": si.Name})
c.Assert(err, gocheck.IsNil)
recorder, request := makeRequestToRemoveInstanceHandler("foo-instance", c)
err = RemoveServiceInstanceHandler(recorder, request, s.user)
c.Assert(err, gocheck.ErrorMatches, "^This service instance is bound to at least one app. Unbind them before removing it$")
}
func (s *ConsumptionSuite) TestRemoveServiceHandlerWithoutPermissionShouldReturn401(c *gocheck.C) {
se := service.Service{Name: "foo"}
err := se.Create()
defer s.conn.Services().Remove(bson.M{"_id": se.Name})
c.Assert(err, gocheck.IsNil)
si := service.ServiceInstance{Name: "foo-instance", ServiceName: "foo"}
err = si.Create()
defer s.conn.ServiceInstances().Remove(bson.M{"name": si.Name})
c.Assert(err, gocheck.IsNil)
recorder, request := makeRequestToRemoveInstanceHandler("foo-instance", c)
err = RemoveServiceInstanceHandler(recorder, request, s.user)
c.Assert(err, gocheck.ErrorMatches, "^This user does not have access to this service instance$")
}
func (s *ConsumptionSuite) TestCreateInstanceHandlerReturnErrorIfTheServiceAPICallFailAndDoesNotSaveTheInstanceInTheDatabase(c *gocheck.C) {
ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
w.WriteHeader(http.StatusInternalServerError)
}))
defer ts.Close()
srvc := service.Service{Name: "mysql", Endpoint: map[string]string{"production": ts.URL}}
err := srvc.Create()
c.Assert(err, gocheck.IsNil)
defer s.conn.Services().Remove(bson.M{"_id": "mysql"})
recorder, request := makeRequestToCreateInstanceHandler(c)
err = CreateInstanceHandler(recorder, request, s.user)
c.Assert(err, gocheck.NotNil)
}
func (s *ConsumptionSuite) TestServiceInfoHandlerReturns403WhenTheUserDoesNotHaveAccessToTheService(c *gocheck.C) {
se := service.Service{Name: "Mysql", IsRestricted: true}
se.Create()
defer s.conn.Services().Remove(bson.M{"_id": se.Name})
request, err := http.NewRequest("DELETE", fmt.Sprintf("/services/%s?:name=%s", se.Name, se.Name), nil)
c.Assert(err, gocheck.IsNil)
recorder := httptest.NewRecorder()
err = ServiceInfoHandler(recorder, request, s.user)
c.Assert(err, gocheck.NotNil)
e, ok := err.(*errors.Http)
c.Assert(ok, gocheck.Equals, true)
c.Assert(e.Code, gocheck.Equals, http.StatusForbidden)
c.Assert(e, gocheck.ErrorMatches, "^This user does not have access to this service$")
}
func (s *S) TestBindAppFailsWhenEndpointIsDown(c *gocheck.C) {
srvc := service.Service{Name: "mysql", Endpoint: map[string]string{"production": ""}}
err := srvc.Create()
c.Assert(err, gocheck.IsNil)
defer s.conn.Services().Remove(bson.M{"_id": "mysql"})
instance := service.ServiceInstance{Name: "my-mysql", ServiceName: "mysql", Teams: []string{s.team.Name}}
instance.Create()
defer s.conn.ServiceInstances().Remove(bson.M{"name": "my-mysql"})
a, err := createTestApp(s.conn, "painkiller", "", []string{s.team.Name}, []app.Unit{{Ip: "10.10.10.10"}})
c.Assert(err, gocheck.IsNil)
defer s.conn.Apps().Remove(bson.M{"name": a.Name})
err = instance.BindApp(&a)
c.Assert(err, gocheck.NotNil)
}
func (s *ProvisionSuite) TestDeleteHandlerReturns403WhenTheUserIsNotOwnerOfTheTeam(c *gocheck.C) {
se := service.Service{Name: "Mysql", Teams: []string{s.team.Name}}
se.Create()
defer s.conn.Services().Remove(bson.M{"_id": se.Name})
request, err := http.NewRequest("DELETE", fmt.Sprintf("/services/%s?:name=%s", se.Name, se.Name), nil)
c.Assert(err, gocheck.IsNil)
recorder := httptest.NewRecorder()
err = DeleteHandler(recorder, request, s.token)
c.Assert(err, gocheck.NotNil)
e, ok := err.(*errors.Http)
c.Assert(ok, gocheck.Equals, true)
c.Assert(e.Code, gocheck.Equals, http.StatusForbidden)
c.Assert(e, gocheck.ErrorMatches, "^This user does not have access to this service$")
}
func CreateInstanceHandler(w http.ResponseWriter, r *http.Request, u *auth.User) error {
log.Print("Receiving request to create a service instance")
b, err := ioutil.ReadAll(r.Body)
if err != nil {
log.Print("Got error while reading request body:")
log.Print(err.Error())
return &errors.Http{Code: http.StatusInternalServerError, Message: err.Error()}
}
var sJson map[string]string
err = json.Unmarshal(b, &sJson)
if err != nil {
log.Print("Got a problem while unmarshalling request's json:")
log.Print(err.Error())
return &errors.Http{Code: http.StatusInternalServerError, Message: err.Error()}
}
var s service.Service
err = validateInstanceForCreation(&s, sJson, u)
if err != nil {
log.Print("Got error while validation:")
log.Print(err.Error())
return err
}
var teamNames []string
teams, err := u.Teams()
if err != nil {
return err
}
for _, t := range teams {
if s.HasTeam(&t) || !s.IsRestricted {
teamNames = append(teamNames, t.Name)
}
}
si := service.ServiceInstance{
Name: sJson["name"],
ServiceName: sJson["service_name"],
Teams: teamNames,
}
if err = s.ProductionEndpoint().Create(&si); err != nil {
log.Print("Error while calling create action from service api.")
log.Print(err.Error())
return err
}
err = si.Create()
if err != nil {
return err
}
fmt.Fprint(w, "success")
return nil
}
func (s *ProvisionSuite) TestAddDocHandlerReturns403WhenTheUserDoesNotHaveAccessToTheService(c *gocheck.C) {
se := service.Service{Name: "Mysql"}
se.Create()
defer s.conn.Services().Remove(bson.M{"_id": se.Name})
b := bytes.NewBufferString("doc")
request, err := http.NewRequest("PUT", fmt.Sprintf("/services/%s/doc?:name=%s", se.Name, se.Name), b)
c.Assert(err, gocheck.IsNil)
recorder := httptest.NewRecorder()
err = AddDocHandler(recorder, request, s.token)
c.Assert(err, gocheck.NotNil)
e, ok := err.(*errors.Http)
c.Assert(ok, gocheck.Equals, true)
c.Assert(e.Code, gocheck.Equals, http.StatusForbidden)
c.Assert(e, gocheck.ErrorMatches, "^This user does not have access to this service$")
}