Revision 120e7e77 test/ganeti.rpc_unittest.py
b/test/ganeti.rpc_unittest.py | ||
---|---|---|
33 | 33 |
from ganeti import errors |
34 | 34 |
from ganeti import serializer |
35 | 35 |
from ganeti import objects |
36 |
from ganeti import backend |
|
36 | 37 |
|
37 | 38 |
import testutils |
38 | 39 |
|
... | ... | |
448 | 449 |
]) |
449 | 450 |
|
450 | 451 |
|
452 |
class TestCompress(unittest.TestCase): |
|
453 |
def test(self): |
|
454 |
for data in ["", "Hello", "Hello World!\nnew\nlines"]: |
|
455 |
self.assertEqual(rpc._Compress(data), |
|
456 |
(constants.RPC_ENCODING_NONE, data)) |
|
457 |
|
|
458 |
for data in [512 * " ", 5242 * "Hello World!\n"]: |
|
459 |
compressed = rpc._Compress(data) |
|
460 |
self.assertEqual(len(compressed), 2) |
|
461 |
self.assertEqual(backend._Decompress(compressed), data) |
|
462 |
|
|
463 |
def testDecompression(self): |
|
464 |
self.assertRaises(AssertionError, backend._Decompress, "") |
|
465 |
self.assertRaises(AssertionError, backend._Decompress, [""]) |
|
466 |
self.assertRaises(AssertionError, backend._Decompress, |
|
467 |
("unknown compression", "data")) |
|
468 |
self.assertRaises(Exception, backend._Decompress, |
|
469 |
(constants.RPC_ENCODING_ZLIB_BASE64, "invalid zlib data")) |
|
470 |
|
|
471 |
|
|
451 | 472 |
if __name__ == "__main__": |
452 | 473 |
testutils.GanetiTestProgram() |
Also available in: Unified diff