mirror of
https://github.com/mozilla/cipherscan.git
synced 2024-12-27 21:23:41 +01:00
46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
|
# Copyright (c) 2015 Hubert Kario
|
||
|
|
# Released under Mozilla Public License Version 2.0
|
||
|
|
|
||
|
|
try:
|
||
|
|
import unittest2 as unittest
|
||
|
|
except ImportError:
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from cscan.config import HugeCipherList, Firefox_42
|
||
|
|
from cscan.modifiers import truncate_ciphers_to_size, append_ciphers_to_size, \
|
||
|
|
extend_with_ext_to_size
|
||
|
|
|
||
|
|
class TestTruncateCiphersToSize(unittest.TestCase):
|
||
|
|
def test_with_big_hello(self):
|
||
|
|
gen = HugeCipherList()
|
||
|
|
|
||
|
|
self.assertGreater(len(gen(b'localhost').write()), 2**14)
|
||
|
|
self.assertEqual(gen(b'localhost').cipher_suites[0], 49196)
|
||
|
|
|
||
|
|
gen = truncate_ciphers_to_size(gen, 2**12)
|
||
|
|
|
||
|
|
self.assertEqual(len(gen(b'localhost').write()), 2**12-1)
|
||
|
|
self.assertEqual(gen(b'localhost').cipher_suites[0], 49196)
|
||
|
|
|
||
|
|
class TestAppendCiphersToSize(unittest.TestCase):
|
||
|
|
def test_with_small_hello(self):
|
||
|
|
gen = Firefox_42()
|
||
|
|
|
||
|
|
self.assertLess(len(gen(b'localhost').write()), 2**10)
|
||
|
|
self.assertEqual(gen(b'localhost').cipher_suites[0], 49195)
|
||
|
|
|
||
|
|
gen = append_ciphers_to_size(gen, 2**12)
|
||
|
|
|
||
|
|
self.assertEqual(len(gen(b'localhost').write()), 2**12)
|
||
|
|
self.assertEqual(gen(b'localhost').cipher_suites[0], 49195)
|
||
|
|
|
||
|
|
class TestExtendWithExtToSize(unittest.TestCase):
|
||
|
|
def test_with_small_hello(self):
|
||
|
|
gen = Firefox_42()
|
||
|
|
|
||
|
|
self.assertLess(len(gen(b'localhost').write()), 2**10)
|
||
|
|
|
||
|
|
gen = extend_with_ext_to_size(gen, 2**12)
|
||
|
|
|
||
|
|
self.assertEqual(len(gen(b'localhost').write()), 2**12)
|