easypost-python/tests/test_tracker.py at c6eaa64099cacd0e6e76810d0b19460afb550a8c · EasyPost/easypost-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
import pytest

from easypost.constant import (
    _FILTERS_KEY,
    _TEST_FAILED_INTENTIONALLY_ERROR,
    NO_MORE_PAGES_ERROR,
)
from easypost.models import Tracker


@pytest.mark.vcr()
def test_tracker_create(test_client):
    tracker = test_client.tracker.create(tracking_code="EZ1000000001")

    assert isinstance(tracker, Tracker)
    assert str.startswith(tracker.id, "trk_")
    assert tracker.status == "pre_transit"


@pytest.mark.vcr()
def test_tracker_retrieve(test_client):
    tracker = test_client.tracker.create(tracking_code="EZ1000000001")

    retrieved_tracker = test_client.tracker.retrieve(tracker.id)

    assert isinstance(retrieved_tracker, Tracker)
    # status changes between creation and retrieval, so we can't compare the whole object
    assert retrieved_tracker.id == tracker.id


@pytest.mark.vcr()
def test_tracker_all(page_size, test_client):
    trackers = test_client.tracker.all(page_size=page_size)

    trackers_array = trackers["trackers"]

    assert len(trackers_array) <= page_size
    assert trackers["has_more"] is not None
    assert all(isinstance(tracker, Tracker) for tracker in trackers_array)


@pytest.mark.vcr()
def test_tracker_retrieve_batch(page_size, test_client):
    tracker = test_client.tracker.create(tracking_code="EZ1000000001")

    trackers = test_client.tracker.retrieve_batch(tracking_codes=[tracker.tracking_code])

    trackers_array = trackers["trackers"]

    assert all(isinstance(tracker, Tracker) for tracker in trackers_array)


@pytest.mark.vcr()
def test_tracker_get_next_page(page_size, test_client):
    try:
        first_page = test_client.tracker.all(page_size=page_size)
        next_page = test_client.tracker.get_next_page(trackers=first_page, page_size=page_size)

        first_id_of_first_page = first_page["trackers"][0].id
        first_id_of_second_page = next_page["trackers"][0].id

        assert first_id_of_first_page != first_id_of_second_page

        # Verify that the filters are being passed along for behind-the-scenes reference
        assert first_page[_FILTERS_KEY] == next_page[_FILTERS_KEY]
    except Exception as e:
        if e.message != NO_MORE_PAGES_ERROR:
            raise Exception(_TEST_FAILED_INTENTIONALLY_ERROR)


@pytest.mark.vcr()
def test_tracker_delete(test_client):
    tracker = test_client.tracker.create(tracking_code="EZ1000000001")

    # Nothing gets returned here, simply ensure no error gets raised
    test_client.tracker.delete(tracker.id)