Merge 42947c0 into 6705a3a

allegro · Nov 30, 2017 · a47f200 · a47f200
2 parents 6705a3a + 42947c0
commit a47f200
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/mesosutils/taskinfo_test.go b/mesosutils/taskinfo_test.go
@@ -263,3 +263,23 @@ func TestGetWeightReturnsWeightFromTagLabel(t *testing.T) {
 	require.Equal(t, 50, weight)
 	require.NoError(t, err)
 }
+
+func TestFindEnvValueReturnsCorrectValue(t *testing.T) {
+	expectedValue := "test env value"
+	key := "TEST_ENV_KEY"
+	vars := []mesos.Environment_Variable{
+		{Name: key, Value: expectedValue},
+	}
+	mesosTaskInfo := mesos.TaskInfo{
+		Command: &mesos.CommandInfo{
+			Environment: &mesos.Environment{Variables: vars},
+		},
+	}
+	taskInfo := TaskInfo{
+		TaskInfo: mesosTaskInfo,
+	}
+
+	returnedValue := taskInfo.FindEnvValue(key)
+
+	require.Equal(t, returnedValue, expectedValue)
+}