Add test for decorator.