residency field support is added

Author: cihangirbesiktas

application.go

@@ -69,6 +69,7 @@ type Application struct {
 	Executor                   *string             `json:"executor,omitempty"`
 	HealthChecks               *[]HealthCheck      `json:"healthChecks,omitempty"`
 	ReadinessChecks            *[]ReadinessCheck   `json:"readinessChecks,omitempty"`
+	Residency                  *Residency          `json:"residency,omitempty"`
 	Instances                  *int                `json:"instances,omitempty"`
 	Mem                        *float64            `json:"mem,omitempty"`
 	Tasks                      []*Task             `json:"tasks,omitempty"`
@@ -462,6 +463,20 @@ func (r *Application) EmptyReadinessChecks() *Application {
 	return r
 }
 
+// SetResidency sets the upgrade strategy.
+func (r *Application) SetResidency(us Residency) *Application {
+	r.Residency = &us
+	return r
+}
+
+// EmptyResidency explicitly empties the residency -- use this if
+// you need to empty the residency of an application that already has
+// the residency set (setting it to nil will keep the current value).
+func (r *Application) EmptyResidency() *Application {
+	r.Residency = &Residency{}
+	return r
+}
+
 // DeploymentIDs retrieves the application deployments IDs
 func (r *Application) DeploymentIDs() []*DeploymentID {
 	var deployments []*DeploymentID

residency.go

@@ -0,0 +1,35 @@
+/*
+Copyright 2014 Rohith All rights reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package marathon
+
+// Residency is the definition for an application residency
+type Residency struct {
+	RelaunchEscalationTimeoutSeconds *int `json:"relaunchEscalationTimeoutSeconds,omitempty"`
+	TaskLostBehavior                 *string `json:"taskLostBehavior,omitempty"`
+}
+
+// SetRelaunchEscalationTimeoutSeconds sets the given relaunch escalation timeout on the residency.
+func (r Residency) SetRelaunchEscalationTimeoutSeconds(i int) Residency {
+	r.RelaunchEscalationTimeoutSeconds = &i
+	return r
+}
+
+// SetRelaunchEscalationTimeoutSeconds sets the given relaunch escalation timeout on the residency.
+func (r Residency) SetTaskLostBehavior(s string) Residency {
+	r.TaskLostBehavior = &s
+	return r
+}

residency_test.go

@@ -0,0 +1,16 @@
+package marathon
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestResidencyCheck(t *testing.T) {
+	rc := Residency{}
+	rc.SetRelaunchEscalationTimeoutSeconds(3600).
+	   SetTaskLostBehavior("WAIT_FOREVER")
+
+	assert.Equal(t, rc.RelaunchEscalationTimeoutSeconds, 3600)
+	assert.Equal(t, rc.TaskLostBehavior, "WAIT_FOREVER")
+}