Start to add some basic testing around runtime workflow. Currently using mocks