Complete tests for decorated class