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')