mirror of
https://github.com/onionshare/onionshare.git
synced 2024-12-27 00:09:50 -05:00
124 lines
3.8 KiB
Python
124 lines
3.8 KiB
Python
from onionshare import *
|
|
import tempfile
|
|
|
|
def test_get_platform_on_tails():
|
|
"get_platform() returns 'Tails' when hostname is 'amnesia'"
|
|
onionshare.platform.uname = lambda: ('Linux', 'amnesia', '3.14-1-amd64', '#1 SMP Debian 3.14.4-1 (2014-05-13)', 'x86_64', '')
|
|
assert get_platform() == 'Tails'
|
|
|
|
def test_get_platform_returns_platform_system():
|
|
"get_platform() returns platform.system() when ONIONSHARE_PLATFORM is not defined"
|
|
onionshare.platform.system = lambda: 'Sega Saturn'
|
|
assert get_platform() == 'Sega Saturn'
|
|
|
|
class MockSubprocess():
|
|
def __init__(self):
|
|
self.last_call = None
|
|
|
|
def call(self, args):
|
|
self.last_call = args
|
|
|
|
def last_call_args(self):
|
|
return self.last_call
|
|
|
|
def test_tails_open_port():
|
|
"tails_open_port() calls iptables with ACCEPT arg"
|
|
onionshare.get_platform = lambda: 'Tails'
|
|
onionshare.strings = {'punching_a_hole': ''}
|
|
|
|
mock_subprocess = MockSubprocess()
|
|
onionshare.subprocess = mock_subprocess
|
|
onionshare.tails_open_port('port')
|
|
|
|
expected_call = [
|
|
'/sbin/iptables', '-I', 'OUTPUT',
|
|
'-o', 'lo', '-p',
|
|
'tcp', '--dport', 'port', '-j', 'ACCEPT'
|
|
]
|
|
actual_call = mock_subprocess.last_call_args()
|
|
assert actual_call == expected_call
|
|
|
|
def test_tails_close_port():
|
|
"tails_close_port() calls iptables with REJECT arg"
|
|
onionshare.get_platform = lambda: 'Tails'
|
|
onionshare.strings = {'closing_hole': ''}
|
|
|
|
mock_subprocess = MockSubprocess()
|
|
onionshare.subprocess = mock_subprocess
|
|
onionshare.tails_close_port('port')
|
|
|
|
expected_call = [
|
|
'/sbin/iptables', '-I', 'OUTPUT',
|
|
'-o', 'lo', '-p',
|
|
'tcp', '--dport', 'port', '-j', 'REJECT'
|
|
]
|
|
actual_call = mock_subprocess.last_call_args()
|
|
assert actual_call == expected_call
|
|
|
|
def test_load_strings_defaults_to_english():
|
|
"load_strings() loads English by default"
|
|
locale.getdefaultlocale = lambda: ('en_US', 'UTF-8')
|
|
load_strings()
|
|
assert onionshare.strings['calculating_sha1'] == "Calculating SHA1 checksum."
|
|
|
|
def test_load_strings_loads_other_languages():
|
|
"load_strings() loads other languages in different locales"
|
|
locale.getdefaultlocale = lambda: ('fr_FR', 'UTF-8')
|
|
load_strings("fr")
|
|
assert onionshare.strings['calculating_sha1'] == "Calculer un hachage SHA-1."
|
|
|
|
def test_generate_slug_length():
|
|
"generates a 26-character slug"
|
|
assert len(slug) == 26
|
|
|
|
def test_generate_slug_characters():
|
|
"generates a base32-encoded slug"
|
|
|
|
def is_b32(string):
|
|
b32_alphabet = "01234556789abcdefghijklmnopqrstuvwxyz"
|
|
return all(char in b32_alphabet for char in string)
|
|
|
|
assert is_b32(slug)
|
|
|
|
def test_starts_with_empty_strings():
|
|
"creates an empty strings dict by default"
|
|
assert strings == {}
|
|
|
|
def test_choose_port_returns_a_port_number():
|
|
"choose_port() returns a port number"
|
|
assert 1024 <= choose_port() <= 65535
|
|
|
|
def test_choose_port_returns_an_open_port():
|
|
"choose_port() returns an open port"
|
|
port = choose_port()
|
|
socket.socket().bind(("127.0.0.1", port))
|
|
|
|
def write_tempfile(text):
|
|
tempdir = tempfile.mkdtemp()
|
|
path = tempdir + "/test-file.txt"
|
|
with open(path, "w") as f:
|
|
f.write(text)
|
|
f.close()
|
|
return path
|
|
|
|
def test_filehash_returns_correct_hash():
|
|
"file_crunching() returns correct hash"
|
|
|
|
text = """
|
|
If you want a picture of the future, imagine a boot stamping on an
|
|
encrypted, redundant, distributed filesystem -- forever.
|
|
"""
|
|
tempfile = write_tempfile(text)
|
|
filehash, _ = file_crunching(tempfile)
|
|
|
|
assert filehash == 'bc004fe72e6530a545570b4c6ce76bcb78ea526b'
|
|
|
|
def test_filehash_returns_correct_size():
|
|
"file_crunching() returns correct size"
|
|
|
|
text = "AUSCANNZUKUS has always been at war with Eastasia."
|
|
tempfile = write_tempfile(text)
|
|
_, filesize = file_crunching(tempfile)
|
|
|
|
assert filesize == 50
|