summaryrefslogtreecommitdiff
path: root/fs/tests/test_multifs.py
blob: 2d70f82a5e0b71e19233355952518fc8c2ca9207 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
from fs.multifs import MultiFS
from fs.memoryfs import MemoryFS
import unittest

from six import b

class TestMultiFS(unittest.TestCase):

    def test_auto_close(self):
        """Test MultiFS auto close is working"""
        multi_fs = MultiFS()
        m1 = MemoryFS()
        m2 = MemoryFS()
        multi_fs.addfs('m1', m1)
        multi_fs.addfs('m2', m2)
        self.assert_(not m1.closed)
        self.assert_(not m2.closed)
        multi_fs.close()
        self.assert_(m1.closed)
        self.assert_(m2.closed)

    def test_no_auto_close(self):
        """Test MultiFS auto close can be disables"""
        multi_fs = MultiFS(auto_close=False)
        m1 = MemoryFS()
        m2 = MemoryFS()
        multi_fs.addfs('m1', m1)
        multi_fs.addfs('m2', m2)
        self.assert_(not m1.closed)
        self.assert_(not m2.closed)
        multi_fs.close()
        self.assert_(not m1.closed)
        self.assert_(not m2.closed)


    def test_priority(self):
        """Test priority order is working"""
        m1 = MemoryFS()
        m2 = MemoryFS()
        m3 = MemoryFS()
        m1.setcontents("name", b("m1"))
        m2.setcontents("name", b("m2"))
        m3.setcontents("name", b("m3"))
        multi_fs = MultiFS(auto_close=False)
        multi_fs.addfs("m1", m1)
        multi_fs.addfs("m2", m2)
        multi_fs.addfs("m3", m3)
        self.assert_(multi_fs.getcontents("name") == b("m3"))

        m1 = MemoryFS()
        m2 = MemoryFS()
        m3 = MemoryFS()
        m1.setcontents("name", b("m1"))
        m2.setcontents("name", b("m2"))
        m3.setcontents("name", b("m3"))
        multi_fs = MultiFS(auto_close=False)
        multi_fs.addfs("m1", m1)
        multi_fs.addfs("m2", m2, priority=10)
        multi_fs.addfs("m3", m3)
        self.assert_(multi_fs.getcontents("name") == b("m2"))

        m1 = MemoryFS()
        m2 = MemoryFS()
        m3 = MemoryFS()
        m1.setcontents("name", b("m1"))
        m2.setcontents("name", b("m2"))
        m3.setcontents("name", b("m3"))
        multi_fs = MultiFS(auto_close=False)
        multi_fs.addfs("m1", m1)
        multi_fs.addfs("m2", m2, priority=10)
        multi_fs.addfs("m3", m3, priority=10)
        self.assert_(multi_fs.getcontents("name") == b("m3"))

        m1 = MemoryFS()
        m2 = MemoryFS()
        m3 = MemoryFS()
        m1.setcontents("name", b("m1"))
        m2.setcontents("name", b("m2"))
        m3.setcontents("name", b("m3"))
        multi_fs = MultiFS(auto_close=False)
        multi_fs.addfs("m1", m1, priority=11)
        multi_fs.addfs("m2", m2, priority=10)
        multi_fs.addfs("m3", m3, priority=10)
        self.assert_(multi_fs.getcontents("name") == b("m1"))