import unittest
import ganesha


EXAMPLE_EXPORT = """## This export is managed by the CephNFS charm ##
EXPORT {
    # Each EXPORT must have a unique Export_Id.
    Export_Id = 1000;

    # The directory in the exported file system this export
    # is rooted on.
    Path = '/volumes/_nogroup/test_ganesha_share/e12a49ef-1b2b-40b3-ba6c';

    # FSAL, Ganesha's module component
    FSAL {
        # FSAL name
        Name = "Ceph";
        User_Id = "ganesha-test_ganesha_share";
        Secret_Access_Key = "AQCT9+9h4cwJOxAAue2fFvvGTWziUiR9koCHEw==";
    }

    # Path of export in the NFSv4 pseudo filesystem
    Pseudo = '/volumes/_nogroup/test_ganesha_share/e12a49ef-1b2b-40b3-ba6c';

    SecType = "sys";
    CLIENT {
        Access_Type = "rw";
        Clients = 0.0.0.0;
    }
    # User id squashing, one of None, Root, All
    Squash = "None";
}
"""


class ExportTest(unittest.TestCase):

    def test_parser(self):
        export = ganesha.Export.from_export(EXAMPLE_EXPORT)
        self.assertEqual(export.export_id, 1000)
        self.assertEqual(export.clients,
                         [{'Access_Type': 'rw', 'Clients': '0.0.0.0'}])
        self.assertEqual(export.name, 'test_ganesha_share')

    def test_add_client(self):
        export = ganesha.Export.from_export(EXAMPLE_EXPORT)
        export.add_client('10.0.0.0/8')
        self.assertEqual(
            export.clients,
            [{'Access_Type': 'rw', 'Clients': '0.0.0.0, 10.0.0.0/8'}])
        # adding again shouldn't duplicate export
        export.add_client('10.0.0.0/8')
        self.assertEqual(
            export.clients,
            [{'Access_Type': 'rw', 'Clients': '0.0.0.0, 10.0.0.0/8'}])

        export.add_client('192.168.0.0/16')
        self.assertEqual(
            export.clients,
            [{
                'Access_Type': 'rw',
                'Clients': '0.0.0.0, 10.0.0.0/8, 192.168.0.0/16'
            }])

    def test_remove_client(self):
        export = ganesha.Export.from_export(EXAMPLE_EXPORT)
        export.add_client('10.0.0.0/8')
        export.add_client('192.168.0.0/16')
        self.assertEqual(
            export.clients,
            [{
                'Access_Type': 'rw',
                'Clients': '0.0.0.0, 10.0.0.0/8, 192.168.0.0/16'
            }])
        export.remove_client('0.0.0.0')
        self.assertEqual(
            export.clients,
            [
                {'Access_Type': 'rw', 'Clients': '10.0.0.0/8, 192.168.0.0/16'},
            ])


class TestGaneshaNFS(unittest.TestCase):

    @unittest.mock.patch.object(ganesha.GaneshaNFS, '_ceph_subvolume_command')
    @unittest.mock.patch.object(ganesha.GaneshaNFS, '_ganesha_add_export')
    @unittest.mock.patch.object(ganesha.GaneshaNFS, '_get_next_export_id')
    @unittest.mock.patch.object(ganesha.GaneshaNFS, 'list_shares')
    @unittest.mock.patch.object(ganesha.GaneshaNFS, '_ceph_auth_key')
    @unittest.mock.patch.object(ganesha.GaneshaNFS, '_rados_get')
    @unittest.mock.patch.object(ganesha.GaneshaNFS, '_rados_put')
    @unittest.mock.patch.object(ganesha.Export, 'to_export')
    def test_create_share(self, mock_export,
                          mock_rados_put,
                          mock_rados_get,
                          mock_auth_key,
                          mock_list_shares,
                          mock_export_id,
                          mock_add_export,
                          mock_subvolume_command):
        mock_subvolume_command.return_value = b'mock-volume'
        mock_list_shares.return_value = []
        mock_export_id.return_value = 1
        mock_auth_key.return_value = b'mock-auth-key'

        inst = ganesha.GaneshaNFS('ceph-client', 'mypool')
        inst.create_share('test-create-share', size=3, access_ips=None)

        mock_subvolume_command.assert_any_call('create', 'ceph-fs',
                                               'test-create-share',
                                               str(3 * 1024 * 1024 * 1024))

    @unittest.mock.patch.object(ganesha.GaneshaNFS, '_ceph_subvolume_command')
    def test_resize_share(self, mock_subvolume_command):
        inst = ganesha.GaneshaNFS('ceph-client', 'mypool')
        inst.resize_share('test-resize-share', 5)
        mock_subvolume_command.assert_any_call('resize', 'ceph-fs',
                                               'test-resize-share',
                                               str(5 * 1024 * 1024 * 1024),
                                               '--no_shrink')