def test_random_alphanumeric512(self): """ Send a payload of 512 random characters. """ payload = ''.join(choice(ascii_letters) for x in range(512)) rawsend(payload) self.assertTrue(puck())
def test_lotsofzeros512k(self): """ Send a payload of 512 KB of zeros. """ payload = '\x00' * 1024 * 512 rawsend(payload) self.assertTrue(puck())
def test_lotsofzeros512(self): """ Send a payload of 512 zeros. """ payload = '\x00' * 512 rawsend(payload) # Check that the server is still alive. self.assertTrue(puck())
def test_bad_port(self): cmd = "{:<8}".format("REGISTER") payload = "127.0.0.1::," l = len(payload) size = "{:<4}".format(l) response = rawsend(cmd + size + payload) self.assertEqual(response, "ERR_BAD_PORT")
def test_badpayload_mismatched_size(self): cmd = "{:<8}".format("BROKER") payload = "127.0.0.1:1811:SomeService" l = len(payload)*2 size = "{:<4}".format(l) response = rawsend(cmd + size + payload) self.assertEqual(response, "ERR_BAD_PAYLOAD")
def test_everything_correct(self): cmd = "{:<8}".format("BROKER") payload = "SomeService" l = len(payload) size = "{:<4}".format(l) response = rawsend(cmd + size + payload) self.assertEqual(response, "OK 0")
def test_badpayload_mismatched_size(self): cmd = "{:<8}".format("BROKER") payload = "127.0.0.1:1811:SomeService" l = len(payload) * 2 size = "{:<4}".format(l) response = rawsend(cmd + size + payload) self.assertEqual(response, "ERR_BAD_PAYLOAD")
def test_bad_host(self): cmd = "{:<8}".format("REGISTER") payload = "1:1811:SomeService," l = len(payload) size = "{:<4}".format(l) response = rawsend(cmd + size + payload) self.assertEqual(response, "ERR_BAD_HOST")
def test_bad_delim(self): cmd = "{:<8}".format("REGISTER") payload = "127:1811:SomeService" l = len(payload) size = "{:<4}".format(l) response = rawsend(cmd + size + payload) self.assertEqual(response, "ERR_BAD_DELIM")
def test_badpayload_negativesize(self): cmd = "{:<8}".format("BROKER") payload = "{:<1024}".format("BLAH BLAH") size = "{:<4}".format(-40) response = rawsend(cmd + size + payload) self.assertEqual(response, "ERR_BAD_PAYLOAD_SIZE")
def test_badpayload_missingsize(self): cmd = "{:<8}".format("BROKER") payload = "{:<1024}".format("BLAH BLAH") response = rawsend(cmd + payload) self.assertEqual(response, "ERR_BAD_PAYLOAD_SIZE")
def test_badpayload(self): cmd = "{:<8}".format("BROKER") response = rawsend(cmd) self.assertEqual(response, "ERR_BAD_PAYLOAD_SIZE")
def test_badcmd(self): payload = "{:<1024}".format("Hi there!") cmd = "{:<8}".format("HI") response = rawsend(cmd+payload) self.assertEqual(response, "ERR_BAD_CMD")
def test_badcmd_toolong(self): cmd = "BROKERBROKER" payload = "{:<1024}".format("Hi there!") response = rawsend(cmd + payload) self.assertEqual(response, "ERR_BAD_CMD")
def test_badcmd_toolong(self): cmd = "BROKERBROKER" payload = "{:<1024}".format("Hi there!") response = rawsend(cmd+payload) self.assertEqual(response, "ERR_BAD_CMD")
def test_badcmd(self): payload = "{:<1024}".format("Hi there!") cmd = "{:<8}".format("HI") response = rawsend(cmd + payload) self.assertEqual(response, "ERR_BAD_CMD")