Refactor to classes and add a test.
Add workflow interface and include Moq.