From f421e90a648b68dbd59af692c537724cacc806b7 Mon Sep 17 00:00:00 2001 From: Steve Howell Date: Thu, 7 Dec 2017 19:46:29 -0800 Subject: [PATCH] bots: Simplify test_encrpyt.py. --- zulip_bots/zulip_bots/bots/encrypt/test_encrypt.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/zulip_bots/zulip_bots/bots/encrypt/test_encrypt.py b/zulip_bots/zulip_bots/bots/encrypt/test_encrypt.py index e03f0e3..a531d5e 100755 --- a/zulip_bots/zulip_bots/bots/encrypt/test_encrypt.py +++ b/zulip_bots/zulip_bots/bots/encrypt/test_encrypt.py @@ -1,19 +1,16 @@ #!/usr/bin/env python -from __future__ import absolute_import -from __future__ import print_function +from zulip_bots.test_lib import StubBotTestCase -from zulip_bots.test_lib import BotTestCase - -class TestEncryptBot(BotTestCase): +class TestEncryptBot(StubBotTestCase): bot_name = "encrypt" def test_bot(self): - expected = [ + dialog = [ ("", "Encrypted/Decrypted text: "), ("Let\'s Do It", "Encrypted/Decrypted text: Yrg\'f Qb Vg"), ("me&mom together..!!", "Encrypted/Decrypted text: zr&zbz gbtrgure..!!"), ("foo bar", "Encrypted/Decrypted text: sbb one"), ("Please encrypt this", "Encrypted/Decrypted text: Cyrnfr rapelcg guvf"), ] - self.check_expected_responses(expected) + self.verify_dialog(dialog)