def test_parse_command(self): command = Command() command.parse(b'User [email protected]\r\n') self.assertEqual('USER', command.name) self.assertEqual('*****@*****.**', command.argument) self.assertRaises(AssertionError, command.parse, b'OOPS\r\n') command = Command() command.parse(b'POKE\r\n') self.assertEqual('POKE', command.name) self.assertEqual('', command.argument) self.assertRaises(AssertionError, command.parse, b'OOPS\r\n')