Source code for mailman_pgp.rest.tests.test_addresses
# Copyright (C) 2017 Jan Jancar
#
# This file is a part of the Mailman PGP plugin.
#
# This program is free software; you can redistribute it and/or modify it under
# the terms of the GNU General Public License as published by the Free
# Software Foundation, either version 3 of the License, or (at your option)
# any later version.
#
# This program is distributed in the hope that it will be useful, but WITHOUT
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
# more details.
#
# You should have received a copy of the GNU General Public License along with
# this program. If not, see <http://www.gnu.org/licenses/>.
""""""
import unittest
from urllib.error import HTTPError
from mailman.interfaces.usermanager import IUserManager
from mailman.testing.helpers import call_api
from zope.component import getUtility
from mailman_pgp.database import mm_transaction, transaction
from mailman_pgp.model.address import PGPAddress
from mailman_pgp.testing.layers import PGPRESTLayer
from mailman_pgp.testing.pgp import load_key
from mailman_pgp.utils.pgp import key_from_blob
[docs]class TestAddresses(unittest.TestCase):
layer = PGPRESTLayer
[docs] def setUp(self):
with mm_transaction():
self.mm_address = getUtility(IUserManager).create_address(
'anne@example.com')
self.address_key = load_key('ecc_p256.pub.asc')
with transaction() as t:
self.address = PGPAddress(self.mm_address)
self.address.key = self.address_key
self.address.key_confirmed = True
t.add(self.address)
[docs] def test_missing_address(self):
with self.assertRaises(HTTPError) as cm:
call_api('http://localhost:9001/3.1/plugins/pgp/addresses/'
'bart@example.com')
self.assertEqual(cm.exception.code, 404)
[docs] def test_all_addresses(self):
json, response = call_api(
'http://localhost:9001/3.1/plugins/pgp/addresses/')
self.assertEqual(json['total_size'], 1)
self.assertEqual(len(json['entries']), 1)
addresses = json['entries']
address = addresses[0]
self.assertEqual(address['email'], self.address.email)
[docs] def test_get_address(self):
json, response = call_api(
'http://localhost:9001/3.1/plugins/pgp/addresses/'
'anne@example.com')
self.assertEqual(json['email'], self.address.email)
[docs] def test_address_key(self):
json, response = call_api(
'http://localhost:9001/3.1/plugins/pgp/addresses/'
'anne@example.com/key')
key = key_from_blob(json['key'])
self.assertEqual(key.fingerprint, self.address_key.fingerprint)
self.assertEqual(json['key_fingerprint'], self.address_key.fingerprint)
self.assertEqual(json['key_confirmed'], True)
[docs] def test_address_no_key(self):
with mm_transaction():
mm_address = getUtility(IUserManager).create_address(
'bart@example.com')
with transaction() as t:
address = PGPAddress(mm_address)
t.add(address)
with self.assertRaises(HTTPError) as cm:
call_api('http://localhost:9001/3.1/plugins/pgp/addresses/'
'bart@example.com/key')
self.assertEqual(cm.exception.code, 404)