def test_get_fingerprint_for_address(self): controller = Mock() self.assertRaises(ValueError, _get_fingerprint, '127.0.0.1:-1', controller) self.assertRaises(ValueError, _get_fingerprint, '127.0.0.901:80', controller) descriptor = Mock() descriptor.address = '127.0.0.1' descriptor.or_port = 80 descriptor.fingerprint = FINGERPRINT controller.get_network_statuses.return_value = [descriptor] self.assertEqual(FINGERPRINT, _get_fingerprint('127.0.0.1', controller)) self.assertEqual(FINGERPRINT, _get_fingerprint('127.0.0.1:80', controller)) self.assertRaises(ValueError, _get_fingerprint, '127.0.0.1:81', controller) self.assertRaises(ValueError, _get_fingerprint, '127.0.0.2', controller)
def test_get_fingerprint_for_ourselves(self): controller = Mock() controller.get_info.side_effect = lambda arg: { 'fingerprint': FINGERPRINT, }[arg] self.assertEqual(FINGERPRINT, _get_fingerprint('', controller)) controller.get_info.side_effect = stem.ControllerError self.assertRaises(ValueError, _get_fingerprint, '', controller)
def test_get_fingerprint_for_nickname(self): controller, descriptor = Mock(), Mock() descriptor.fingerprint = FINGERPRINT controller.get_network_status.side_effect = lambda arg: { 'moria1': descriptor, }[arg] self.assertEqual(FINGERPRINT, _get_fingerprint('moria1', controller)) controller.get_network_status.side_effect = stem.ControllerError self.assertRaises(ValueError, _get_fingerprint, 'moria1', controller)
def test_get_fingerprint_for_fingerprint(self): self.assertEqual(FINGERPRINT, _get_fingerprint(FINGERPRINT, Mock()))