class DocumentTestCase(unittest.TestCase):
def setUp(self):
self.gpg = GPG()
self.gpg.import_keys(PUB_KEY)
def testGetCertificateID(self):
"Test result value of document ID"
source = DocumentSourceMock((200, CORRECT_CERTIFICATE_WITH_VALID_SIGNATURE))
document = Document(source)
self.assertTrue(document.verify())
self.assertEquals(document.get_id(), '64D0 F8CA 822B BDF0 FF59 42F6 A363 ED80 D8A5 CEFF')
def testGetSignature(self):
"Test result value of document signature"
source = DocumentSourceMock((200, CORRECT_CERTIFICATE_WITH_VALID_SIGNATURE))
document = Document(source)
self.assertTrue(document.verify())
self.assertEquals(document.get_signature(), VALID_SIGNATURE)
def testGetBody(self):
"Test result value of document body"
source = DocumentSourceMock((200, CORRECT_CERTIFICATE_WITH_VALID_SIGNATURE))
document = Document(source)
self.assertTrue(document.verify())
self.assertEquals(document.get_body(), VALID_BODY)
def testGetRawData(self):
"Test result value of document raw data"
source = DocumentSourceMock((200, CORRECT_CERTIFICATE_WITH_VALID_SIGNATURE))
document = Document(source)
self.assertTrue(document.verify())
self.assertEquals(document.get_raw_data(), CORRECT_CERTIFICATE_WITH_VALID_SIGNATURE)
class DocumentVerificationTestCase(unittest.TestCase):
def setUp(self):
self.gpg = GPG()
self.gpg.import_keys(PUB_KEY)
def testCorrectCertificateWithValidSignature(self):
"Test correct certificate with valid signature"
source = DocumentSourceMock((200, CORRECT_CERTIFICATE_WITH_VALID_SIGNATURE))
document = Document(source)
self.assertTrue(document.verify())
def testCorrectCertificateWithInvalidSignature(self):
"Test correct certificate with invalid signature"
source = DocumentSourceMock((200, CORRECT_CERTIFICATE_WITH_INVALID_SIGNATURE))
document = Document(source)
self.assertFalse(document.verify())
def testIncorrectCertificateWithValidSignature(self):
"Test incorrect certificate with valid signature"
source = DocumentSourceMock((200, INCORRECT_CERTIFICATE_WITH_VALID_SIGNATURE))
document = Document(source)
self.assertTrue(document.verify())