class HttpSuccess(unittest.TestCase):
def setUp(self):
self.mock = Mock()
self.one = Connection(self.mock, '*****@*****.**', 'password',
'client')
self.mock.getresponse.return_value.status = 200
self.mock.getresponse.return_value.read.return_value = '<success />'
def test_command(self):
data = self.one.command('document.info', id=1)
self.assertEqual(etree.tostring(data), '<success />')
def test_command_error(self):
self.mock.getresponse.return_value.status = 200
self.mock.getresponse.return_value.read.return_value = '<error />'
self.assertRaises(CommandError, self.one.command, 'command', id=1)
