Add Unit Testing to `pkg/workflow`

[jonathan-innis]

Describe the solution you'd like
Unit tests should be added to the files in the workflow package