Fix test for slack slash using knowledge graph API (GoogleCloudPlatform#1613)

yukuairoy · engelke · commit f241636eafa6 · 2018-12-06T17:50:21.000-08:00
* Fix test

* remove import os that's no longer needed
diff --git a/functions/slack/main_test.py b/functions/slack/main_test.py
@@ -12,21 +12,19 @@
 # limitations under the License.
 
 import json
-import os
 
 import apiclient
 import mock
 import pytest
 
 import main
 
-API_KEY = os.environ['API_KEY']
-
 with open('config.json', 'r') as f:
     data = f.read()
 config = json.loads(data)
 
-kg_search = apiclient.discovery.build('kgsearch', 'v1', developerKey=API_KEY)
+kg_search = apiclient.discovery.build('kgsearch', 'v1',
+                                      developerKey=config['KG_API_KEY'])
 example_response = kg_search.entities().search(query='lion', limit=1).execute()
 
 
