def setUp(self): self.fs = FileSystem()
class SimpleTests(unittest.TestCase):
def setUp(self):
self.fs = FileSystem()
def test_add(self):
m = Mock()
self.fs.add("/foo/bar", m)
self.assertEqual(self.fs.get_node("/foo/bar"), m)
def test_remove(self):
self.fs.add("/foo/bar", "baz")
self.fs.remove("/foo/bar")
self.assertRaises(Exception, lambda: self.fs.get_node("/foo/bar"))
self.assertEqual(self.fs.list_dir(), {})
def test_list_dir(self):
m = Mock()
self.fs.add("/foo/bar", m)
self.assertEqual(self.fs.list_dir(),
{'foo': {'bar': m.type}})
self.assertEqual(self.fs.list_dir("/"),
{'foo': {'bar': m.type}})
self.assertEqual(self.fs.list_dir("/", False),
{'foo': 'dir'})
self.assertEqual(self.fs.list_dir("/foo", False),
{'bar': m.type})
class ComplexTests(unittest.TestCase):
def setUp(self):
self.fs = FileSystem()
def test_same_dir_add_remove(self):
foo = Mock()
bar = Mock()
self.fs.add("/baz/foo", foo)
self.fs.add("/baz/bar", bar)
self.assertEqual(self.fs.list_dir(),
{'baz': {'foo': foo.type, 'bar': bar.type}})
self.fs.remove("/baz/foo")
self.assertEqual(self.fs.list_dir(),
{'baz': {'bar': bar.type}})
self.fs.remove("/baz/bar")
self.assertEqual(self.fs.list_dir(),
{})
def test_add_remove_with_sub_dir(self):
foo = Mock()
baz = Mock()
self.fs.add("/dir/foo", foo)
self.fs.add("/dir/bar/baz", baz)
self.assertEqual(self.fs.list_dir(),
{'dir': {'foo': foo.type, 'bar': {'baz': baz.type}}})
self.fs.remove("/dir/foo")
self.assertEqual(self.fs.list_dir(),
{'dir': {'bar': {'baz': baz.type}}})
self.fs.remove("/dir/bar/baz")
self.assertEqual(self.fs.list_dir(),
{})
class EdgeCaseTests(unittest.TestCase):
def setUp(self):
self.fs = FileSystem()
def test_add_same(self):
self.fs.add("/foo/bar", "baz")
self.assertRaises(Exception, lambda: self.fs.add("/foo/bar", "foo"))
def test_add_over_dir(self):
self.fs.add("/foo/bar", "baz")
self.assertRaises(Exception, lambda: self.fs.add("/foo", "foo"))
def test_remove_nonexistant(self):
self.assertRaises(Exception, lambda: self.fs.remove("/foo"))
self.assertRaises(Exception, lambda: self.fs.remove("/foo/bar"))
def test_add_deeper(self):
m = Mock()
self.fs.add("/foo/bar", m)
self.assertRaises(Exception, lambda: self.fs.add("/foo/bar/baz", "quux"))
def test_get_deeper(self):
m = Mock()
self.fs.add("/foo/bar", m)
self.assertRaises(Exception, lambda: self.fs.get_node("/foo/bar/baz"))
