Add test data and assertion