provision/swarm: ignores task status for register unit

tsuru · Jun 13, 2017 · 3c38fb4 · 3c38fb4
1 parent 624b214
commit 3c38fb4
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/provision/swarm/provisioner.go b/provision/swarm/provisioner.go
@@ -369,11 +369,16 @@ func (p *swarmProvisioner) RegisterUnit(a provision.App, unitId string, customDa
 	if err != nil {
 		return err
 	}
-	units, err := tasksToUnits(client, []swarm.Task{*task})
+	service, err := client.InspectService(task.ServiceID)
 	if err != nil {
-		return err
+		return errors.WithStack(err)
+	}
+	node, err := client.InspectNode(task.NodeID)
+	if err != nil {
+		return errors.WithStack(err)
 	}
-	err = bindUnit(a, &units[0])
+	unit := taskToUnit(task, service, node, a)
+	err = bindUnit(a, &unit)
 	if err != nil {
 		return err
 	}