# Copyright 2018, Google, LLC.

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

#

# http://www.apache.org/licenses/LICENSE-2.0

#

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# 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)

example_response = kg_search.entities().search(query='lion', limit=1).execute()

class Request(object):

def __init__(self):

pass

class TestGCFPySlackSample(object):

def test_verify_web_hook_request_form_empty(self):

with pytest.raises(ValueError):

main.verify_web_hook({})

def test_verify_web_hook_token_incorrect(self):

with pytest.raises(ValueError):

main.verify_web_hook({'token': 123})

def test_verify_web_hook_valid_request(self):

main.verify_web_hook({'token': config['SLACK_TOKEN']})

def test_format_slack_message(self):

message = main.format_slack_message('lion', example_response)

assert 'lion' in message['text'].lower()

assert 'lion' in message['attachments'][0]['title'].lower()

assert message['attachments'][0]['color'] == '#3367d6'

def test_make_search_request(self):

with mock.patch.object(main, 'kgsearch'):

entities = main.kgsearch.entities.return_value

search = entities.search.return_value

search.execute.return_value = example_response

message = main.make_search_request('lion')

assert 'lion' in message['text'].lower()

assert 'lion' in message['attachments'][0]['title'].lower()

assert message['attachments'][0]['color'] == '#3367d6'

def test_kg_search(self):

with mock.patch.object(main, 'kgsearch'):

entities = main.kgsearch.entities.return_value

search = entities.search.return_value

search.execute.return_value = example_response

request = Request()

request.method = 'POST'

request.form = {

'token': config['SLACK_TOKEN'],

'text': 'lion'

}

with mock.patch('main.jsonify', side_effect=json.dumps):

response = main.kg_search(request)

assert 'lion' in response.lower()

assert 'color' in response.lower()