update key_value_store_test.cc