Commit fcb5ce41 authored by Michael Tremer's avatar Michael Tremer

tests: Add a simple test that lists all networks

Signed-off-by: default avatarMichael Tremer <michael.tremer@ipfire.org>
parent 12ac5c5b
...@@ -320,6 +320,9 @@ TESTS_LDADD = \ ...@@ -320,6 +320,9 @@ TESTS_LDADD = \
src/libloc.la \ src/libloc.la \
src/libloc-internal.la src/libloc-internal.la
TESTS_ENVIRONMENT = \
TEST_DATA_DIR="$(abs_top_srcdir)/tests/data"
TESTS = \ TESTS = \
src/test-libloc \ src/test-libloc \
src/test-stringpool \ src/test-stringpool \
...@@ -328,7 +331,11 @@ TESTS = \ ...@@ -328,7 +331,11 @@ TESTS = \
src/test-network \ src/test-network \
src/test-country \ src/test-country \
src/test-signature \ src/test-signature \
src/test-address src/test-address \
tests/python/test-export.py
EXTRA_DIST += \
tests/data/location-2022-03-30.db
CLEANFILES += \ CLEANFILES += \
testdata.db testdata.db
......
#!/usr/bin/python3
###############################################################################
# #
# libloc - A library to determine the location of someone on the Internet #
# #
# Copyright (C) 2022 IPFire Development Team <info@ipfire.org> #
# #
# This library is free software; you can redistribute it and/or #
# modify it under the terms of the GNU Lesser General Public #
# License as published by the Free Software Foundation; either #
# version 2.1 of the License, or (at your option) any later version. #
# #
# This library 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 #
# Lesser General Public License for more details. #
# #
###############################################################################
import location
import os
import unittest
TEST_DATA_DIR = os.environ["TEST_DATA_DIR"]
class Test(unittest.TestCase):
def setUp(self):
path = os.path.join(TEST_DATA_DIR, "location-2022-03-30.db")
# Load the database
self.db = location.Database(path)
def test_list_networks(self):
"""
Lists all available networks
"""
for network in self.db.networks:
print(network)
def test_list_networks_flattened(self):
"""
Lists all networks but flattened
"""
for network in self.db.networks_flattened:
print(network)
if __name__ == "__main__":
unittest.main()
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment