def test_validate_timestamp(self, mock_time):
timestamp1 = 1577804700
mock_time.return_value = 1577804400
actual = FileControllerHelper.valid_timestamp(timestamp1)
self.assertTrue(actual)
timestamp1 = 1577804800
mock_time.return_value = 1577804400
actual = FileControllerHelper.valid_timestamp(timestamp1)
self.assertFalse(actual)
def test_parse_text(self):
text1 = ' from=2020/01/01 to=2020/11/11'
actual1 = FileControllerHelper.parse_text(text1)
self.assertEqual('2020/01/01', actual1.date_from)
self.assertEqual('2020/11/11', actual1.date_to)
self.assertEqual(actual1.all_users, False)
self.assertEqual(actual1.all_channels, False)
text2 = ' from=2020/01/01 to=2020/11/11 -ac -au'
actual2 = FileControllerHelper.parse_text(text2)
self.assertEqual('2020/01/01', actual2.date_from)
self.assertEqual('2020/11/11', actual2.date_to)
self.assertEqual(actual2.all_users, True)
self.assertEqual(actual2.all_channels, True)
text3 = ' from=2020/01/01 to=2020/11/11 --all-channels --all-users'
actual3 = FileControllerHelper.parse_text(text3)
self.assertEqual('2020/01/01', actual3.date_from)
self.assertEqual('2020/11/11', actual3.date_to)
self.assertEqual(actual3.all_users, True)
self.assertEqual(actual3.all_channels, True)
def test_get_command_header(self):
event: Dict = {
'headers': {
'x-slack-request-timestamp': '1577804400',
'x-slack-signature': 'sig1'
}
}
actual = FileControllerHelper.get_command_header(event)
self.assertEqual(actual.slack_signature, "sig1")
self.assertEqual(actual.slack_request_timestamp, 1577804400)
def test_valid_signature(self):
headers1 = SlashCommandHeaders(
slack_request_timestamp=1531420618,
slack_signature=
'v0=a2114d57b48eac39b9ad189dd8316235a7b4a8d21a10bd27519666489c69b503'
)
body1 = 'token=xyzz0WbapA4vBCDEFasx0q6G&team_id=T1DC2JH3J&team_domain=testteamnow&channel_id=G8PSS9T3V' \
'&channel_name=foobar&user_id=U2CERLKJA&user_name=roadrunner&command=%2Fwebhook-collect&text' \
'=&response_url=https%3A%2F%2Fhooks.slack.com%2Fcommands%2FT1DC2JH3J%2F397700885554' \
'%2F96rGlfmibIGlgcZRskXaIFfN&trigger_id=398738663015.47445629121.803a0bc887a14d10d2c447fce8b6703c'
actual1 = FileControllerHelper.valid_signature(
headers1, body1, '8f742231b10e8888abcd99yyyzzz85a5')
self.assertTrue(actual1)
headers2 = SlashCommandHeaders(slack_request_timestamp=1531420618,
slack_signature='v0=wrong')
body2 = 'token=xyzz0WbapA4vBCDEFasx0q6G&team_id=T1DC2JH3J&team_domain=testteamnow&channel_id=G8PSS9T3V' \
'&channel_name=foobar&user_id=U2CERLKJA&user_name=roadrunner&command=%2Fwebhook-collect&text' \
'=&response_url=https%3A%2F%2Fhooks.slack.com%2Fcommands%2FT1DC2JH3J%2F397700885554' \
'%2F96rGlfmibIGlgcZRskXaIFfN&trigger_id=398738663015.47445629121.803a0bc887a14d10d2c447fce8b6703c'
actual2 = FileControllerHelper.valid_signature(
headers2, body2, '8f742231b10e8888abcd99yyyzzz85a5')
self.assertFalse(actual2)
def test_get_command_body(self):
body: Dict = {
'token': 'token1',
'user_id': 'user_id1',
'channel_id': 'channel_id1',
'command': 'command1',
'text': 'text1',
}
encoded_dict = urlencode(body)
encoded = base64.b64encode(encoded_dict.encode('utf-8'))
event: Dict = {'body': encoded}
actual = FileControllerHelper.get_command_body(event)
self.assertEqual('token1', actual.token)
self.assertEqual('user_id1', actual.user_id)
self.assertEqual('channel_id1', actual.channel_id)
self.assertEqual('command1', actual.command)
self.assertEqual('text1', actual.text)
def list(self, event: Dict) -> Dict[str, Union[int, Dict[str, str], str]]:
try:
is_local = self.__env_vars.stage == 'local'
valid_timestamp = True
valid_signature = True
if not is_local:
headers: SlashCommandHeaders = FileControllerHelper.get_command_header(event)
raw_body = FileControllerHelper.get_raw_body(event)
valid_signature = FileControllerHelper.valid_signature(
headers,
raw_body,
self.__env_vars.slack_signing_secret)
valid_timestamp = FileControllerHelper.valid_timestamp(headers.slack_request_timestamp)
if not valid_timestamp:
raise InvalidTimestampException(event['headers'])
if not valid_signature:
raise InvalidSignatureException(event['headers'])
body: SlashCommandBody = FileControllerHelper.get_command_body(event)
FileControllerHelper.validate_body(body)
parsed_text: ParsedText = FileControllerHelper.parse_text(body.text)
input_data: FileListInput = FileListInput(
token=self.__env_vars.token,
date_from=parsed_text.date_from,
date_to=parsed_text.date_to,
channel=body.channel_id if parsed_text.all_channels else None,
user=body.user_id if parsed_text.all_users else None,
)
output: FileListOutput = self.__usecase.handle(input_data)
return self.__presenter.complete(output, None)
except Exception as ex:
return self.__presenter.complete(FileListOutput([]), ex)
def test_validate_body__ok(self):
body1 = SlashCommandBody(
token='token1',
user_id='user_id1',
channel_id='channel_id1',
command='command1',
text='to=2020/01/01 from=2020/11/11',
)
FileControllerHelper.validate_body(body1)
body2 = SlashCommandBody(
token='token2',
user_id='user_id2',
channel_id='channel_id2',
command='command2',
text=' to=2020/01/01 from=2020/11/11',
)
FileControllerHelper.validate_body(body2)
body3 = SlashCommandBody(
token='token3',
user_id='user_id3',
channel_id='channel_id3',
command='command3',
text=' to=2020/01/01 from=2020/11/11 -ac -au',
)
FileControllerHelper.validate_body(body3)
body4 = SlashCommandBody(
token='token4',
user_id='user_id4',
channel_id='channel_id4',
command='command4',
text=' to=2020/01/01 from=2020/11/11 --all-channels --all-users',
)
FileControllerHelper.validate_body(body4)
body5 = SlashCommandBody(
token='token5',
user_id='user_id5',
channel_id='channel_id5',
command='command5',
text=' to=2020/01/01 from=2020/11/11 -ac',
)
FileControllerHelper.validate_body(body5)
body6 = SlashCommandBody(
token='token6',
user_id='user_id6',
channel_id='channel_id6',
command='command6',
text=' to=2020/01/01 from=2020/11/11 -au',
)
FileControllerHelper.validate_body(body6)
body7 = SlashCommandBody(
token='token7',
user_id='user_id7',
channel_id='channel_id7',
command='command7',
text=' to=2020/01/01 from=2020/11/11 --all-users',
)
FileControllerHelper.validate_body(body7)
body8 = SlashCommandBody(
token='token8',
user_id='user_id8',
channel_id='channel_id8',
command='command8',
text=' to=2020/01/01 from=2020/11/11 --all-channels',
)
FileControllerHelper.validate_body(body8)
def test_validate_body__raise(self):
body1 = SlashCommandBody(
token='',
user_id='user_id1',
channel_id='channel_id1',
command='command1',
text='to=2020/01/01 from=2020/11/11',
)
with self.assertRaises(ValidationException):
FileControllerHelper.validate_body(body1)
body2 = SlashCommandBody(
token='token2',
user_id='',
channel_id='channel_id2',
command='command2',
text='to=2020/01/01 from=2020/11/11',
)
with self.assertRaises(ValidationException):
FileControllerHelper.validate_body(body2)
body3 = SlashCommandBody(
token='token3',
user_id='user_id3',
channel_id='',
command='command3',
text='to=2020/01/01 from=2020/11/11',
)
with self.assertRaises(ValidationException):
FileControllerHelper.validate_body(body3)
body4 = SlashCommandBody(
token='token4',
user_id='user_id4',
channel_id='channel_id4',
command='',
text='to=2020/01/01 from=2020/11/11',
)
with self.assertRaises(ValidationException):
FileControllerHelper.validate_body(body4)
body5 = SlashCommandBody(
token='token5',
user_id='user_id5',
channel_id='channel_id5',
command='command5',
text='',
)
with self.assertRaises(ValidationException):
FileControllerHelper.validate_body(body5)
body6 = SlashCommandBody(
token='token6',
user_id='user_id6',
channel_id='channel_id6',
command='command6',
text='to=2020/01/01 from=xxxx/xx/xx',
)
with self.assertRaises(ValidationException):
FileControllerHelper.validate_body(body6)
body7 = SlashCommandBody(
token='token7',
user_id='user_id7',
channel_id='channel_id7',
command='command7',
text='to=xxxx/xx/xx from=2020/10/10',
)
with self.assertRaises(ValidationException):
FileControllerHelper.validate_body(body7)