Add test for the StatefullStoryRunner