mirror of
https://github.com/freedombox/FreedomBox.git
synced 2026-03-11 09:04:54 +00:00
0cacb21983
Signed-off-by: Sunil Mohan Adapa <sunil@medhas.org> Reviewed-by: Joseph Nuthalapati <njoseph@thoughtworks.com>
119 lines
3.9 KiB
Python
119 lines
3.9 KiB
Python
#
|
|
# This file is part of FreedomBox.
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
"""
|
|
Tests for names module.
|
|
"""
|
|
|
|
import pytest
|
|
|
|
from .. import domain_types, domains
|
|
from .. import on_domain_added, on_domain_removed
|
|
from .. import get_domain_types, get_description
|
|
from .. import get_domain, get_enabled_services, get_services_status
|
|
|
|
|
|
@pytest.fixture(name='clean_domains')
|
|
def fixture_clean_domains():
|
|
"""Test fixture to start a test with clean domains list."""
|
|
old_domain_types = dict(domain_types)
|
|
old_domains = dict(domains)
|
|
domain_types.clear()
|
|
domains.clear()
|
|
yield
|
|
domain_types.clear()
|
|
domains.clear()
|
|
domain_types.update(old_domain_types)
|
|
domains.update(old_domains)
|
|
|
|
|
|
@pytest.mark.usefixtures('clean_domains')
|
|
def test_on_domain_added():
|
|
"""Test adding a domain to the global list."""
|
|
on_domain_added('', '')
|
|
assert '' not in domain_types
|
|
assert '' not in domains
|
|
|
|
on_domain_added('', 'hiddenservice', 'ddddd.onion')
|
|
on_domain_added('', 'hiddenservice', 'eeeee.onion')
|
|
assert 'ddddd.onion' in domains['hiddenservice']
|
|
assert 'eeeee.onion' in domains['hiddenservice']
|
|
|
|
|
|
@pytest.mark.usefixtures('clean_domains')
|
|
def test_on_domain_removed():
|
|
"""Test removing a domain from the global list."""
|
|
on_domain_added('', 'domainname', 'fffff')
|
|
on_domain_removed('', 'domainname', 'fffff')
|
|
assert 'fffff' not in domains['domainname']
|
|
|
|
on_domain_added('', 'pagekite', 'ggggg.pagekite.me')
|
|
on_domain_added('', 'pagekite', 'hhhhh.pagekite.me')
|
|
on_domain_removed('', 'pagekite')
|
|
assert 'ggggg.pagekite.me' not in domains['pagekite']
|
|
assert 'hhhhh.pagekite.me' not in domains['pagekite']
|
|
|
|
# try to remove things that don't exist
|
|
on_domain_removed('', '')
|
|
on_domain_removed('', 'domainname', 'iiiii')
|
|
|
|
|
|
@pytest.mark.usefixtures('clean_domains')
|
|
def test_get_domain_types():
|
|
"""Test getting domain types."""
|
|
on_domain_added('', 'domainname')
|
|
assert 'domainname' in get_domain_types()
|
|
|
|
|
|
@pytest.mark.usefixtures('clean_domains')
|
|
def test_get_description():
|
|
"""Test getting domain type description."""
|
|
on_domain_added('', 'pagekite', '', 'Pagekite')
|
|
assert get_description('pagekite') == 'Pagekite'
|
|
|
|
assert get_description('asdfasdf') == 'asdfasdf'
|
|
|
|
|
|
@pytest.mark.usefixtures('clean_domains')
|
|
def test_get_domain():
|
|
"""Test getting a domain of domain_type."""
|
|
on_domain_added('', 'hiddenservice', 'aaaaa.onion')
|
|
assert get_domain('hiddenservice') == 'aaaaa.onion'
|
|
|
|
assert get_domain('abcdef') is None
|
|
|
|
on_domain_removed('', 'hiddenservice')
|
|
assert get_domain('hiddenservice') is None
|
|
|
|
|
|
@pytest.mark.usefixtures('clean_domains')
|
|
def test_get_enabled_services():
|
|
"""Test getting enabled services for a domain."""
|
|
on_domain_added('', 'domainname', 'bbbbb', '', ['http', 'https', 'ssh'])
|
|
assert get_enabled_services('domainname',
|
|
'bbbbb') == ['http', 'https', 'ssh']
|
|
|
|
assert get_enabled_services('xxxxx', 'yyyyy') == []
|
|
assert get_enabled_services('domainname', 'zzzzz') == []
|
|
|
|
|
|
@pytest.mark.usefixtures('clean_domains')
|
|
def test_get_services_status():
|
|
"""Test getting whether each service is enabled for a domain."""
|
|
on_domain_added('', 'pagekite', 'ccccc.pagekite.me', '', ['http', 'https'])
|
|
assert get_services_status('pagekite', 'ccccc.pagekite.me') == \
|
|
[True, True, False]
|