33 lines
878 B
Python
33 lines
878 B
Python
# -*- coding: utf-8 -*-
|
|
import unittest
|
|
|
|
import mock
|
|
|
|
from ..managers import SQRLNutManager
|
|
from ..models import SQRLNut
|
|
|
|
|
|
TESTING_MODULE = 'sqrl.models'
|
|
|
|
|
|
class TestSQRLNut(unittest.TestCase):
|
|
def test_objects(self):
|
|
self.assertIsInstance(SQRLNut.objects, SQRLNutManager)
|
|
|
|
def test_str(self):
|
|
self.assertEqual(
|
|
str(SQRLNut(nonce='nonce')),
|
|
'nonce'
|
|
)
|
|
|
|
@mock.patch(TESTING_MODULE + '.generate_randomness')
|
|
@mock.patch.object(SQRLNut, 'delete')
|
|
@mock.patch.object(SQRLNut, 'save')
|
|
def test_renew(self, mock_save, mock_delete, mock_generate_randomness):
|
|
nut = SQRLNut(nonce='nonce')
|
|
|
|
self.assertIsNone(nut.renew())
|
|
self.assertEqual(nut.nonce, mock_generate_randomness.return_value)
|
|
mock_save.assert_called_once_with()
|
|
mock_generate_randomness.assert_called_once_with()
|