diff --git a/tests/web/rpc/v1/test_endpoint_trace_item_attribute_names.py b/tests/web/rpc/v1/test_endpoint_trace_item_attribute_names.py index 367d73f5da..c80045f146 100644 --- a/tests/web/rpc/v1/test_endpoint_trace_item_attribute_names.py +++ b/tests/web/rpc/v1/test_endpoint_trace_item_attribute_names.py @@ -246,6 +246,7 @@ def test_page_token_offset_filter(self) -> None: ) res = EndpointTraceItemAttributeNames().execute(req) page_token = res.page_token + assert res.WhichOneOf("page_token") == "filter_offset" assert res.attributes == expected_attributes[:at_a_time] expected_attributes = expected_attributes[at_a_time:] done += at_a_time diff --git a/tests/web/rpc/v1/test_trace_item_attribute_values_v1.py b/tests/web/rpc/v1/test_trace_item_attribute_values_v1.py index 441260795c..fb2515fd65 100644 --- a/tests/web/rpc/v1/test_trace_item_attribute_values_v1.py +++ b/tests/web/rpc/v1/test_trace_item_attribute_values_v1.py @@ -145,7 +145,7 @@ def test_page_token(self, setup_teardown: Any) -> None: done += at_a_time assert expected_values == [] - def test_filter_offset_is_not_implemented_right_now(self) -> None: + def test_page_token_filter_offset(self) -> None: expected_values = [ "blah", "derpderp", @@ -170,6 +170,7 @@ def test_filter_offset_is_not_implemented_right_now(self) -> None: ) res = AttributeValuesRequest().execute(req) page_token = res.page_token + assert res.WhichOneOf("page_token") == "filter_offset" assert res.values == expected_values[:at_a_time] expected_values = expected_values[at_a_time:] done += at_a_time