diff --git a/tests/unit/test_adapter_session.py b/tests/unit/test_adapter_session.py index 75f4bcc..bf4fcf1 100644 --- a/tests/unit/test_adapter_session.py +++ b/tests/unit/test_adapter_session.py @@ -33,3 +33,22 @@ def test_get_segment_by_id(rf, monkeypatch): segment_x = adapter.get_segment_by_id(segment_2.pk) assert segment_x == segment_2 + + +@pytest.mark.django_db +def test_refresh_removes_disabled(rf, monkeypatch): + request = rf.get('/') + + adapter = adapters.SessionSegmentsAdapter(request) + + segment_1 = SegmentFactory(name='segment-1', persistent=True) + segment_2 = SegmentFactory(name='segment-2', persistent=True) + + adapter.set_segments([segment_1, segment_2]) + + adapter = adapters.SessionSegmentsAdapter(request) + segment_1.status = segment_1.STATUS_DISABLED + segment_1.save() + adapter.refresh() + + assert adapter.get_segments() == [segment_2]