boards/tests/test_keys.py

import logging

from django.test import TestCase
from boards.models import KeyPair, GlobalId, Post


logger = logging.getLogger(__name__)


class KeyTest(TestCase):
    def test_create_key(self):
        key = KeyPair.objects.generate_key('ecdsa')

        self.assertIsNotNone(key, 'The key was not created.')

    def test_validation(self):
        key = KeyPair.objects.generate_key(key_type='ecdsa')
        message = 'msg'
        signature = key.sign(message)
        valid = KeyPair.objects.verify(key.public_key, message, signature,
                                       key_type='ecdsa')

        self.assertTrue(valid, 'Message verification failed.')

    def test_primary_constraint(self):
        KeyPair.objects.generate_key(key_type='ecdsa', primary=True)

        try:
            KeyPair.objects.generate_key(key_type='ecdsa', primary=True)
            self.fail('Exception should be thrown indicating there can be only'
                      ' one primary key.')
        except Exception:
            pass

    def test_model_id_save(self):
        model_id = GlobalId(key_type='test', key='test key', local_id='1')
        model_id.save()

    def test_request_get(self):
        post = self._create_post_with_key()

        request = Post.objects.generate_request_get([post])
        logger.debug(request)

        self.assertTrue('<request type="get" version="1.0">'
                        '<model name="post" version="1.0">'
                        '<id key="pubkey" local-id="1" type="test_key_type" />'
                        '</model>'
                        '</request>' in request,
                        'Wrong XML generated for the GET request.')

    def test_response_get(self):
        post = self._create_post_with_key()
        reply_post = Post.objects.create_post(title='test_title',
                text='[post]%d[/post]' % post.id, thread=post.get_thread())
        reply_reply_post = Post.objects.create_post(title='',
                text='[post]%d[/post]' % reply_post.id,
                thread=post.get_thread())

        response = Post.objects.generate_response_get([reply_post])
        logger.debug(response)

        self.assertTrue('<response>'
                        '<status>success</status>'
                        '<models>'
                        '<model name="post" ref-id="1">'
                        '<id key="pubkey" local-id="%d" type="test_key_type" />'
                        '<title>test_title</title>'
                        '<text>[post]%d[/post]</text>'
                        '<thread>%d</thread>'
                        '<pub-time>%s</pub-time>'
                        '<edit-time>%s</edit-time>'
                        '<previous>'
                        '<id key="pubkey" local-id="%d" type="test_key_type" />'
                        '</previous>'
                        '<next>'
                        '<id key="pubkey" local-id="%d" type="test_key_type" />'
                        '</next>'
                        '</model>'
                        '</models>'
                        '</response>' % (
                            reply_post.id,
                            post.id,
                            post.id,
                            str(reply_post.get_edit_time_epoch()),
                            str(reply_post.get_pub_time_epoch()),
                            post.id,
                            reply_reply_post.id,
                        ) in response,
                        'Wrong XML generated for the GET response.')

    def _create_post_with_key(self):
        key = KeyPair(public_key='pubkey', private_key='privkey',
                      key_type='test_key_type', primary=True)
        key.save()

        return Post.objects.create_post(title='test_title', text='test_text')