Add stream_alias to Catalog.to_dict. Add to unit test.

flash716 · flash716 · commit 219b8eb6d582 · 2017-12-10T15:47:40.000-08:00
diff --git a/singer/catalog.py b/singer/catalog.py
@@ -55,6 +55,8 @@ def to_dict(self):
             result['stream'] = self.stream
         if self.row_count is not None:
             result['row_count'] = self.row_count
+        if self.stream_alias is not None:
+            result['stream_alias'] = self.stream_alias
         if self.metadata is not None:
             result['metadata'] = self.metadata
         return result
diff --git a/tests/test_catalog.py b/tests/test_catalog.py
@@ -10,6 +10,7 @@ class TestToDictAndFromDict(unittest.TestCase):
             {
                 'stream': 'users',
                 'tap_stream_id': 'prod_users',
+                'stream_alias': 'users_alias',
                 'database_name': 'prod',
                 'table_name': 'users',
                 'schema': {
@@ -42,6 +43,7 @@ class TestToDictAndFromDict(unittest.TestCase):
         CatalogEntry(
             stream='users',
             tap_stream_id='prod_users',
+            stream_alias='users_alias',
             database='prod',
             table='users',
             schema=Schema(
