Refactor StageCollection in service of testing

[#54320562]

Signed-off-by: Tony Hansmann <thansmann@pivotallabs.com>