Revision a6c43c02 test/py/ganeti.utils.security_unittest.py
| b/test/py/ganeti.utils.security_unittest.py | ||
|---|---|---|
| 22 | 22 |
"""Script for unittesting the ganeti.utils.storage module""" |
| 23 | 23 |
|
| 24 | 24 |
import mock |
| 25 |
import os |
|
| 26 |
import shutil |
|
| 27 |
import tempfile |
|
| 25 | 28 |
import unittest |
| 26 | 29 |
|
| 30 |
from ganeti import constants |
|
| 27 | 31 |
from ganeti.utils import security |
| 28 | 32 |
|
| 29 | 33 |
import testutils |
| ... | ... | |
| 88 | 92 |
self.assertFalse(digest1 == digest2) |
| 89 | 93 |
|
| 90 | 94 |
|
| 95 |
class TestCertVerification(testutils.GanetiTestCase): |
|
| 96 |
def setUp(self): |
|
| 97 |
testutils.GanetiTestCase.setUp(self) |
|
| 98 |
|
|
| 99 |
self.tmpdir = tempfile.mkdtemp() |
|
| 100 |
|
|
| 101 |
def tearDown(self): |
|
| 102 |
shutil.rmtree(self.tmpdir) |
|
| 103 |
|
|
| 104 |
def testVerifyCertificate(self): |
|
| 105 |
security.VerifyCertificate(testutils.TestDataFilename("cert1.pem"))
|
|
| 106 |
|
|
| 107 |
nonexist_filename = os.path.join(self.tmpdir, "does-not-exist") |
|
| 108 |
|
|
| 109 |
(errcode, msg) = security.VerifyCertificate(nonexist_filename) |
|
| 110 |
self.assertEqual(errcode, constants.CV_ERROR) |
|
| 111 |
|
|
| 112 |
# Try to load non-certificate file |
|
| 113 |
invalid_cert = testutils.TestDataFilename("bdev-net.txt")
|
|
| 114 |
(errcode, msg) = security.VerifyCertificate(invalid_cert) |
|
| 115 |
self.assertEqual(errcode, constants.CV_ERROR) |
|
| 116 |
|
|
| 117 |
|
|
| 91 | 118 |
if __name__ == "__main__": |
| 92 | 119 |
testutils.GanetiTestProgram() |
Also available in: Unified diff