1 | #!/usr/bin/python |
---|
2 | |
---|
3 | import unittest |
---|
4 | from mock import Mock, patch |
---|
5 | |
---|
6 | from allmydata.web.root import Root |
---|
7 | from allmydata.client import Client |
---|
8 | |
---|
9 | class TestRoot(unittest.TestCase): |
---|
10 | |
---|
11 | @patch('allmydata.web.root.Root') |
---|
12 | def test_introducer_furls(self, MockRoot): |
---|
13 | # items needed to call data_introducer_furl() |
---|
14 | mockctx = Mock() |
---|
15 | mockdata = Mock() |
---|
16 | myclient = Client() |
---|
17 | furls = myclient.introducer_furls |
---|
18 | furl_count = len(furls) |
---|
19 | #print "furls: %d" %furl_count |
---|
20 | # Pass mock value to Root |
---|
21 | myroot = Root(myclient) |
---|
22 | |
---|
23 | # make the call |
---|
24 | s = myroot.data_introducers(mockctx, mockdata) |
---|
25 | |
---|
26 | #assertions: compare return value with preset value |
---|
27 | self.failUnlessEqual(furl_count, len(s)) |
---|
28 | |
---|
29 | if __name__ == "__main__": |
---|
30 | unittest.main() |
---|