models/runs.py: Add flavor to runs output

Signed-off-by: Vallari Agrawal <val.agl002@gmail.com>

Author: VallariAg

paddles/models/runs.py

@@ -168,6 +168,7 @@ def __json__(self):
             machine_type=self.machine_type,
             sha1=results['sha1'],
             priority=self.priority,
+            flavor=results['flavor'],
         )
 
     @classmethod
@@ -227,6 +228,7 @@ def get_results(self):
         unknown = jobs_status.count(None) + jobs_status.count('unknown')
         total = len(jobs_status)
         sha1 = next(self.jobs.values(Job.sha1), ['none'])[0]
+        flavor = next(self.jobs.values(Job.flavor), ['none'])[0]
         return {
             'queued': queued,
             'pass': passing,
@@ -237,6 +239,7 @@ def get_results(self):
             'unknown': unknown,
             'total': total,
             'sha1': sha1,
+            'flavor': flavor,
         }
 
     def set_status(self, results=None):

paddles/tests/models/test_runs.py

@@ -228,7 +228,7 @@ def test_run_results(self):
         run_name = 'teuthology-2014-03-27_00:00:00-x-x-x-x-x'
         new_run = Run(run_name)
         stats_in = {'pass': 9, 'fail': 1, 'dead': 6, 'running': 5, 'sha1': None,
-                    'waiting': 1, 'unknown': 1, 'queued': 1}
+                    'waiting': 1, 'unknown': 1, 'queued': 1, 'flavor': None}
         statuses = stats_in.keys()
         stats_in['total'] = sum(_ for _ in stats_in.values() if _)
         stats_out = {}
@@ -248,3 +248,10 @@ def test_run_priority(self):
         new_run = Run(run_name)
         Job(dict(job_id=1, id=1, status='queued', priority=99), new_run)
         assert new_run.priority == 99
+
+    def test_run_flavor(self):
+        run_name = "run_flavor"
+        new_run = Run(run_name)
+        Job(dict(job_id=1, id=1, status='queued', flavor="blah"), new_run)
+        run_result = new_run.get_results()
+        assert run_result.get('flavor') == "blah"