refactor: update unit tests to use Track4PX API
Replaced DPD API references with Track4PX in unit tests to align with the updated API structure. Removed unused imports to clean up the code. This ensures compatibility with the current API implementation and improves maintainability.
This commit is contained in:
parent
5a05062267
commit
c37fe815c1
2 changed files with 5 additions and 28 deletions
|
@ -2,8 +2,6 @@ from datetime import datetime
|
||||||
|
|
||||||
import json
|
import json
|
||||||
|
|
||||||
import bs4
|
|
||||||
|
|
||||||
from .http import HTTPRequest
|
from .http import HTTPRequest
|
||||||
from .shipment import Shipment, Event
|
from .shipment import Shipment, Event
|
||||||
|
|
||||||
|
|
31
test.py
31
test.py
|
@ -1,9 +1,6 @@
|
||||||
from unittest import TestCase, main
|
from unittest import TestCase
|
||||||
from configparser import ConfigParser
|
|
||||||
|
|
||||||
import json
|
from track4px import HTTPRequest, Track4PX
|
||||||
|
|
||||||
from dpdtrack import *
|
|
||||||
|
|
||||||
class TestHTTPRequest(TestCase):
|
class TestHTTPRequest(TestCase):
|
||||||
def test_http_request(self):
|
def test_http_request(self):
|
||||||
|
@ -11,30 +8,12 @@ class TestHTTPRequest(TestCase):
|
||||||
response = http.execute()
|
response = http.execute()
|
||||||
self.assertEqual(response["headers"]["User-Agent"], http.USER_AGENT)
|
self.assertEqual(response["headers"]["User-Agent"], http.USER_AGENT)
|
||||||
|
|
||||||
class TestDPD(TestCase):
|
class TestTrack4PX(TestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
self.api = DPD()
|
self.api = Track4PX()
|
||||||
|
|
||||||
def test_tracking(self):
|
def test_tracking(self):
|
||||||
tracking_number = "01155036780055"
|
tracking_number = "4PX3001291278502CN"
|
||||||
response = self.api.tracking(tracking_number, wrap=True)
|
response = self.api.tracking(tracking_number, wrap=True)
|
||||||
self.assertTrue(response.events)
|
self.assertTrue(response.events)
|
||||||
self.assertEqual(response.tracking_number, tracking_number)
|
self.assertEqual(response.tracking_number, tracking_number)
|
||||||
|
|
||||||
def test_tracking_with_postal_code(self):
|
|
||||||
tracking_number = "01155036780055"
|
|
||||||
postal_code = "8010"
|
|
||||||
response = self.api.tracking(tracking_number, postal_code=postal_code, wrap=True)
|
|
||||||
self.assertTrue(response.events)
|
|
||||||
self.assertEqual(response.tracking_number, tracking_number)
|
|
||||||
|
|
||||||
class DPDROTest(TestCase):
|
|
||||||
def setUp(self):
|
|
||||||
self.api = DPDRO()
|
|
||||||
|
|
||||||
def test_tracking(self):
|
|
||||||
tracking_number = "80720052822"
|
|
||||||
response = self.api.tracking(tracking_number, wrap=True)
|
|
||||||
self.assertTrue(response.events)
|
|
||||||
self.assertEqual(response.tracking_number, tracking_number)
|
|
||||||
self.assertEqual(response.remote[0][1], "DPDDE")
|
|
Loading…
Reference in a new issue