[Backport 2.x ]Fix test for single node and consume numNodes (#1091)

[monusingh-1]

Description

backport #1091 to 2.x branch.

Issues Resolved

[List any issues this PR will resolve]

Check List

• New functionality includes testing.
  • All tests pass
• New functionality has been documented.
  • New functionality has javadoc added
• Commits are signed per the DCO using --signoff

By submitting this pull request, I confirm that my contribution is made under the terms of the Apache 2.0 license.
For more information on following Developer Certificate of Origin and signing off your commits, please check here.

[codecov]

Codecov Report

Merging #1105 (b86ac7c) into 2.x (9e2587c) will increase coverage by 1.36%.
The diff coverage is n/a.

❗ Current head b86ac7c differs from pull request most recent head d7d76e8. Consider uploading reports for the commit d7d76e8 to get more accurate results

@@             Coverage Diff              @@
##                2.x    #1105      +/-   ##
============================================
+ Coverage     75.12%   76.48%   +1.36%     
- Complexity     1026     1042      +16     
============================================
  Files           141      141              
  Lines          4763     4763              
  Branches        521      521              
============================================
+ Hits           3578     3643      +65     
+ Misses          851      780      -71     
- Partials        334      340       +6     

see 24 files with indirect coverage changes