class GetHomeFolder(MockTestCase):

    layer = FAVOURITES_ZCML_LAYER

    def setUp(self):
        super(GetHomeFolder, self).setUp()

        self.request = {}
        self.context = self.stub()

        self.mtool = self.mocker.mock(count=False)
        self.mock_tool(self.mtool, 'portal_membership')

        self.handler = DefaultFavouritesHandler(self.context, self.request)

    def test_with_homefolder(self):

        folder = object()
        self.expect(self.mtool.getHomeFolder()).result(folder)

        self.replay()

        result = self.handler.get_home_folder()

        self.assertEquals(folder, result)

    def test_without_homefolder(self):

        self.expect(self.mtool.getHomeFolder()).result(None)

        self.replay()

        self.assertRaises(NoHomeFolderError, self.handler.get_home_folder)
    def setUp(self):
        super(GetFavouriteFolderName, self).setUp()

        self.request = {}
        self.context = self.stub()

        self.rtool = self.mocker.mock(count=False)
        self.mock_utility(self.rtool, IRegistry)

        self.handler = DefaultFavouritesHandler(self.context, self.request)
    def setUp(self):
        super(GetHomeFolder, self).setUp()

        self.request = {}
        self.context = self.stub()

        self.mtool = self.mocker.mock(count=False)
        self.mock_tool(self.mtool, 'portal_membership')

        self.handler = DefaultFavouritesHandler(self.context, self.request)
class GetFavouriteFolderName(MockTestCase):

    layer = FAVOURITES_ZCML_LAYER

    def setUp(self):
        super(GetFavouriteFolderName, self).setUp()

        self.request = {}
        self.context = self.stub()

        self.rtool = self.mocker.mock(count=False)
        self.mock_utility(self.rtool, IRegistry)

        self.handler = DefaultFavouritesHandler(self.context, self.request)

    def test_with_registry(self):

        folder = object()
        self.expect(self.rtool.get(
            'ftw.dashboard.portlets.favourites.foldername')).result(folder)

        self.replay()

        result = self.handler.get_favourite_folder_name()

        self.assertEquals(folder, result)

    def test_without_registry(self):

        self.expect(self.rtool.get(
            'ftw.dashboard.portlets.favourites.foldername')).result(None)

        self.replay()

        self.assertRaises(
            RegistryKeyError, self.handler.get_favourite_folder_name)