From 2654c795485e7cc439e9af050a9880c96a634efd Mon Sep 17 00:00:00 2001
From: Jacob Beck <jake@fishtownanalytics.com>
Date: Wed, 3 Apr 2019 10:01:36 -0600
Subject: [PATCH] PR feedback

---
 core/dbt/rpc.py                                   |  3 +--
 test/integration/042_sources_test/test_sources.py | 12 ++++--------
 2 files changed, 5 insertions(+), 10 deletions(-)

diff --git a/core/dbt/rpc.py b/core/dbt/rpc.py
index 2e25b9e5655..de1eae7f826 100644
--- a/core/dbt/rpc.py
+++ b/core/dbt/rpc.py
@@ -306,8 +306,7 @@ def process_listing(self, active=True, completed=False):
             ))
         table.sort(key=lambda r: (r.state, r.start))
         result = {
-            'columns': list(TaskRow._fields),
-            'rows': [list(r) for r in table],
+            'rows': [dict(r._asdict()) for r in table],
         }
         return result
 
diff --git a/test/integration/042_sources_test/test_sources.py b/test/integration/042_sources_test/test_sources.py
index 0e4c92a00ed..ab56d077443 100644
--- a/test/integration/042_sources_test/test_sources.py
+++ b/test/integration/042_sources_test/test_sources.py
@@ -696,8 +696,7 @@ def test_ps_kill_postgres(self):
         sleeper_ps_result = self.query('ps', completed=False, active=True).json()
         result = self.assertIsResult(sleeper_ps_result)
         self.assertEqual(len(result['rows']), 1)
-        self.assertEqual(len(result['rows'][0]), len(result['columns']))
-        rowdict = [{k: v for k, v in zip(result['columns'], row)} for row in result['rows']]
+        rowdict = result['rows']
         self.assertEqual(rowdict[0]['request_id'], request_id)
         self.assertEqual(rowdict[0]['method'], 'run')
         self.assertEqual(rowdict[0]['state'], 'running')
@@ -706,8 +705,7 @@ def test_ps_kill_postgres(self):
         complete_ps_result = self.query('ps', completed=True, active=False).json()
         result = self.assertIsResult(complete_ps_result)
         self.assertEqual(len(result['rows']), 1)
-        self.assertEqual(len(result['rows'][0]), len(result['columns']))
-        rowdict = [{k: v for k, v in zip(result['columns'], row)} for row in result['rows']]
+        rowdict = result['rows']
         self.assertEqual(rowdict[0]['request_id'], 1)
         self.assertEqual(rowdict[0]['method'], 'compile')
         self.assertEqual(rowdict[0]['state'], 'finished')
@@ -716,9 +714,7 @@ def test_ps_kill_postgres(self):
         all_ps_result = self.query('ps', completed=True, active=True).json()
         result = self.assertIsResult(all_ps_result)
         self.assertEqual(len(result['rows']), 2)
-        self.assertEqual(len(result['rows'][0]), len(result['columns']))
-        self.assertEqual(len(result['rows'][1]), len(result['columns']))
-        rowdict = [{k: v for k, v in zip(result['columns'], row)} for row in result['rows']]
+        rowdict = result['rows']
         rowdict.sort(key=lambda r: r['start'])
         self.assertEqual(rowdict[0]['request_id'], 1)
         self.assertEqual(rowdict[0]['method'], 'compile')
@@ -765,7 +761,7 @@ def _get_sleep_query(self):
             time.sleep(0.2)
             sleeper_ps_result = self.query('ps', completed=False, active=True).json()
             result = self.assertIsResult(sleeper_ps_result)
-            rows = [{k: v for k, v in zip(result['columns'], row)} for row in result['rows']]
+            rows = result['rows']
             for row in rows:
                 if row['request_id'] == request_id and row['state'] == 'running':
                     return pg_sleeper, row['task_id'], request_id