class TestAddItems(TestCase): def setUp(self): sys.argv = [ 'doesnotmatter.py', 'interactive', 'special://my.plugin.id/testpath/' ] self.plugin = Plugin('My Plugin', 'my.plugin.id') @self.plugin.route('/videos/') def show_videos(): return 'videos' def test_1_item(self): items = [ { 'label': 'My video', 'url': self.plugin.url_for('show_videos'), }, { 'label': 'My video2', 'url': self.plugin.url_for('show_videos'), }, ] urls = self.plugin.add_items(items) self.assertEqual(urls, [ 'special://my.plugin.id/videos/', 'special://my.plugin.id/videos/' ])
class TestAddItems(TestCase): def setUp(self): sys.argv = ["doesnotmatter.py", "interactive", "special://my.plugin.id/testpath/"] self.plugin = Plugin("My Plugin", "my.plugin.id") @self.plugin.route("/videos/") def show_videos(): return "videos" def test_1_item(self): items = [ {"label": "My video", "url": self.plugin.url_for("show_videos")}, {"label": "My video2", "url": self.plugin.url_for("show_videos")}, ] urls = self.plugin.add_items(items) self.assertEqual(urls, ["special://my.plugin.id/videos/", "special://my.plugin.id/videos/"])