Cucumber tests wrapped around original code