Add test for setting block