def testunit_sock_sendcmd_WHEN_receive_bad_reply_RETURNS_zero(self):
fake_sock = FakeSocketIO('not-a-number junk\n')
returned_value = check_collectd_mlab.sock_sendcmd(
fake_sock, 'GETVAL "whatever"')
self.assertEqual(returned_value, 0)
self.assertEqual(fake_sock.getvalue(), 'GETVAL "whatever"\n')
def testunit_sock_sendcmd_RETURNS_successfully(self):
fake_sock = FakeSocketIO('1 default reply\n')
returned_value = check_collectd_mlab.sock_sendcmd(
fake_sock, 'GETVAL "whatever"')
self.assertEqual(returned_value, 1)
self.assertEqual(fake_sock.getvalue(), 'GETVAL "whatever"\n')
