add tests for decorator