#!/usr/bin/env python

from __future__ import absolute_import
from __future__ import print_function

from zulip_bots.test_lib import BotTestCase

class TestThesaurusBot(BotTestCase):
    bot_name = "thesaurus"

    def test_bot(self):
        expected = {
            "synonym good": "great, satisfying, exceptional, positive, acceptable",
            "synonym nice": "cordial, kind, good, okay, fair",
            "synonym foo": "bar, thud, X, baz, corge",
            "antonym dirty": "ordered, sterile, spotless, moral, clean",
            "antonym bar": "loss, whole, advantage, aid, failure",
            "": ("To use this bot, start messages with either "
                 "@mention-bot synonym (to get the synonyms of a given word) "
                 "or @mention-bot antonym (to get the antonyms of a given word). "
                 "Phrases are not accepted so only use single words "
                 "to search. For example you could search '@mention-bot synonym hello' "
                 "or '@mention-bot antonym goodbye'."),
        }
        self.check_expected_responses(expected)
