Added the unit-test for key-value class