From d4b5a51da059e65507cbddeacfa7e17d658d30e0 Mon Sep 17 00:00:00 2001
From: Peter Jausovec <peterj@microsoft.com>
Date: Thu, 8 Dec 2016 16:10:25 -0800
Subject: [PATCH] Update Mesos unit tests and add new unit tests for
 MarathonEvent

---
 .../acs-dcos/test_marathon_event.py           | 97 +++++++++++++++++++
 src/tasks/dockerDeploy/acs-dcos/test_mesos.py |  2 +-
 2 files changed, 98 insertions(+), 1 deletion(-)
 create mode 100644 src/tasks/dockerDeploy/acs-dcos/test_marathon_event.py

diff --git a/src/tasks/dockerDeploy/acs-dcos/test_marathon_event.py b/src/tasks/dockerDeploy/acs-dcos/test_marathon_event.py
new file mode 100644
index 0000000..604b804
--- /dev/null
+++ b/src/tasks/dockerDeploy/acs-dcos/test_marathon_event.py
@@ -0,0 +1,97 @@
+import unittest
+
+from marathon_deployments import MarathonEvent
+
+
+class MarathonEventTest(unittest.TestCase):
+    def test_not_none(self):
+        m = MarathonEvent({})
+        self.assertIsNotNone(m)
+
+    def test_get_event_type(self):
+        m = MarathonEvent({'eventType': 'SomeEvent'})
+        self.assertEqual(m._get_event_type(), 'SomeEvent') 
+
+    def test_get_event_type_unknown(self):
+        m = MarathonEvent({'blah': 'SomeEvent'})
+        self.assertEqual(m._get_event_type(), 'UNKNOWN') 
+
+    def test_get_app_id(self):
+        m = MarathonEvent({'appId': 'appid'})
+        self.assertEqual(m.app_id(), 'appid') 
+
+    def test_get_app_id_missing(self):
+        m = MarathonEvent({'blah': 'appid'})
+        self.assertRaises(KeyError, m.app_id)
+
+    def test_get_task_id(self):
+        m = MarathonEvent({'taskId': 'taskid'})
+        self.assertEqual(m.task_id(), 'taskid') 
+
+    def test_get_task_id_missing(self):
+        m = MarathonEvent({'blah': 'blah'})
+        self.assertRaises(KeyError, m.task_id)
+
+    def test_get_slave_id(self):
+        m = MarathonEvent({'slaveId': 'slaveid'})
+        self.assertEqual(m.slave_id(), 'slaveid') 
+
+    def test_get_slave_id_missing(self):
+        m = MarathonEvent({'blah': 'blah'})
+        self.assertRaises(KeyError, m.slave_id)
+
+    def test_get_task_status(self):
+        m = MarathonEvent({'taskStatus': 'status'})
+        self.assertEqual(m._get_task_status(), 'status') 
+
+    def test_get_task_status_missing(self):
+        m = MarathonEvent({'blah': 'blah'})
+        self.assertRaises(KeyError, m._get_task_status)
+
+    def test_is_status_update_true(self):
+        m = MarathonEvent({'eventType': 'status_update_event'})
+        self.assertTrue(m.is_status_update())
+
+    def test_is_status_update_false(self):
+        m = MarathonEvent({'eventType': 'BLAH'})
+        self.assertFalse(m.is_status_update())
+
+    def test_is_deployment_succeeded_true(self):
+        m = MarathonEvent({'eventType': 'deployment_success'})
+        self.assertTrue(m.is_deployment_succeeded())
+
+    def test_is_deployment_succeeded_false(self):
+        m = MarathonEvent({'eventType': 'BLAH'})
+        self.assertFalse(m.is_deployment_succeeded())
+
+    def test_is_task_failed_true(self):
+        m = MarathonEvent({'taskStatus': 'TASK_FAILED'})
+        self.assertTrue(m.is_task_failed())
+
+    def test_is_task_failed_false(self):
+        m = MarathonEvent({'taskStatus': 'BLAH'})
+        self.assertFalse(m.is_task_failed())
+
+    def test_is_task_staging_true(self):
+        m = MarathonEvent({'taskStatus': 'TASK_STAGING'})
+        self.assertTrue(m.is_task_staging())
+
+    def test_is_task_staging_false(self):
+        m = MarathonEvent({'taskStatus': 'BLAH'})
+        self.assertFalse(m.is_task_staging())
+
+    def test_is_task_killed_true(self):
+        m = MarathonEvent({'taskStatus': 'TASK_KILLED'})
+        self.assertTrue(m.is_task_killed())
+
+    def test_is_task_killed_false(self):
+        m = MarathonEvent({'taskStatus': 'BLAH'})
+        self.assertFalse(m.is_task_killed())
+
+    def test_is_task_running_true(self):
+        m = MarathonEvent({'taskStatus': 'TASK_RUNNING'})
+        self.assertTrue(m.is_task_running())
+
+    def test_is_task_running_false(self):
+        m = MarathonEvent({'taskStatus': 'BLAH'})
+        self.assertFalse(m.is_task_running())
diff --git a/src/tasks/dockerDeploy/acs-dcos/test_mesos.py b/src/tasks/dockerDeploy/acs-dcos/test_mesos.py
index 17bdd13..7e7787a 100644
--- a/src/tasks/dockerDeploy/acs-dcos/test_mesos.py
+++ b/src/tasks/dockerDeploy/acs-dcos/test_mesos.py
@@ -137,7 +137,7 @@ class MesosTest(unittest.TestCase):
     def test_get_latest_task(self, mock_acs_client):
         mock_acs_client.get_request.side_effect = mocked_requests_get
         m = Mesos(mock_acs_client)
-        actual = m.get_latest_task('service_id')
+        actual = m.get_task('service_id')
         self.assertEqual(actual.task_id, 'service_id')
         self.assertEqual(actual.slave_id, 'slave_id')
         self.assertEqual(actual.framework_id, 'framework_id_2')