forked from GoogleCloudPlatform/python-docs-samples
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain_test.py
More file actions
116 lines (92 loc) · 3.7 KB
/
main_test.py
File metadata and controls
116 lines (92 loc) · 3.7 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
# 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 time
import googleapiclient.discovery
import mock
import pytest
from slack.signature import SignatureVerifier
import main
kg_search = googleapiclient.discovery.build(
'kgsearch', 'v1',
developerKey=os.environ['KG_API_KEY'])
example_response = kg_search.entities().search(query='lion', limit=1).execute()
class Request(object):
def __init__(self, data='', headers={}):
self.data = data
self.headers = headers
def get_data(self):
return self.data
class TestGCFPySlackSample(object):
def test_verify_signature_request_form_empty(self):
with pytest.raises(ValueError):
request = Request()
main.verify_signature(request)
def test_verify_signature_token_incorrect(self):
with pytest.raises(ValueError):
request = Request(headers={'X-Slack-Signature': '12345'})
main.verify_signature(request)
def test_verify_web_hook_valid_request(self):
request = Request()
request.body = ''
now = str(int(time.time()))
verifier = SignatureVerifier(os.environ['SLACK_SECRET'])
test_signature = verifier.generate_signature(
timestamp=now,
body=''
)
request.headers = {
'X-Slack-Request-Timestamp': now,
'X-Slack-Signature': test_signature
}
main.verify_signature(request)
def test_format_slack_message(self):
message = main.format_slack_message('lion', example_response)
# Just make sure there's a result.
assert 'title' in message['attachments'][0]
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')
# Just make sure there's a result.
assert 'title' in message['attachments'][0]
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.form = {
'text': 'lion'
}
request.data = json.dumps(request.form)
now = str(int(time.time()))
verifier = SignatureVerifier(os.environ['SLACK_SECRET'])
test_signature = verifier.generate_signature(
timestamp=now,
body=request.data
)
request.method = 'POST'
request.headers = {
'X-Slack-Request-Timestamp': now,
'X-Slack-Signature': test_signature
}
with mock.patch('main.jsonify', side_effect=json.dumps):
response = main.kg_search(request)
assert 'lion' in response.lower()
assert 'color' in response.lower()