def test_dump(self):
self.protocol.answers.append(
"""info/names=
multi/path/arg/* a documentation string
"""
)
info = TorInfo(self.protocol)
info.dump()
def test_simple(self):
self.protocol.answers.append('''info/names=
something a documentation string
multi/path a documentation string
''')
info = TorInfo(self.protocol)
self.assertTrue(hasattr(info, 'something'))
self.assertTrue(hasattr(info, 'multi'))
self.assertTrue(hasattr(getattr(info, 'multi'), 'path'))
self.protocol.answers.append('something=\nfoo\nOK')
d = info.something()
d.addCallback(CheckAnswer(self, 'foo'))
return d
def test_simple(self):
self.protocol.answers.append('''info/names=
something a documentation string
multi/path a documentation string
''')
info = TorInfo(self.protocol)
self.assertTrue(hasattr(info, 'something'))
self.assertTrue(hasattr(info, 'multi'))
self.assertTrue(hasattr(getattr(info, 'multi'), 'path'))
self.protocol.answers.append('something=\nfoo')
d = info.something()
d.addCallback(CheckAnswer(self, 'foo'))
return d
def test_prefix_error(self):
self.protocol.answers.append('''info/names=
something not allowed I hope
something/one a documentation string
''')
self.error_happened = False
TorInfo(self.protocol, self.handle_error)
self.assertTrue(self.error_happened)
def test_prefix_error_other_order(self):
self.protocol.answers.append('''info/names=
other/one a documentation string
other not allowed I hope
''')
self.error_happened = False
TorInfo(self.protocol, self.handle_error)
self.assertTrue(self.error_happened)
def test_simple(self):
self.protocol.answers.append(
"""info/names=
something a documentation string
multi/path a documentation string
"""
)
info = TorInfo(self.protocol)
self.assertTrue(hasattr(info, "something"))
self.assertTrue(hasattr(info, "multi"))
self.assertTrue(hasattr(getattr(info, "multi"), "path"))
self.protocol.answers.append("something=\nfoo\nOK")
d = info.something()
d.addCallback(CheckAnswer(self, "foo"))
return d
def test_config_star_workaround(self):
'''
ensure we ignore config/* for now
'''
self.protocol.answers.append('''info/names=
config/* a documentation string
''')
info = TorInfo(self.protocol)
self.assertTrue(dir(info) == [])
def test_accessors_not_setup(self):
info = TorInfo(self.protocol)
self.assertTrue(info.__dict__['_setup'] is False)
self.assertRaises(TypeError, len, info)
dir(info)
try:
info[0]
self.fail("Should have raised TypeError")
except TypeError:
pass
def test_str(self):
'''rather silly test to cover string creation'''
self.protocol.answers.append('''info/names=
version docstring
foo/* bar
''')
info = TorInfo(self.protocol)
## not the end of the world if this fails
self.assertTrue(str(info.version) == "version()")
self.assertTrue(str(info.foo) == "foo(arg)")
def test_with_arg_error(self):
self.protocol.answers.append('''info/names=
multi/no-arg docstring
''')
info = TorInfo(self.protocol)
try:
info.multi.no_arg('an argument')
self.assertTrue(False)
except TypeError:
pass
def test_member_access(self):
self.protocol.answers.append('info/names blam a thinkg\r\n')
info = TorInfo(self.protocol)
from txtorcon import torinfo
c = torinfo.MagicContainer(None)
c._setup = True
self.assertEqual([], c.__members__)
self.assertEqual(['info'], info.__members__)
# make sure __magic__ attr access doesn't throw
c.__class__
self.assertRaises(AttributeError, lambda: c.foo_mc_bar_bar)
def test_with_arg(self):
info = TorInfo(self.protocol)
pb = info.post_bootstrap
# now we hook up the protocol like it connected to a real Tor
self.protocol.makeConnection(self.transport)
# answer all the requests generated by TorControlProtocol
# boostrapping etc.
self.send(b'250-AUTH METHODS=HASHEDPASSWORD')
self.send(b'250 OK')
# response to AUTHENTICATE
self.send(b'250 OK')
# now we're in _bootstrap() in TorControlProtocol()
self.send(b"250-signal/names=")
self.send(b"250 OK")
self.send(b"250-version=foo")
self.send(b"250 OK")
self.send(b"250-events/names=")
self.send(b"250 OK")
self.send(b"250 OK") # for USEFEATURE
# do the TorInfo magic
self.send(b'250-info/names=')
self.send(b'250-multi/path/arg/* a documentation string')
self.send(b'250 OK')
# we had to save this up above due to the "interesting" way
# TorInfo switches to become a possible-nice magic thingy
# that does attribute-access for you.
yield pb
self.assertTrue(hasattr(info, 'multi'))
self.assertTrue(hasattr(getattr(info, 'multi'), 'path'))
self.assertTrue(hasattr(getattr(getattr(info, 'multi'), 'path'),
'arg'))
# Finally! The test! We see if we can get this multi-path
# value with an argument...
# a "real" tor example is "net/listeners/socks" which shows
# up in info/names as "net/listeners/*"
d = info.multi.path.arg('quux')
d.addCallback(CheckAnswer(self, 'foo'))
self.send(b"250-multi/path/arg/quux=foo")
self.send(b"250 OK")
yield d
def test_attribute_access(self):
'''
test that our post-setup TorInfo pretends to only have
attributes that correspond to (valid) GETINFO calls.
'''
self.protocol.answers.append('''info/names=
something/one a documentation string
something/two a second documentation string
''')
info = TorInfo(self.protocol)
yield self.protocol.post_bootstrap
self.assertTrue('something' in dir(info))
self.assertTrue(dir(info.something) == ['one', 'two'] or
dir(info.something) == ['two', 'one'])
def test_same_prefix(self):
self.protocol.answers.append('''info/names=
something/one a documentation string
something/two a second documentation string
''')
info = TorInfo(self.protocol)
self.assertTrue(hasattr(info, 'something'))
self.assertTrue(hasattr(info.something, 'one'))
self.assertTrue(hasattr(info.something, 'two'))
self.protocol.answers.append('something/two=bar\nOK')
d = info.something.two()
d.addCallback(CheckAnswer(self, 'bar'))
return d
def test_with_arg(self):
self.protocol.answers.append('''info/names=
multi/path/arg/* a documentation string
''')
info = TorInfo(self.protocol)
self.assertTrue(hasattr(info, 'multi'))
self.assertTrue(hasattr(getattr(info, 'multi'), 'path'))
self.assertTrue(hasattr(getattr(getattr(info, 'multi'), 'path'),
'arg'))
self.protocol.answers.append(
'multi/path/arg/quux=\nbar\nbaz\nquux\nOK')
try:
info.multi.path.arg()
self.assertTrue(False)
except TypeError:
pass
d = info.multi.path.arg('quux')
d.addCallback(CheckAnswer(self, 'bar\nbaz\nquux'))
return d
def test_iterator_access(self):
'''
confirm we can use the iterator protocol
'''
self.protocol.answers.append('''info/names=
something/one a documentation string
something/two a second documentation string
''')
info = TorInfo(self.protocol)
self.assertTrue(len(info) == 1)
all = []
for x in info:
all.append(x)
self.assertTrue(len(all) == 1)
self.assertTrue(len(info.something) == 2)
all = []
for x in info.something:
all.append(x)
self.assertTrue(len(all) == 2)
def test_with_arg(self):
self.protocol.answers.append('''info/names=
multi/path/arg/* a documentation string
''')
info = TorInfo(self.protocol)
self.assertTrue(hasattr(info, 'multi'))
self.assertTrue(hasattr(getattr(info, 'multi'), 'path'))
self.assertTrue(hasattr(getattr(getattr(info, 'multi'), 'path'), 'arg'))
## FIXME should have a test that "really" goes out through TorControlProtocol instance for this stuff...
## TorControlProtocol now strips the OK line...
self.protocol.answers.append('multi/path/arg/quux=\nbar\nbaz\nquux')
try:
info.multi.path.arg()
self.assertTrue(False)
except TypeError:
pass
d = info.multi.path.arg('quux')
d.addCallback(CheckAnswer(self, 'bar\nbaz\nquux'))
return d
def bootstrap(c):
info = TorInfo(c)
info.post_bootstrap.addCallback(setup_complete).addErrback(setup_failed)
def test_other_bootstrap(self):
self.protocol.answers.append('''info/names=
multi/path/arg/* a documentation string
''')
self.protocol.post_bootstrap = None
TorInfo(self.protocol)
def test_dump(self):
self.protocol.answers.append('''info/names=
multi/path/arg/* a documentation string
''')
info = TorInfo(self.protocol)
info.dump()
