-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
test: Add test cases and CI pipeline (#3)
Previously, there were no unittests or CI in this project. This adds test cases for every functionality of the semantic_id_resolver and its service. Furthermore, this adds the corresponding GitHub CI Workflows to run the tests. Fixes #2
- Loading branch information
1 parent
d8dd2a6
commit 17812f4
Showing
10 changed files
with
213 additions
and
13 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,35 @@ | ||
name: test | ||
|
||
on: | ||
push: | ||
branches: | ||
- wip/testing | ||
|
||
jobs: | ||
build: | ||
runs-on: ${{ matrix.os }} | ||
strategy: | ||
matrix: | ||
os: | ||
- ubuntu-latest | ||
- ubuntu-22.04 | ||
- ubuntu-20.04 | ||
- windows-latest | ||
- windows-2022 | ||
- windows-2019 | ||
|
||
steps: | ||
- name: Checkout code | ||
uses: actions/checkout@v4 | ||
|
||
- name: Set up Python | ||
uses: actions/setup-python@v4 | ||
with: | ||
python-version: "3.10" | ||
architecture: x64 | ||
|
||
- name: Install Python dependencies | ||
run: pip install -r requirements.txt | ||
|
||
- name: Run Python Tests | ||
run: python -m unittest discover |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -5,6 +5,7 @@ | |
|
||
# Typical virtualenv dir | ||
/venv/ | ||
/.venv/ | ||
|
||
# IDE settings | ||
/.idea/ | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Binary file not shown.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Empty file.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,145 @@ | ||
import os | ||
import configparser | ||
from typing import Dict | ||
import multiprocessing | ||
|
||
import dns | ||
import requests | ||
import unittest | ||
|
||
from fastapi import FastAPI | ||
import uvicorn | ||
|
||
from semantic_id_resolver import resolver | ||
from semantic_id_resolver.service import SemanticIdResolvingService, SMSRequest | ||
|
||
from contextlib import contextmanager | ||
import signal | ||
import time | ||
|
||
|
||
def run_server(): | ||
# Load test configuration | ||
config = configparser.ConfigParser() | ||
config.read([ | ||
os.path.abspath(os.path.join(os.path.dirname(__file__), "../test_resources/config.ini")), | ||
]) | ||
|
||
# Define test configuration | ||
IRDI_MATCHER_DICT: Dict[resolver.IRDISources, str] = { | ||
resolver.IRDISources.ECLASS: config["RESOLVER"]["eclass_semantic_matching_service"], | ||
resolver.IRDISources.IEC_CDD: config["RESOLVER"]["cdd_semantic_matching_service"] | ||
} | ||
|
||
try: | ||
DEBUG_ENDPOINTS = resolver.DebugSemanticMatchingServiceEndpoints.from_file( | ||
config["RESOLVER"]["debug_semantic_matching_service_endpoints"] | ||
) | ||
print(f"USING DEBUG ENDPOINTS FROM {config['RESOLVER']['debug_semantic_matching_service_endpoints']}") | ||
except FileNotFoundError: | ||
DEBUG_ENDPOINTS = resolver.DebugSemanticMatchingServiceEndpoints(debug_endpoints={}) | ||
|
||
# Mock SemanticIdResolvingService for testing | ||
mock_resolver = resolver.SemanticIdResolver(IRDI_MATCHER_DICT, DEBUG_ENDPOINTS) | ||
semantic_id_resolver_service = SemanticIdResolvingService( | ||
endpoint=config["SERVICE"]["endpoint"], | ||
fallback_semantic_matching_service_endpoint=config["RESOLVER"]["fallback_semantic_matching_service"], | ||
semantic_id_resolver=mock_resolver | ||
) | ||
|
||
# Mock TXT record | ||
class MockTXTRecord: | ||
def __init__(self, strings): | ||
self.strings = strings | ||
|
||
# Mock DNS resolving manually, mock for unittest does not work in this context | ||
def mock_dns_resolver_resolve(qname, rdtype): | ||
return [MockTXTRecord([b"semantic_matcher: https://example.org/iri_semantic_matching_service"])] | ||
dns.resolver.resolve = mock_dns_resolver_resolve | ||
|
||
# Run server | ||
app = FastAPI() | ||
app.include_router(semantic_id_resolver_service.router) | ||
uvicorn.run(app, host=str(config["SERVICE"]["ENDPOINT"]), port=int(config["SERVICE"]["PORT"]), log_level="error") | ||
|
||
|
||
@contextmanager | ||
def run_server_context(): | ||
server_process = multiprocessing.Process(target=run_server) | ||
server_process.start() | ||
try: | ||
time.sleep(2) # Wait for the server to start | ||
yield | ||
finally: | ||
server_process.terminate() | ||
server_process.join(timeout=5) | ||
if server_process.is_alive(): | ||
os.kill(server_process.pid, signal.SIGKILL) | ||
server_process.join() | ||
|
||
|
||
class TestSemanticMatchingService(unittest.TestCase): | ||
|
||
def test_semantic_matching_service_iri(self): | ||
with run_server_context(): | ||
sms_request = SMSRequest(semantic_id="foo://example.org:1234/over/there?name=bar#page=3") | ||
response = requests.get( | ||
"http://localhost:8125/get_semantic_matching_service", | ||
data=sms_request.model_dump_json() | ||
) | ||
self.assertEqual( | ||
"https://example.org/iri_semantic_matching_service", | ||
response.json()["semantic_matching_service_endpoint"] | ||
) | ||
|
||
def test_semantic_matching_service_irdi_eclass(self): | ||
with run_server_context(): | ||
sms_request = SMSRequest(semantic_id="0173-0001#01-ACK323#7") | ||
response = requests.get( | ||
"http://localhost:8125/get_semantic_matching_service", | ||
data=sms_request.model_dump_json() | ||
) | ||
self.assertEqual( | ||
"https://example.org/eclass_semantic_matching_service", | ||
response.json()["semantic_matching_service_endpoint"] | ||
) | ||
|
||
def test_semantic_matching_service_irdi_cdd(self): | ||
with run_server_context(): | ||
sms_request = SMSRequest(semantic_id="0112-0001#01-ACK323#7") | ||
response = requests.get( | ||
"http://localhost:8125/get_semantic_matching_service", | ||
data=sms_request.model_dump_json() | ||
) | ||
self.assertEqual( | ||
"https://example.org/cdd_semantic_matching_service", | ||
response.json()["semantic_matching_service_endpoint"] | ||
) | ||
|
||
def test_semantic_matching_service_fallback(self): | ||
with run_server_context(): | ||
sms_request = SMSRequest(semantic_id="nothing") | ||
response = requests.get( | ||
"http://localhost:8125/get_semantic_matching_service", | ||
data=sms_request.model_dump_json() | ||
) | ||
self.assertEqual( | ||
"https://example.org/fallback_semantic_matching_service", | ||
response.json()["semantic_matching_service_endpoint"] | ||
) | ||
|
||
def test_semantic_matching_service_debug(self): | ||
with run_server_context(): | ||
sms_request = SMSRequest(semantic_id="https://example.org/semanticIDone") | ||
response = requests.get( | ||
"http://localhost:8125/get_semantic_matching_service", | ||
data=sms_request.model_dump_json() | ||
) | ||
self.assertEqual( | ||
"https://example.org/debug_semantic_matching_service", | ||
response.json()["semantic_matching_service_endpoint"] | ||
) | ||
|
||
|
||
if __name__ == '__main__': | ||
unittest.main() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,9 @@ | ||
[SERVICE] | ||
endpoint=127.0.0.1 | ||
port=8125 | ||
|
||
[RESOLVER] | ||
fallback_semantic_matching_service=https://example.org/fallback_semantic_matching_service | ||
eclass_semantic_matching_service=https://example.org/eclass_semantic_matching_service | ||
cdd_semantic_matching_service=https://example.org/cdd_semantic_matching_service | ||
debug_semantic_matching_service_endpoints=test_resources/debug_endpoints.json |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,3 @@ | ||
{ | ||
"https://example.org/semanticIDone": "https://example.org/debug_semantic_matching_service" | ||
} |