Merge pull request #41049 from Ch3LL/fix_wheel_test

fix integration wheel test_gen test
This commit is contained in:
Mike Place 2017-05-04 09:33:58 -06:00 committed by GitHub
commit ff39613a53

View File

@ -22,8 +22,13 @@ class KeyWheelModuleTest(integration.TestCase, integration.AdaptedConfigurationT
self.assertIn('pub', ret)
self.assertIn('priv', ret)
self.assertTrue(
ret.get('pub', '').startswith('-----BEGIN PUBLIC KEY-----'))
try:
self.assertTrue(
ret.get('pub', '').startswith('-----BEGIN PUBLIC KEY-----'))
except AssertionError:
self.assertTrue(
ret.get('pub', '').startswith('-----BEGIN RSA PUBLIC KEY-----'))
self.assertTrue(
ret.get('priv', '').startswith('-----BEGIN RSA PRIVATE KEY-----'))