I see you actually included the tests, do you mind adding the @should annotation and use the test generator plugin if you can figure out how to use it