import sys import unittest from pathlib import Path from unittest import mock SOURCE = Path(__file__).resolve().parents[1] / "src" sys.path.insert(0, str(SOURCE)) from coin_transfer import AUTHOR, DESCRIPTION, HELP_ENTRIES, plugin class PluginTests(unittest.TestCase): def test_registers_declared_commands_and_author(self): context = mock.Mock() plugin(context) registered = [ call.args[0] for call in context.commands.register.call_args_list ] self.assertEqual(AUTHOR, 'jwebot') self.assertEqual(DESCRIPTION, '金币转账') self.assertEqual(HELP_ENTRIES, ('/转账 对方 金额;/transfer',)) self.assertIn('/转账', registered) self.assertEqual(len(registered), 2) if __name__ == "__main__": unittest.main()