Revision a55474c7 test/ganeti.utils_unittest.py

b/test/ganeti.utils_unittest.py
1085 1085
      self.failUnlessEqual(UnescapeAndSplit(sep.join(a), sep=sep), b)
1086 1086

  
1087 1087

  
1088
class TestGenerateSelfSignedSslCert(unittest.TestCase):
1089
  def setUp(self):
1090
    self.tmpdir = tempfile.mkdtemp()
1091

  
1092
  def tearDown(self):
1093
    shutil.rmtree(self.tmpdir)
1094

  
1095
  def _checkPrivateRsaKey(self, key):
1096
    lines = key.splitlines()
1097
    self.assert_("-----BEGIN RSA PRIVATE KEY-----" in lines)
1098
    self.assert_("-----END RSA PRIVATE KEY-----" in lines)
1099

  
1100
  def _checkRsaCertificate(self, cert):
1101
    lines = cert.splitlines()
1102
    self.assert_("-----BEGIN CERTIFICATE-----" in lines)
1103
    self.assert_("-----END CERTIFICATE-----" in lines)
1104

  
1105
  def testSingleFile(self):
1106
    cert1_filename = os.path.join(self.tmpdir, "cert1.pem")
1107

  
1108
    utils.GenerateSelfSignedSslCert(cert1_filename, validity=1)
1109

  
1110
    cert1 = utils.ReadFile(cert1_filename)
1111

  
1112
    self._checkPrivateRsaKey(cert1)
1113
    self._checkRsaCertificate(cert1)
1114

  
1115

  
1088 1116
if __name__ == '__main__':
1089 1117
  testutils.GanetiTestProgram()

Also available in: Unified diff