def test_creating_worker(self):
obj = convert_from_json("""[{
"type": "HandleSoftBan"
}]""")
builder = TreeConfigBuilder(self.bot, obj)
tree = builder.build()
self.assertIsInstance(tree[0], HandleSoftBan)
self.assertIs(tree[0].bot, self.bot)
def test_creating_worker(self):
obj = convert_from_json("""[{
"type": "HandleSoftBan"
}]""")
builder = TreeConfigBuilder(self.bot, obj)
tree = builder.build()
self.assertIsInstance(tree[0], HandleSoftBan)
self.assertIs(tree[0].bot, self.bot)
def test_task_with_config(self):
obj = convert_from_json("""[{
"type": "IncubateEggs",
"config": {
"longer_eggs_first": true
}
}]""")
builder = TreeConfigBuilder(self.bot, obj)
tree = builder.build()
self.assertTrue(tree[0].config.get('longer_eggs_first', False))
def test_task_with_config(self):
obj = convert_from_json("""[{
"type": "IncubateEggs",
"config": {
"longer_eggs_first": true
}
}]""")
builder = TreeConfigBuilder(self.bot, obj)
tree = builder.build()
self.assertTrue(tree[0].config.get('longer_eggs_first', False))
def test_load_plugin_task(self):
package_path = os.path.join(os.path.abspath(os.path.dirname(__file__)), 'resources', 'plugin_fixture')
plugin_loader = PluginLoader()
plugin_loader.load_plugin(package_path)
obj = convert_from_json("""[{
"type": "plugin_fixture.FakeTask"
}]""")
builder = TreeConfigBuilder(self.bot, obj)
tree = builder.build()
result = tree[0].work()
self.assertEqual(result, 'FakeTask')
def test_load_plugin_task(self):
package_path = os.path.join(os.path.abspath(os.path.dirname(__file__)),
'resources', 'plugin_fixture')
plugin_loader = PluginLoader()
plugin_loader.load_plugin(package_path)
obj = convert_from_json("""[{
"type": "plugin_fixture.FakeTask"
}]""")
builder = TreeConfigBuilder(self.bot, obj)
tree = builder.build()
result = tree[0].work()
self.assertEqual(result, 'FakeTask')
def test_disabling_task(self):
obj = convert_from_json("""[{
"type": "HandleSoftBan",
"config": {
"enabled": false
}
}, {
"type": "CatchPokemon",
"config": {
"enabled": true
}
}]""")
builder = TreeConfigBuilder(self.bot, obj)
tree = builder.build()
self.assertTrue(len(tree) == 1)
self.assertIsInstance(tree[0], CatchPokemon)
def test_disabling_task(self):
obj = convert_from_json("""[{
"type": "HandleSoftBan",
"config": {
"enabled": false
}
}, {
"type": "CatchLuredPokemon",
"config": {
"enabled": true
}
}]""")
builder = TreeConfigBuilder(self.bot, obj)
tree = builder.build()
self.assertTrue(len(tree) == 1)
self.assertIsInstance(tree[0], CatchLuredPokemon)
