Source:
public void testFillAccountsToNewZNode() throws Exception { if (useNewZNodePath) { return; } helixConfigProps.put(HelixAccountServiceConfig.BACKFILL_ACCOUNTS_TO_NEW_ZNODE, "true"); vHelixConfigProps = new VerifiableProperties(helixConfigProps); storeConfig = new HelixPropertyStoreConfig(vHelixConfigProps); String updaterThreadPrefix = UUID.randomUUID().toString(); MockHelixAccountServiceFactory mockHelixAccountServiceFactory = new MockHelixAccountServiceFactory(vHelixConfigProps, new MetricRegistry(), notifier, updaterThreadPrefix, mockRouter); accountService = mockHelixAccountServiceFactory.getAccountService(); ((HelixAccountService) accountService).setupRouter(mockRouter); Account newAccountWithoutContainer = new AccountBuilder(refAccountId, refAccountName, refAccountStatus).build(); List<Account> accountsToUpdate = Collections.singletonList(newAccountWithoutContainer); boolean hasUpdateAccountSucceed = accountService.updateAccounts(accountsToUpdate); assertTrue("Wrong update return status", hasUpdateAccountSucceed); assertAccountsInAccountService(accountsToUpdate, 1, accountService); HelixPropertyStore<ZNRecord> helixStore = mockHelixAccountServiceFactory.getHelixStore(ZK_CONNECT_STRING, storeConfig); HelixAccountService helixAccountService = (HelixAccountService) accountService; RouterStore routerStore = new RouterStore(helixAccountService.getAccountServiceMetrics(), helixAccountService.getBackup(), helixStore, new AtomicReference<>(mockRouter), false); Map<String, String> accountMap = routerStore.fetchAccountMetadata(); assertNotNull("Accounts should be backfilled to new znode", accountMap); assertAccountMapEquals(accountService.getAllAccounts(), accountMap); }

Target:
public void testFillAccountsToNewZNode() throws Exception { assumeTrue(!useNewZNodePath); helixConfigProps.put(HelixAccountServiceConfig.BACKFILL_ACCOUNTS_TO_NEW_ZNODE, "true"); vHelixConfigProps = new VerifiableProperties(helixConfigProps); storeConfig = new HelixPropertyStoreConfig(vHelixConfigProps); String updaterThreadPrefix = UUID.randomUUID().toString(); MockHelixAccountServiceFactory mockHelixAccountServiceFactory = new MockHelixAccountServiceFactory(vHelixConfigProps, new MetricRegistry(), notifier, updaterThreadPrefix, mockRouter); accountService = mockHelixAccountServiceFactory.getAccountService(); ((HelixAccountService) accountService).setupRouter(mockRouter); Account newAccountWithoutContainer = new AccountBuilder(refAccountId, refAccountName, refAccountStatus).build(); List<Account> accountsToUpdate = Collections.singletonList(newAccountWithoutContainer); boolean hasUpdateAccountSucceed = accountService.updateAccounts(accountsToUpdate); assertTrue("Wrong update return status", hasUpdateAccountSucceed); assertAccountsInAccountService(accountsToUpdate, 1, accountService); HelixPropertyStore<ZNRecord> helixStore = mockHelixAccountServiceFactory.getHelixStore(ZK_CONNECT_STRING, storeConfig); HelixAccountService helixAccountService = (HelixAccountService) accountService; RouterStore routerStore = new RouterStore(helixAccountService.getAccountServiceMetrics(), helixAccountService.getBackup(), helixStore, new AtomicReference<>(mockRouter), false); Map<String, String> accountMap = routerStore.fetchAccountMetadata(); assertNotNull("Accounts should be backfilled to new znode", accountMap); assertAccountMapEquals(accountService.getAllAccounts(), accountMap); }