func (dt *DaoTest) TestDao_UpdateService(t *C) {
dt.Dao.RemoveService("default", &unused)
svc, _ := service.NewService()
svc.ID = "default0"
svc.Name = "default0"
svc.PoolID = "default"
svc.Launch = "auto"
svc.DeploymentID = "deployment_id"
err := dt.Dao.AddService(*svc, &id)
t.Assert(err, IsNil)
svc.Name = "name"
err = dt.Dao.UpdateService(*svc, &unused)
if err != nil {
t.Errorf("Failure updating service %-v with error: %s", svc, err)
t.Fail()
}
result := service.Service{}
dt.Dao.GetService("default0", &result)
//XXX the time.Time types fail comparison despite being equal...
// as far as I can tell this is a limitation with Go
result.UpdatedAt = svc.UpdatedAt
result.CreatedAt = svc.CreatedAt
if !svc.Equals(&result) {
t.Errorf("Expected Service %+v, Actual Service %+v", result, *svc)
t.Fail()
}
svc, _ = service.NewService()
svc.ID = "default1"
svc.Name = "default1"
svc.PoolID = "default"
svc.Launch = "auto"
svc.DeploymentID = "deployment_id"
err = dt.Dao.AddService(*svc, &id)
t.Assert(err, IsNil)
svc.Name = "name"
err = dt.Dao.UpdateService(*svc, &unused)
if err == nil {
t.Errorf("Expected error updating service with same name and parent", svc)
t.Fail()
}
}
func (dt *DaoTest) TestDao_GetService(t *C) {
svc, _ := service.NewService()
svc.Name = "testname"
svc.PoolID = "default"
svc.Launch = "auto"
svc.DeploymentID = "deployment_id"
err := dt.Dao.AddService(*svc, &id)
t.Assert(err, IsNil)
var result service.Service
err = dt.Dao.GetService(svc.ID, &result)
t.Assert(err, IsNil)
//XXX the time.Time types fail comparison despite being equal...
// as far as I can tell this is a limitation with Go
result.UpdatedAt = svc.UpdatedAt
result.CreatedAt = svc.CreatedAt
if !svc.Equals(&result) {
t.Errorf("GetService Failed: expected=%+v, actual=%+v", svc, result)
}
}
func restAddService(w *rest.ResponseWriter, r *rest.Request, client *node.ControlClient) {
var svc service.Service
var serviceID string
err := r.DecodeJsonPayload(&svc)
if err != nil {
glog.V(1).Info("Could not decode service payload: ", err)
restBadRequest(w, err)
return
}
if id, err := utils.NewUUID36(); err != nil {
restBadRequest(w, err)
return
} else {
svc.ID = id
}
now := time.Now()
svc.CreatedAt = now
svc.UpdatedAt = now
//for each endpoint, evaluate its EndpointTemplates
getSvc := func(svcID string) (service.Service, error) {
svc := service.Service{}
err := client.GetService(svcID, &svc)
return svc, err
}
findChild := func(svcID, childName string) (service.Service, error) {
svc := service.Service{}
err := client.FindChildService(dao.FindChildRequest{svcID, childName}, &svc)
return svc, err
}
if err = svc.EvaluateEndpointTemplates(getSvc, findChild); err != nil {
glog.Errorf("Unable to evaluate service endpoints: %v", err)
restServerError(w, err)
return
}
tags := map[string][]string{
"controlplane_service_id": []string{svc.ID},
}
profile, err := svc.MonitoringProfile.ReBuild("1h-ago", tags)
if err != nil {
glog.Errorf("Unable to rebuild service monitoring profile: %v", err)
restServerError(w, err)
return
}
svc.MonitoringProfile = *profile
//add the service to the data store
err = client.AddService(svc, &serviceID)
if err != nil {
glog.Errorf("Unable to add service: %v", err)
restServerError(w, err)
return
}
//automatically assign virtual ips to new service
request := dao.AssignmentRequest{ServiceID: svc.ID, IPAddress: "", AutoAssignment: true}
if err := client.AssignIPs(request, nil); err != nil {
glog.Errorf("Failed to automatically assign IPs: %+v -> %v", request, err)
restServerError(w, err)
return
}
glog.V(0).Info("Added service ", serviceID)
w.WriteJson(&simpleResponse{"Added service", serviceLinks(serviceID)})
}