mirror of
https://github.com/spalencsar/bearhub.git
synced 2026-07-06 21:44:16 +02:00
e95a23bc20
- Native runtime under bearhub/ (api, commons, gems, view/util, view/core, view/qt) - bauh/ compatibility re-export shims for legacy imports - Phase D: canonical __version__/__app_name__ in bearhub/__init__.py; pyproject.toml and setup.py point at bearhub package - Native gem loader at bearhub/view/core/gems.py - Namespace compatibility tests (159 tests passing) - Documentation: CHANGELOG [Unreleased], NAMESPACE_MIGRATION, ROADMAP, README, CONTRIBUTING, docs/qt6-migration.md, packaging/aur/README.md
92 lines
4.0 KiB
Python
92 lines
4.0 KiB
Python
from unittest import TestCase
|
|
|
|
from bearhub.commons.util import size_to_byte, sanitize_command_input
|
|
|
|
|
|
class SizeToByteTest(TestCase):
|
|
|
|
def test_must_return_right_number_of_bytes_for_bit_size(self):
|
|
self.assertEqual(0.125, size_to_byte(1, 'b'))
|
|
self.assertEqual(0.250, size_to_byte(2, 'b'))
|
|
self.assertEqual(0.40625, size_to_byte(3.25, 'b'))
|
|
|
|
def test_must_return_right_number_of_bytes_for_byte_based_units(self):
|
|
self.assertEqual(1.0, size_to_byte(1, 'B'))
|
|
self.assertEqual(587, size_to_byte(587, 'B'))
|
|
self.assertEqual(1000, size_to_byte(1, 'K'))
|
|
self.assertEqual(57300, size_to_byte(57.3, 'K'))
|
|
self.assertEqual(1000000, size_to_byte(1, 'M'))
|
|
self.assertEqual(57300000, size_to_byte(57.3, 'M'))
|
|
self.assertEqual(1000000000, size_to_byte(1, 'G'))
|
|
self.assertEqual(57300000000, size_to_byte(57.3, 'G'))
|
|
self.assertEqual(1000000000000, size_to_byte(1, 'T'))
|
|
self.assertEqual(57300000000000, size_to_byte(57.3, 'T'))
|
|
self.assertEqual(1000000000000000, size_to_byte(1, 'P'))
|
|
self.assertEqual(5670000000000000, size_to_byte(5.67, 'P'))
|
|
|
|
def test_must_return_right_number_of_bytes_for_bibyte_units(self):
|
|
self.assertEqual(1024.0, size_to_byte(1, 'KiB'))
|
|
self.assertEqual(1579683.84, size_to_byte(1542.66, 'KiB'))
|
|
self.assertEqual(1048576, size_to_byte(1, 'MiB'))
|
|
self.assertEqual(4372561.92, size_to_byte(4.17, 'MiB'))
|
|
self.assertEqual(1073741824, size_to_byte(1, 'GiB'))
|
|
self.assertEqual(2168958484.48, size_to_byte(2.02, 'GiB'))
|
|
self.assertEqual(1099511627776, size_to_byte(1, 'TiB'))
|
|
self.assertEqual(3375500697272.32, size_to_byte(3.07, 'TiB'))
|
|
self.assertEqual(1125899906842624, size_to_byte(1, 'PiB'))
|
|
self.assertEqual(1294784892869017.5, size_to_byte(1.15, 'PiB'))
|
|
|
|
def test_must_return_converted_string_sizes(self):
|
|
self.assertEqual(1000, size_to_byte('1', 'K'))
|
|
self.assertEqual(57300, size_to_byte('57.3', 'K'))
|
|
self.assertEqual(57300000, size_to_byte('57.3', 'M'))
|
|
self.assertEqual(57300000000, size_to_byte('57.3', 'G'))
|
|
self.assertEqual(57300000000000, size_to_byte('57.3', 'T'))
|
|
self.assertEqual(5670000000000000, size_to_byte('5.67', 'P'))
|
|
|
|
def test_must_treat_string_sizes_before_converting(self):
|
|
self.assertEqual(57300, size_to_byte(' 57 , 3 ', ' K '))
|
|
|
|
|
|
class SanitizeCommandInputTest(TestCase):
|
|
|
|
def test__must_remove_any_forbidden_symbols(self):
|
|
input_ = ' #$%* abc-def@ #%<<>> '
|
|
res = sanitize_command_input(input_)
|
|
self.assertEqual('abc-def@', res)
|
|
|
|
def test__must_remove_single_quotes(self):
|
|
input_ = " 'abc'-'xpto' "
|
|
res = sanitize_command_input(input_)
|
|
self.assertEqual('abc-xpto', res)
|
|
|
|
def test__must_remove_double_quotes(self):
|
|
input_ = ' "abc"-"xpto" '
|
|
res = sanitize_command_input(input_)
|
|
self.assertEqual('abc-xpto', res)
|
|
|
|
def test__must_remove_the_pipe_operator(self):
|
|
input_ = ' abc | ls /home/xpto | cat /home/xpto/secret.txt'
|
|
res = sanitize_command_input(input_)
|
|
self.assertEqual('abc', res)
|
|
|
|
def test__must_remove_the_and_operator(self):
|
|
input_ = ' abc && ls /home/xpto & cat /home/xpto/secret.txt'
|
|
res = sanitize_command_input(input_)
|
|
self.assertEqual('abc', res)
|
|
|
|
def test__must_remove_several_operator(self):
|
|
input_ = ' abc | ls /home/xpto && cat /home/xpto/secret.txt'
|
|
res = sanitize_command_input(input_)
|
|
self.assertEqual('abc', res)
|
|
|
|
def test__must_remove_single_dash_parameters(self):
|
|
input_ = '-cat abc-def -user -system ghi--jkl -xpto '
|
|
res = sanitize_command_input(input_)
|
|
self.assertEqual('abc-def ghi--jkl', res)
|
|
|
|
def test__must_remove_double_dash_parameters(self):
|
|
input_ = '--cat abc-def --user -system ghi--jkl --xpto '
|
|
res = sanitize_command_input(input_)
|
|
self.assertEqual('abc-def ghi--jkl', res)
|