JezK
Edit File: hostconfig.py
# Unix SMB/CIFS implementation. Tests for shares # Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2009 # # This program is free software; you can redistribute it and/or modify # it under the terms of the GNU General Public License as published by # the Free Software Foundation; either version 3 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program. If not, see <http://www.gnu.org/licenses/>. # """Tests for samba.hostconfig.""" from samba.hostconfig import SharesContainer from samba.tests import TestCase class MockService(object): def __init__(self, data): self.data = data def __getitem__(self, name): return self.data[name] class MockLoadParm(object): def __init__(self, data): self.data = data def __getitem__(self, name): return MockService(self.data[name]) def __len__(self): return len(self.data) def services(self): return self.data.keys() class ShareTests(TestCase): def _get_shares(self, conf): return SharesContainer(MockLoadParm(conf)) def test_len_no_global(self): shares = self._get_shares({}) self.assertEqual(0, len(shares)) def test_iter(self): self.assertEqual([], list(self._get_shares({}))) self.assertEqual([], list(self._get_shares({"global": {}}))) self.assertEqual( ["bla"], list(self._get_shares({"global": {}, "bla": {}}))) def test_len(self): shares = self._get_shares({"global": {}}) self.assertEqual(0, len(shares)) def test_getitem_nonexistent(self): shares = self._get_shares({"global": {}}) self.assertRaises(KeyError, shares.__getitem__, "bla") def test_getitem_global(self): shares = self._get_shares({"global": {}}) self.assertRaises(KeyError, shares.__getitem__, "global")