track4px/test.py

from unittest import TestCase, main
from configparser import ConfigParser

import json

from dpdtrack import *

class TestHTTPRequest(TestCase):
    def test_http_request(self):
        http = HTTPRequest("https://httpbin.kumi.systems/get")
        response = http.execute()
        self.assertEqual(response["headers"]["User-Agent"], http.USER_AGENT)

class TestDPD(TestCase):
    def setUp(self):
        self.api = DPD()

    def test_tracking(self):
        tracking_number = "01155036780055"
        response = self.api.tracking(tracking_number, wrap=True)
        self.assertTrue(response.events)
        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")