testing fixes

singer-io · Jan 29, 2018 · 8d6e6c1 · 8d6e6c1
1 parent 2768f12
commit 8d6e6c1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tap_hubspot/__init__.py b/tap_hubspot/__init__.py
@@ -801,13 +801,13 @@ def do_sync(STATE, catalogs):
 class Context(object):
     def __init__(self, catalog):
         self.selected_stream_ids = set(
-            s.get('tap_stream_id') for s in catalogs.get('streams')
+            s.get('tap_stream_id') for s in catalog.get('streams')
             if s.get('schema').get('selected'))
         self.catalog = catalog
 
-    def get_catalog_from_id(tap_stream_id):
-        return [c for c in catalogs.get('streams')
-               if c.get('stream') == stream.tap_stream_id][0]
+    def get_catalog_from_id(self,tap_stream_id):
+        return [c for c in self.catalog.get('streams')
+               if c.get('stream') == tap_stream_id][0]
 
 # stream a is dependent on stream STREAM_DEPENDENCIES[a]
 STREAM_DEPENDENCIES = {