Add test coverage for ContainerAwareTrait