@Test
@PrepareForTest(ZooKeeperMap.class)
public void testRegisterWatchAllFileSets() throws Exception {

  PowerMockito.mockStatic(ZooKeeperMap.class);
  when(ZooKeeperMap.create(any(ZooKeeperClient.class), anyString(), any(Function.class)))
      .thenReturn(null);
  ArgumentCaptor<String> pathCaptor = ArgumentCaptor.forClass(String.class);
  zkManager.registerWatchAllFileSets();
  verifyStatic(times(2));
  ZooKeeperMap.create(any(ZooKeeperClient.class), pathCaptor.capture(), any(Function.class));
  Set<String> allPaths = Sets.newHashSet(pathCaptor.getAllValues());
  assertTrue(allPaths.contains(FILE_SET_DIR));
  assertTrue(allPaths.contains(VIEWS_DIR));
}