Add test coverage for demo component