I copied the original StateFlowGraphTest and modified it for testing the new class but I will submit the abstracted version in a new pull request.