Skip to content

[FLINK-37944][test] Add utils for initializing process function test harnesses with async state. #26697

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

AlexYinHan
Copy link
Contributor

What is the purpose of the change

This PR introduces new util functions in ProcessFunctionTestHarnesses, for easy initialization of process function test harnesses with async state.

Brief change log

  • In ProcessFunctionTestHarnesses, add new functions forKeyedAsyncProcessFunction and forKeyedAsyncCoProcessFunction

Verifying this change

This change added tests and can be verified as follows:

  • Add UTs in testHarnessForAsyncKeyedProcessFunction, which access states through async APIs.

Does this pull request potentially affect one of the following parts:

  • Dependencies (does it add or upgrade a dependency): (no)
  • The public API, i.e., is any changed class annotated with @Public(Evolving): (no)
  • The serializers: (no)
  • The runtime per-record code paths (performance sensitive): (no)
  • Anything that affects deployment or recovery: JobManager (and its components), Checkpointing, Kubernetes/Yarn, ZooKeeper: (no)
  • The S3 file system connector: (no)

Documentation

  • Does this pull request introduce a new feature? (no)

@flinkbot
Copy link
Collaborator

flinkbot commented Jun 18, 2025

CI report:

Bot commands The @flinkbot bot supports the following commands:
  • @flinkbot run azure re-run the last Azure build

@@ -87,6 +91,24 @@ KeyedOneInputStreamOperatorTestHarness<K, IN, OUT> forKeyedProcessFunction(
return testHarness;
}

public static <K, IN, OUT>
KeyedOneInputStreamOperatorTestHarness<K, IN, OUT> forKeyedAsyncProcessFunction(
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'd suggest naming it forKeyedProcessFunctionWithStateV2, since the function still is KeyedProcessFunction, WDYT?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants