Function test_basic

tests/sentry/tasks/test_reprocessing2.py:97–202

(
    task_runner,
    default_project,
    change_groups,
    reset_snuba,
    process_and_save,
    register_event_preprocessor,
    django_cache,
)

@pytest.mark.snuba
@pytest.mark.parametrize("change_groups", (True, False), ids=("new_group", "same_group"))
def test_basic(
    task_runner,
    default_project,
    change_groups,
    reset_snuba,
    process_and_save,
    register_event_preprocessor,
    django_cache,
):
    from sentry import eventstream

    tombstone_calls = []
    old_tombstone_fn = eventstream.backend.tombstone_events_unsafe

    def tombstone_called(*args, **kwargs):
        tombstone_calls.append((args, kwargs))
        old_tombstone_fn(*args, **kwargs)

    with mock.patch("sentry.eventstream.backend.tombstone_events_unsafe", tombstone_called):
        abs_count = 0

        @register_event_preprocessor
        def event_preprocessor(data):
            nonlocal abs_count

            tags = data.setdefault("tags", [])
            assert all(not x or x[0] != "processing_counter" for x in tags)
            tags.append(("processing_counter", f"x{abs_count}"))
            abs_count += 1

            if change_groups:
                data["fingerprint"] = [uuid.uuid4().hex]
            else:
                data["fingerprint"] = ["foo"]

            return data

        event_id = process_and_save({"tags": [["key1", "value"], None, ["key2", "value"]]})

        def get_event_by_processing_counter(n: str) -> list[Event]:
            return list(
                eventstore.backend.get_events(
                    eventstore.Filter(
                        project_ids=[default_project.id],
                        conditions=[["tags[processing_counter]", "=", n]],
                    ),
                    tenant_ids={"organization_id": 1234, "referrer": "eventstore.get_events"},
                )
            )

        event = eventstore.backend.get_event_by_id(
            default_project.id,
            event_id,
            tenant_ids={"organization_id": 1234, "referrer": "eventstore.get_events"},
        )
        assert event is not None
        assert event.get_tag("processing_counter") == "x0"
        assert not event.data.get("errors")