Refactor test with DSL for assert message