82 lines
2.4 KiB
Python
82 lines
2.4 KiB
Python
# DRF Rack
|
|
# Библиотека классов расширений для Django REST Framework
|
|
#
|
|
# Автор: Николай В. Анохин <n.anokhin@xeaf.net>
|
|
# Все права защищены. Лицензия: MIT
|
|
|
|
"""
|
|
Описание класса CryptoTests
|
|
"""
|
|
|
|
from net.xeaf.rack.core import CoreTestCase
|
|
from net.xeaf.rack.util import Crypto
|
|
|
|
|
|
class CryptoTests(CoreTestCase):
|
|
"""
|
|
Тесты для Crypto
|
|
"""
|
|
|
|
def test_gen_token(self):
|
|
"""
|
|
Тест генерации токена
|
|
"""
|
|
|
|
token = Crypto.gen_token()
|
|
self.assertEqual(len(token), Crypto.DEFAULT_TOKEN_LENGTH)
|
|
self.assertLowerHex(token)
|
|
|
|
def test_gen_token_length(self):
|
|
"""
|
|
Тест генерации токена с указанием длины
|
|
"""
|
|
|
|
token = Crypto.gen_token(length=15)
|
|
self.assertEqual(len(token), 15)
|
|
|
|
def test_gen_token_different(self):
|
|
"""
|
|
Тест генерации разных токенов
|
|
"""
|
|
|
|
token1 = Crypto.gen_token()
|
|
token2 = Crypto.gen_token()
|
|
self.assertNotEqual(token1, token2)
|
|
|
|
def test_gen_pin_code(self):
|
|
"""
|
|
Тест генерации PIN-кода
|
|
"""
|
|
|
|
pin_code = Crypto.gen_pin_code()
|
|
self.assertEqual(len(pin_code), Crypto.DEFAULT_PIN_CODE_LENGTH)
|
|
self.assertTrue(all(c in Crypto.DEFAULT_PIN_CODE_ALPHABET for c in pin_code))
|
|
|
|
def test_gen_pin_code_length(self):
|
|
"""
|
|
Тест генерации PIN-кода с указанием длины
|
|
"""
|
|
|
|
pin_code = Crypto.gen_pin_code(length=8)
|
|
self.assertEqual(len(pin_code), 8)
|
|
self.assertTrue(all(c in Crypto.DEFAULT_PIN_CODE_ALPHABET for c in pin_code))
|
|
|
|
def test_gen_pin_code_alphabet(self):
|
|
"""
|
|
Тест генерации PIN-кода с указанием алфавита
|
|
"""
|
|
|
|
letters = 'ABCD'
|
|
pin_code = Crypto.gen_pin_code(alphabet=letters)
|
|
self.assertEqual(len(pin_code), Crypto.DEFAULT_PIN_CODE_LENGTH)
|
|
self.assertTrue(all(c in letters for c in pin_code))
|
|
|
|
def test_gen_pin_code_different(self):
|
|
"""
|
|
Тест генерации разных PIN-кодов
|
|
"""
|
|
|
|
pin_code_1 = Crypto.gen_pin_code()
|
|
pin_code_2 = Crypto.gen_pin_code()
|
|
self.assertNotEqual(pin_code_1, pin_code_2)
|