@Test
public void testUpdateClaimStatus() {

  ShadowUser shadowUser =
      new ShadowUser.ShadowUserBuilder()
          .setChannel("anyChannel")
          .setUserExtId("anyUserExtId")
          .build();
  Map<String, Object> compositeKeysMap = new HashMap<>();
  compositeKeysMap.put(JsonKey.USER_EXT_ID, "anyUserExtId");
  compositeKeysMap.put(JsonKey.CHANNEL, "anyChannel");
  when(cassandraOperationImpl.updateRecord(
          JsonKey.SUNBIRD, JsonKey.SHADOW_USER, new HashMap<>(), compositeKeysMap))
      .thenReturn(response);
  boolean isRecordUpdated =
      MigrationUtils.updateClaimStatus(shadowUser, ClaimStatus.ELIGIBLE.getValue());
  Assert.assertEquals(true, isRecordUpdated);
}