@Test
public void testUpdateRecord() {

  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.updateRecord(new HashMap<>(), "anyChannel", "anyUserExtId");
  Assert.assertEquals(true, isRecordUpdated);
}