diff --git a/contrib_bots/bots/encrypt/test_encrypt.py b/contrib_bots/bots/encrypt/test_encrypt.py new file mode 100644 index 0000000000..7501fbda14 --- /dev/null +++ b/contrib_bots/bots/encrypt/test_encrypt.py @@ -0,0 +1,38 @@ +#!/usr/bin/env python + +from __future__ import absolute_import +from __future__ import print_function + +import os +import sys + +our_dir = os.path.dirname(os.path.abspath(__file__)) +# For dev setups, we can find the API in the repo itself. +if os.path.exists(os.path.join(our_dir, '..')): + sys.path.insert(0, '..') +from bots_test_lib import BotTestCase + +class TestEncryptBot(BotTestCase): + bot_name = "encrypt" + + def test_bot(self): + self.assert_bot_output( + {'content': "Please encrypt this", 'type': "private", 'sender_email': "foo@gmail.com"}, + "Encrypted/Decrypted text: Cyrnfr rapelcg guvf" + ) + self.assert_bot_output( + {'content': "Let\'s Do It", 'type': "stream", 'display_recipient': "foo", 'subject': "foo"}, + "Encrypted/Decrypted text: Yrg\'f Qb Vg" + ) + self.assert_bot_output( + {'content': "", 'type': "stream", 'display_recipient': "foo", 'subject': "foo"}, + "Encrypted/Decrypted text: " + ) + self.assert_bot_output( + {'content': "me&mom together..!!", 'type': "stream", 'display_recipient': "foo", 'subject': "foo"}, + "Encrypted/Decrypted text: zr&zbz gbtrgure..!!" + ) + self.assert_bot_output( + {'content': "foo bar", 'type': "stream", 'display_recipient': "foo", 'subject': "foo"}, + "Encrypted/Decrypted text: sbb one" + ) diff --git a/contrib_bots/bots/encrypt/tests.py b/contrib_bots/bots/encrypt/tests.py deleted file mode 100644 index 0e1a3cd3a1..0000000000 --- a/contrib_bots/bots/encrypt/tests.py +++ /dev/null @@ -1,33 +0,0 @@ -from . import encrypt - -def test(): - for cmd, expected_response in sample_conversation(): - message = {'content': cmd, 'subject': 'foo', - 'display_recipient': 'bar'} - - class ClientDummy(object): - def __init__(self): - self.output = '' - - def send_message(self, params): - self.output = params['content'] - handler = encrypt.EncryptHandler() - client_dummy = ClientDummy() - handler.handle_message(message, client_dummy, '') - if client_dummy.output != expected_response: - raise AssertionError(''' - cmd: %s - expected: %s - but got : %s - ''' % (cmd, expected_response, client_dummy.output)) - -def sample_conversation(): - return [ - ('@encrypt Please encrypt this', 'Encrypted/Decrypted text: Cyrnfr rapelcg guvf'), - ('@encrypt Let\'s Do It', 'Encrypted/Decrypted text: Yrg\'f Qb Vg'), - ('@encrypt ', 'Encrypted/Decrypted text: '), - ('@encrypt me&mom together..!!', 'Encrypted/Decrypted text: zr&zbz gbtrgure..!!'), - ] - -if __name__ == '__main__': - test()