Add test cases for set key