nrupal
diff --git a/‎functions/slack/main_test.py
Copy file name to clipboardExpand all lines: functions/slack/main_test.py
+2-4Lines changed: 2 additions & 4 deletions b/‎functions/slack/main_test.py
Copy file name to clipboardExpand all lines: functions/slack/main_test.py
+2-4Lines changed: 2 additions & 4 deletions
@@ -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()