SED-4340 find-usages-of-keyword-and-plan-referenced-by-attribute-is-b… #579

david-stephan · 2026-01-08T13:25:22Z

…roken

* SED-4305: unit tests for AutomationPackageManager * SED-4305: forbid libraries in local execution (CLI) * SED-4305: fix test * SED-4305: new unit test * SED-4305: remove redundant todo * SED-4350: new unit test * SED-4350: merge master into SED-4350 * SED-4350: fix test * SED-4350: unit test * SED-4350: unit test (cherry picked from commit 6be0173)

) * SED-4409 Keyword calls with no agent response have no measurements * SED-4409 PR feedback * SED-4409 PR feedback + Junit

SED-4404: improve error messages (cherry picked from commit 011fbcb)

… fails (#565)

* SED-4413 added connect and read timeouts to AbstractRemoteClient * SED-4413 PR feedback (cherry picked from commit 984f98e)

…r not released (#567) * SED-4401 Application context of step-functions-handler-initializer.jar not released * SED-4413 added connect and read timeouts to AbstractRemoteClient (#563) * SED-4413 added connect and read timeouts to AbstractRemoteClient * SED-4413 PR feedback (cherry picked from commit 984f98e) * SED-4401 Application context of step-functions-handler-initializer.jar not released * Revert "SED-4413 added connect and read timeouts to AbstractRemoteClient (#563)" This reverts commit 5eff1e9. * SED-4401 Fixing NoClassDefFound * SED-4401 Fixing NPE --------- Co-authored-by: Jonathan Rubiero <30461894+rubij@users.noreply.github.com>

…fter merge of SED-4401 (#563)

#569) SED-4439 Optimize reporting request and query performance for Step 29 branch

… resources of a global project (#568) * SED-4387 delete resource services do not check write access for tenant * SED-4387 improve error message displyed in the UI

* SED-4444 timeseries-re-ingestion-doesnt-work-for-psql * SED-4444 timeseries-re-ingestion-doesnt-work-for-psql

* SED-4443 Fixing ClassCastException in FunctionMessageHandler * SED-4443 Fixing ClassCastException in FunctionMessageHandler

…#575) * SED-4418 Calling Keywords in an after section does not release tokens * SED-4418 Calling Keywords in an after section does not release tokens * SED-4418 Adding a junit test

* SED-4287 ForEach may try to use uninitialized tempWriter * SED-4287 propagating exception

…roken

gemini-code-assist · 2026-01-08T13:25:58Z

Summary of Changes

Hello @david-stephan, I'm Gemini Code Assist¹! I'm currently reviewing this pull request and will post my feedback shortly. In the meantime, here's a summary to help you and other reviewers quickly get up to speed!

This pull request resolves a defect in the 'find usages' functionality, specifically targeting issues where keywords and plans referenced by attributes were not correctly identified. The changes involve a significant refactoring of the reference finding mechanism, moving the core logic into a new, dedicated class and enhancing the underlying entity dependency traversal. This ensures that the system can now accurately and comprehensively resolve all potential references, improving the reliability of usage analysis for dynamically referenced entities.

Highlights

New ReferenceFinder Class: Introduced a dedicated ReferenceFinder class to centralize and encapsulate the logic for finding references to plans, functions (keywords), and resources. This promotes better separation of concerns.
Refactored ReferenceFinderServices: The ReferenceFinderServices class has been refactored to delegate its core reference-finding functionality to the new ReferenceFinder, simplifying the REST endpoint implementation.
Enhanced Entity Dependency Tree Traversal: The EntityDependencyTreeVisitor now uses a VISIT_MODE enum (SINGLE, RECURSIVE, RESOLVE_ALL) instead of a boolean flag, providing more granular control over how entity dependencies are traversed and resolved. The RESOLVE_ALL mode is crucial for finding all references.
Improved Plan and Function Entity Resolution: Updated PlanEntity and FunctionEntity to correctly resolve all matching plans and functions, respectively, when the EntityDependencyTreeVisitor is in RESOLVE_ALL mode. This specifically addresses issues with attribute-based lookups.
Flexible Locator Helper: The FunctionLocator and PlanLocator now support a non-strict resolution mode, allowing them to find all potential matches for entities, even if dynamic selection criteria are unresolvable or activation expressions are bypassed.
Comprehensive Unit Tests: Added new unit tests for the ReferenceFinder to ensure the correct identification of references for various entity types and lookup methods (by ID and by name).

Using Gemini Code Assist

The full guide for Gemini Code Assist can be found on our documentation page, here are some quick tips.

Invoking Gemini

You can request assistance from Gemini at any point by creating a comment using either /gemini <command> or @gemini-code-assist <command>. Below is a summary of the supported commands on the current page.

Feature	Command	Description
Code Review	`/gemini review`	Performs a code review for the current pull request in its current state.
Pull Request Summary	`/gemini summary`	Provides a summary of the current pull request in its current state.
Comment	@gemini-code-assist	Responds in comments when explicitly tagged, both in pull request comments and review comments.
Help	`/gemini help`	Displays a list of available commands.

Customization

To customize Gemini Code Assist for GitHub experience, repository maintainers can create a configuration file and/or provide a custom code review style guide (such as PEP-8 for Python) by creating and adding files to a .gemini/ folder in the base of the repository. Detailed instructions can be found here.

Limitations & Feedback

Gemini Code Assist may make mistakes. Please leave feedback on any instances where its feedback is incorrect or counter productive. You can react with 👍 and 👎 on @gemini-code-assist comments. If you're interested in giving your feedback about your experience with Gemini Code Assist for Github and other Google products, sign up here.

You can also get AI-powered code generation, chat, as well as code reviews directly in the IDE at no cost with the Gemini Code Assist IDE Extension.

Review the Privacy Notices, Generative AI Prohibited Use Policy, Terms of Service, and learn how to configure Gemini Code Assist in GitHub here. Gemini can make mistakes, so double check it and use code with caution. ↩

gemini-code-assist

Code Review

This pull request introduces a significant and well-executed refactoring of the reference finding logic to address an issue with locating usages of keywords and plans referenced by attributes. The key improvements include extracting the core logic into a dedicated ReferenceFinder class, introducing a more flexible VISIT_MODE for dependency traversal, and enhancing the PlanLocator and FunctionLocator to support a non-strict resolution mode. This non-strict mode is crucial for finding all potential references without a full execution context. The addition of a comprehensive test suite for the new ReferenceFinder is also a great enhancement.

My review includes a high-severity bug fix for a potential NullPointerException and a couple of medium-severity suggestions to further improve code maintainability and test quality. Overall, these are excellent changes that effectively solve the underlying problem.

step-controller/step-controller-server/src/main/java/step/core/references/ReferenceFinder.java

gemini-code-assist · 2026-01-08T13:28:13Z

step-controller/step-controller-server/src/main/java/step/core/references/ReferenceFinder.java

+        List<Object> referencedObjects = getReferencedObjects(entityType, object).stream().filter(o -> (o != null &&  !o.equals(object))).collect(Collectors.toList());
+        //System.err.println("objects referenced from plan: " + planToString(plan) + ": "+ referencedObjects.stream().map(ReferenceFinderServices::objectToString).collect(Collectors.toList()));
+        return referencedObjects.stream().filter(o -> doesRequestMatch(request, o)).collect(Collectors.toList());


This method can be simplified into a single, more efficient stream pipeline that avoids creating an intermediate list. Additionally, there's a commented-out debug line that should be removed as it's no longer relevant and references code that has been moved.

return getReferencedObjects(entityType, object).stream() .filter(o -> o != null && !o.equals(object)) .filter(o -> doesRequestMatch(request, o)) .collect(Collectors.toList());

...ontroller/step-controller-server/src/test/java/step/core/references/ReferenceFinderTest.java

jeromecomte

The changes look good but I indeed see a risk in the changes related to the FunctionLocator and PlanLocator. The new JUnit test ReferenceFinderTest looks good but I'm not sure if it covers all the aspects of the Locators, especially the ones related to versions. In order to merge this to 29, we should ensure that we cover this properly and add some tests if required

step-core/src/main/java/step/core/entities/EntityDependencyTreeVisitor.java

step-controller/step-controller-server/src/main/java/step/core/references/ReferenceFinder.java

step-plans/step-plans-base-artefacts/src/main/java/step/artefacts/handlers/FunctionLocator.java

david-stephan

Thanks for the review, all points should be addressed with my last commits.

step-controller/step-controller-server/src/main/java/step/core/references/ReferenceFinder.java

jeromecomte · 2026-01-09T11:39:05Z

step-controller/step-controller-server/src/main/java/step/core/references/ReferenceFinder.java

+                        results.add(new FindReferencesResponse(function));
+                    }
+                } catch (Exception e) {
+                    logger.error("Unable to find references for function {}", function.getId(), e);


SHouldn't we rethrow this or add the error to the response?

Not sure how relevant that is. The implementation will browse all plans and all composite keywords to see if they use the searched reference, if an exception is raised because the context could not be rebuilt for one of them, I don't think this should break the operation, adding it as a warning would be doable (with quite some effort) but I also don't think this warning would be relevant for the user doing the request.

OK, get it. I would say it depends a bit on when we expect such exceptions to occur. If this an exception that we except, then I would indeed "ignore" it and even log it as worning or even debug. If we don't expect it, I would rethrow it. Let you decide

I looked at it again, since the "expected" exceptions can only be thrown when rebuilding the context I moved the error handling to that specific location. The only error case I could see is when plans or composite KWs are orphans (i.e. a project has been deleted with leftover plans or composites).

jeromecomte · 2026-01-09T11:39:51Z

step-controller/step-controller-server/src/main/java/step/core/references/ReferenceFinder.java

+                        results.add(new FindReferencesResponse(plan));
+                    }
+                } catch (Exception e) {
+                    logger.error("Unable to find references for plan {}", plan.getId(), e);


* SED-4471 Add audit logging to parameters and schedules * SED-4471 Code simplification * SED-4471 More logging

…580) * SED-4430 Improve Agent and CLI resilience in case of network errors * SED-4430 Reverting method signatures and correct comment * EI-485 new nexus staging host * SED-4430 bumping grid and FW after PR merge --------- Co-authored-by: Jonathan Rubiero <30461894+rubij@users.noreply.github.com>

* SED-4498 Parameters priority not respected when deployed with AP * SED-4498 PR feedbacks * SED-4498 fixing junit

#585) SED-4506 Clean-up of isolated AP resources delete all revisions and files but keep the resource entry in DB

…datasets after re ingestion (#578) * SED-4451 The performance of the analytics view is very poor on large datasets after re-ingestion * SED-4451 The performance of the analytics view is very poor on large datasets after re-ingestion * SED-4451 The performance of the analytics view is very poor on large datasets after re-ingestion * SED-4451 adapting TimeSeriesExecutionPlugin * SED-4451 index optimizations and other perf fixes * SED-4451 bumping framework * SED-4451 PR Feedbacks * SED-4451 adding missing TS index and fixing default resolution * SED-4451 Bumping framework to 0.0.0-25-SNAPSHOT before merge

jeromecomte

Approving with on minor comment

jeromecomte · 2026-02-02T09:06:52Z

step-controller/step-controller-server/src/main/java/step/core/references/ReferenceFinder.java

+            } catch (Exception e) {
+                //The getReferencedObjects method is invoked for all entities found in the system, for some entities (for example plans that belongs to a deleted project), the context cannot be rebuilt.
+                //These expected errors are ignored
+                logger.warn("Unable to inspect the {} with id {}", entityType, object.getId(), e);


Admins might not understand in which context this warning occurs. We might give some more details in order to understand that it relates to the find usage

…enced-by-attribute-is-broken

…eferenced-by-attribute-is-broken # Conflicts: # pom.xml

The base branch was changed.

iegorov777 and others added 18 commits November 27, 2025 15:45

SED-4350 adapting Junit after merge

1deafe9

SED-4409 Keyword calls with no agent response have no measurements (#561

be23145

) * SED-4409 Keyword calls with no agent response have no measurements * SED-4409 PR feedback * SED-4409 PR feedback + Junit

SED-000 AP Junit test flakiness

23a70a3

SED-4403 deploy library error messages are not clear enough (#562)

7078864

SED-4404: improve error messages (cherry picked from commit 011fbcb)

SED-4428 Uploading an AP defining K6 keywords without scriptDirectory…

291833f

… fails (#565)

SED-4413 added connect and read timeouts to AbstractRemoteClient (#563)

5eff1e9

* SED-4413 added connect and read timeouts to AbstractRemoteClient * SED-4413 PR feedback (cherry picked from commit 984f98e)

SED-4413 Re-added connect and read timeouts to AbstractRemoteClient a…

2f8e20e

…fter merge of SED-4401 (#563)

SED-4439 Optimize reporting request and query performance for Step 29… (

732a312

#569) SED-4439 Optimize reporting request and query performance for Step 29 branch

SED-4387 from a project it is possible to use bulk deletion to delete…

480f1f9

… resources of a global project (#568) * SED-4387 delete resource services do not check write access for tenant * SED-4387 improve error message displyed in the UI

SED-4444 timeseries-re-ingestion-doesnt-work-for-psql (#570)

1a1506a

* SED-4444 timeseries-re-ingestion-doesnt-work-for-psql * SED-4444 timeseries-re-ingestion-doesnt-work-for-psql

SED-4443 ClassCastException in FunctionMessageHandler (#572)

0790bc5

* SED-4443 Fixing ClassCastException in FunctionMessageHandler * SED-4443 Fixing ClassCastException in FunctionMessageHandler

SED-4443 Switching to explicit creation of initializer CL (#574)

6098a47

SED-4445 bumping framework to 2.5.1

38ea1cf

SED-4418 Calling Keywords in an after section does not release tokens (…

2407ee2

…#575) * SED-4418 Calling Keywords in an after section does not release tokens * SED-4418 Calling Keywords in an after section does not release tokens * SED-4418 Adding a junit test

SED-4287 ForEach may try to use uninitialized tempWriter (#576)

b5e41d7

* SED-4287 ForEach may try to use uninitialized tempWriter * SED-4287 propagating exception

SED-4340 find-usages-of-keyword-and-plan-referenced-by-attribute-is-b…

174c638

…roken

gemini-code-assist bot reviewed Jan 8, 2026

View reviewed changes

SED-4340 PR feedbacks

5498447

david-stephan requested a review from jeromecomte January 8, 2026 13:46

jeromecomte reviewed Jan 9, 2026

View reviewed changes

david-stephan added 2 commits January 9, 2026 10:50

SED-4340 PR feedbacks

a431643

SED-4340 fixing incorrect context and predicate usage

e0c4c97

david-stephan commented Jan 9, 2026

View reviewed changes

jeromecomte reviewed Jan 9, 2026

View reviewed changes

cl-exense and others added 3 commits January 13, 2026 14:57

SED-4471 Add audit logging to parameters, schedules, resources (#581)

340fc42

* SED-4471 Add audit logging to parameters and schedules * SED-4471 Code simplification * SED-4471 More logging

EI-485 new nexus staging host

8c255de

david-stephan and others added 6 commits January 19, 2026 10:31

SED-4480 BE Junit test flakiness

9ad83df

SED-4498 Parameters priority not respected when deployed with AP (#584)

443eb0d

* SED-4498 Parameters priority not respected when deployed with AP * SED-4498 PR feedbacks * SED-4498 fixing junit

SED-4506 Clean-up of isolated AP resources delete all revisions and f… (

ff8d80d

#585) SED-4506 Clean-up of isolated AP resources delete all revisions and files but keep the resource entry in DB

SED-4340 fixing logging

e459cf6

SED-4340 fixing error handling

36fb4ed

jeromecomte previously approved these changes Feb 2, 2026

View reviewed changes

david-stephan added 3 commits February 2, 2026 11:02

SED-4430 improving warning message

25fd2b8

Merge branch '29' into SED-4340-find-usages-of-keyword-and-plan-refer…

9a52a2e

…enced-by-attribute-is-broken

Merge branch 'master' into SED-4340-find-usages-of-keyword-and-plan-r…

eac1f22

…eferenced-by-attribute-is-broken # Conflicts: # pom.xml

david-stephan changed the base branch from 29 to master February 2, 2026 10:39

SED-4340 find-usages-of-keyword-and-plan-referenced-by-attribute-is-b… #579

Are you sure you want to change the base?

SED-4340 find-usages-of-keyword-and-plan-referenced-by-attribute-is-b… #579

Uh oh!

Conversation

david-stephan commented Jan 8, 2026

Uh oh!

gemini-code-assist bot commented Jan 8, 2026

Summary of Changes

Highlights

Footnotes

Uh oh!

gemini-code-assist bot left a comment

Choose a reason for hiding this comment

Code Review

Uh oh!

Uh oh!

gemini-code-assist bot Jan 8, 2026

Choose a reason for hiding this comment

Uh oh!

Uh oh!

jeromecomte left a comment

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

david-stephan left a comment

Choose a reason for hiding this comment

Uh oh!

Uh oh!

Uh oh!

Uh oh!

jeromecomte Jan 9, 2026

Choose a reason for hiding this comment

Uh oh!

david-stephan Jan 9, 2026

Choose a reason for hiding this comment

Uh oh!

jeromecomte Jan 30, 2026

Choose a reason for hiding this comment

Uh oh!

david-stephan Feb 2, 2026

Choose a reason for hiding this comment

Uh oh!

jeromecomte Jan 9, 2026

Choose a reason for hiding this comment

Uh oh!

jeromecomte left a comment

Choose a reason for hiding this comment

Uh oh!

jeromecomte Feb 2, 2026

Choose a reason for hiding this comment

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

5 participants