86 lines
2.5 KiB
Python
86 lines
2.5 KiB
Python
import importlib.util
|
|
from pathlib import Path
|
|
import sys
|
|
import types
|
|
import unittest
|
|
from unittest.mock import patch
|
|
|
|
|
|
MODULE_PATH = (
|
|
Path(__file__).resolve().parents[1] / "core" / "identification_feedback.py"
|
|
)
|
|
SPEC = importlib.util.spec_from_file_location(
|
|
"identification_feedback_under_test", MODULE_PATH
|
|
)
|
|
FEEDBACK = importlib.util.module_from_spec(SPEC)
|
|
SPEC.loader.exec_module(FEEDBACK)
|
|
|
|
|
|
class FakeResponse:
|
|
def __init__(self, body):
|
|
self.body = body
|
|
|
|
def raise_for_status(self):
|
|
return None
|
|
|
|
def json(self):
|
|
return self.body
|
|
|
|
|
|
class IdentificationFeedbackTests(unittest.TestCase):
|
|
def call_with_response(self, response_body, callback):
|
|
calls = []
|
|
requests_module = types.ModuleType("requests")
|
|
|
|
def post(url, json, timeout):
|
|
calls.append((url, json, timeout))
|
|
return FakeResponse(response_body)
|
|
|
|
requests_module.post = post
|
|
api_module = types.ModuleType("api")
|
|
api_module.data_record_url = "https://cloud/data_record"
|
|
api_module.the_folder = "客户A"
|
|
with patch.dict(sys.modules, {
|
|
"requests": requests_module,
|
|
"api": api_module,
|
|
}):
|
|
value = callback()
|
|
return value, calls
|
|
|
|
def test_registers_uploaded_csv(self):
|
|
_, calls = self.call_with_response(
|
|
{"success": True},
|
|
lambda: FEEDBACK.register_identification_result("result.csv", 7),
|
|
)
|
|
self.assertEqual(calls[0][1], {
|
|
"type": "registerIdentificationResult",
|
|
"deviceId": "客户A",
|
|
"runId": "result.csv",
|
|
"fileName": "result.csv",
|
|
})
|
|
|
|
def test_pending_feedback_returns_none(self):
|
|
value, _ = self.call_with_response(
|
|
{"success": True, "ready": False},
|
|
lambda: FEEDBACK.get_identification_feedback("result.csv"),
|
|
)
|
|
self.assertIsNone(value)
|
|
|
|
def test_feedback_returns_only_zero_or_one(self):
|
|
for result in (0, 1):
|
|
value, _ = self.call_with_response(
|
|
{"success": True, "ready": True, "result": result},
|
|
lambda: FEEDBACK.get_identification_feedback("result.csv"),
|
|
)
|
|
self.assertEqual(value, result)
|
|
|
|
with self.assertRaisesRegex(ValueError, "0 或 1"):
|
|
self.call_with_response(
|
|
{"success": True, "ready": True, "result": 2},
|
|
lambda: FEEDBACK.get_identification_feedback("result.csv"),
|
|
)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|