from . import GalaxyTestBase

FOO_DATA = 'foo\nbar\n'


class TestGalaxyFolders(GalaxyTestBase.GalaxyTestBase):

    def setUp(self):
        super().setUp()
        self.name = 'automated test folder'
        self.library = self.gi.libraries.create_library(self.name, description='automated test', synopsis='automated test synopsis')
        self.folder = self.gi.folders.create_folder(self.library['root_folder_id'], self.name,
                                                    description="automatically created folder")

    def tearDown(self):
        self.gi.libraries.delete_library(self.library['id'])

    def test_create_folder(self):
        self.assertEqual(self.folder['name'], self.name)
        self.assertEqual(self.folder['description'], 'automatically created folder')

    def test_show_folder(self):
        f2 = self.gi.folders.show_folder(self.folder['id'])
        self.assertEqual(f2['id'], self.folder['id'])

    def test_show_folder_contents(self):
        f2 = self.gi.folders.show_folder(self.folder['id'], contents=True)
        self.assertIn('folder_contents', f2)
        self.assertIn('metadata', f2)
        self.assertEqual(self.name, f2['metadata']['folder_name'])

    def test_delete_folder(self):
        self.sub_folder = self.gi.folders.create_folder(self.folder['id'], self.name)
        self.gi.folders.delete_folder(self.sub_folder['id'])

    def test_update_folder(self):
        self.folder = self.gi.folders.update_folder(self.folder['id'], 'new-name', 'new-description')
        self.assertEqual(self.folder['name'], 'new-name')
        self.assertEqual(self.folder['description'], 'new-description')

    def test_get_set_permissions(self):
        empty_permission = {'add_library_item_role_list': [], 'modify_folder_role_list': [], 'manage_folder_role_list': []}
        # They should be empty to start with
        self.assertEqual(
            self.gi.folders.get_permissions(self.folder['id'], scope='current'),
            empty_permission
        )
        self.assertEqual(
            self.gi.folders.get_permissions(self.folder['id'], scope='available'),
            empty_permission
        )
        # Then we'll add a role
        role = self.gi.roles.get_roles()[0]
        self.gi.folders.set_permissions(self.folder['id'], add_ids=[role['id']])
        self.assertTrue(
            role['id'] in
            self.gi.folders.get_permissions(self.folder['id'], scope='available')['add_library_item_role_list'][0]
        )