com.google.common.collect.Sets Java Examples

The following examples show how to use com.google.common.collect.Sets. You can vote up the ones you like or vote down the ones you don't like, and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example #1
Source Project: calcite   Author: apache   File: Matcher.java    License: Apache License 2.0 6 votes vote down vote up
public Matcher<E> build() {
  final Set<String> predicateSymbolsNotInGraph =
      Sets.newTreeSet(symbolPredicates.keySet());
  predicateSymbolsNotInGraph.removeAll(automaton.symbolNames);
  if (!predicateSymbolsNotInGraph.isEmpty()) {
    throw new IllegalArgumentException("not all predicate symbols ["
        + predicateSymbolsNotInGraph + "] are in graph ["
        + automaton.symbolNames + "]");
  }
  final ImmutableMap.Builder<String, Predicate<MemoryFactory.Memory<E>>> builder =
      ImmutableMap.builder();
  for (String symbolName : automaton.symbolNames) {
    // If a symbol does not have a predicate, it defaults to true.
    // By convention, "STRT" is used for the start symbol, but it could be
    // anything.
    builder.put(symbolName,
        symbolPredicates.getOrDefault(symbolName, e -> true));
  }
  return new Matcher<>(automaton, builder.build());
}
 
Example #2
Source Project: blueocean-plugin   Author: jenkinsci   File: BlueIssueFactory.java    License: MIT License 6 votes vote down vote up
/**
 * Finds any issues associated with the changeset
 * e.g. a commit message could be "TICKET-123 fix all the things" and be associated with TICKET-123 in JIRA
 * @param changeSetEntry entry
 * @return issues representing the change
 */
public static Collection<BlueIssue> resolve(ChangeLogSet.Entry changeSetEntry) {
    LinkedHashSet<BlueIssue> allIssues = Sets.newLinkedHashSet();
    for (BlueIssueFactory factory : ExtensionList.lookup(BlueIssueFactory.class)) {
        try {
            Collection<BlueIssue> issues = factory.getIssues(changeSetEntry);
            if (issues == null) {
                continue;
            }
            allIssues.addAll(issues);
        } catch (Exception e) {
            LOGGER.log(Level.WARNING,"Unable to fetch issues for changeSetEntry " + e.getMessage(), e);
        }
    }
    return allIssues;
}
 
Example #3
Source Project: tac-kbp-eal   Author: BBN-E   File: ScorerTest.java    License: MIT License 6 votes vote down vote up
private AnswerKey makeAnswerKeyFromCorrectAndIncorrect(final ImmutableSet<Response> correct,
    final ImmutableSet<Response> incorrect, final CorefAnnotation coref) {
  final ImmutableSet.Builder<AssessedResponse> correctAssessedResponses = ImmutableSet.builder();
  for (final Response correctResponse : correct) {
    correctAssessedResponses.add(
        AssessedResponse.assessCorrectly(correctResponse, FillerMentionType.NAME));
  }
  final ImmutableSet.Builder<AssessedResponse> incorrectAssessedResponses =
      ImmutableSet.builder();
  for (final Response incorrectResponse : incorrect) {
    incorrectAssessedResponses
        .add(AssessedResponse.assessWithIncorrectEventType(incorrectResponse));
  }
  return AnswerKey.from(DOC, Sets.union(correctAssessedResponses.build(),
      incorrectAssessedResponses.build()), ImmutableSet.<Response>of(), coref);
}
 
Example #4
Source Project: SearchServices   Author: Alfresco   File: SolrResponsesComparator.java    License: GNU Lesser General Public License v3.0 6 votes vote down vote up
public static String compare(Object[] a, Object[] b, int flags, Map<String, Integer> handle)
{

    boolean ordered = (flags & UNORDERED) == 0;

    if (a.length != b.length)
    {
        return ".length:" + a.length + "!=" + b.length;
    }

    if (!ordered)
    {
        Set<Object> setA = Sets.newHashSet(a);
        Set<Object> setB = Sets.newHashSet(b);
        return compare(setA, setB, flags, handle);
    }


    for (int i = 0; i < a.length; i++)
    {
        String cmp = compare(a[i], b[i], flags, handle);
        if (cmp != null)
            return "[" + i + "]" + cmp;
    }
    return null;
}
 
Example #5
Source Project: secure-data-service   Author: inbloom   File: ExtractorHelperTest.java    License: Apache License 2.0 6 votes vote down vote up
@Test
public void testBuildSubToParentEdOrgCache() {
    EntityToEdOrgCache cache = new EntityToEdOrgCache();
    cache.addEntry("lea-1", "school-1");
    cache.addEntry("lea-1", "school-2");
    cache.addEntry("lea-1", "school-3");
    cache.addEntry("lea-2", "school-4");
    cache.addEntry("lea-2", "school-5");
    cache.addEntry("lea-3", "school-6");

    Map<String, Collection<String>> result = helper.buildSubToParentEdOrgCache(cache);
    Assert.assertEquals(6, result.keySet().size());
    Assert.assertEquals(Sets.newHashSet("lea-1"), result.get("school-1"));
    Assert.assertEquals(Sets.newHashSet("lea-1"), result.get("school-2"));
    Assert.assertEquals(Sets.newHashSet("lea-1"), result.get("school-3"));
    Assert.assertEquals(Sets.newHashSet("lea-2"), result.get("school-4"));
    Assert.assertEquals(Sets.newHashSet("lea-2"), result.get("school-5"));
    Assert.assertEquals(Sets.newHashSet("lea-3"), result.get("school-6"));
}
 
Example #6
Source Project: flow-platform-x   Author: gy2006   File: AgentHostServiceTest.java    License: Apache License 2.0 6 votes vote down vote up
@Test(expected = NotAvailableException.class)
public void should_create_unix_local_host() {
    // when: create host
    AgentHost host = new LocalUnixAgentHost();
    host.setName("test-host");
    host.setTags(Sets.newHashSet("local", "test"));
    agentHostService.createOrUpdate(host);

    // then:
    Assert.assertNotNull(host.getId());
    Assert.assertEquals(AgentHost.Type.LocalUnixSocket, host.getType());
    Assert.assertEquals(1, agentHostService.list().size());
    Assert.assertEquals(host, agentHostService.list().get(0));

    // when: create other
    AgentHost another = new LocalUnixAgentHost();
    another.setName("test-host-failure");
    another.setTags(Sets.newHashSet("local", "test"));
    agentHostService.createOrUpdate(another);
}
 
Example #7
Source Project: genie   Author: Netflix   File: JpaPersistenceServiceImpl.java    License: Apache License 2.0 6 votes vote down vote up
/**
 * {@inheritDoc}
 */
@Override
public void setApplicationsForCommand(
    @NotBlank final String id,
    @NotNull final List<@NotBlank String> applicationIds
) throws NotFoundException, PreconditionFailedException {
    log.debug("[setApplicationsForCommand] Called to set {} for {}", applicationIds, id);
    if (Sets.newHashSet(applicationIds).size() != applicationIds.size()) {
        throw new PreconditionFailedException("Duplicate application id in " + applicationIds);
    }
    final CommandEntity commandEntity = this.commandRepository
        .getCommandAndApplications(id)
        .orElseThrow(() -> new NotFoundException("No command with id " + id + " exists"));
    final List<ApplicationEntity> applicationEntities = Lists.newArrayList();
    for (final String applicationId : applicationIds) {
        applicationEntities.add(
            this.applicationRepository
                .getApplicationAndCommands(applicationId)
                .orElseThrow(() -> new NotFoundException("No application with id " + applicationId + " exists"))
        );
    }
    commandEntity.setApplications(applicationEntities);
}
 
Example #8
Source Project: Quicksql   Author: Qihoo360   File: Matcher.java    License: MIT License 6 votes vote down vote up
public Matcher<E> build() {
  final Set<String> predicateSymbolsNotInGraph =
      Sets.newTreeSet(symbolPredicates.keySet());
  predicateSymbolsNotInGraph.removeAll(automaton.symbolNames);
  if (!predicateSymbolsNotInGraph.isEmpty()) {
    throw new IllegalArgumentException("not all predicate symbols ["
        + predicateSymbolsNotInGraph + "] are in graph ["
        + automaton.symbolNames + "]");
  }
  final ImmutableMap.Builder<String, Predicate<MemoryFactory.Memory<E>>> builder =
      ImmutableMap.builder();
  for (String symbolName : automaton.symbolNames) {
    // If a symbol does not have a predicate, it defaults to true.
    // By convention, "STRT" is used for the start symbol, but it could be
    // anything.
    builder.put(symbolName,
        symbolPredicates.getOrDefault(symbolName, e -> true));
  }
  return new Matcher<>(automaton, builder.build());
}
 
Example #9
Source Project: genie   Author: Netflix   File: JobLauncher.java    License: Apache License 2.0 6 votes vote down vote up
/**
 * Starts the job setup and launch process once the thread is activated.
 */
@Override
public void run() {
    final long start = System.nanoTime();
    final Set<Tag> tags = Sets.newHashSet();
    try {
        this.jobSubmitterService.submitJob(
            this.jobRequest,
            this.cluster,
            this.command,
            this.applications,
            this.memory
        );
        MetricsUtils.addSuccessTags(tags);
    } catch (final GenieException e) {
        log.error("Unable to submit job due to exception: {}", e.getMessage(), e);
        MetricsUtils.addFailureTagsWithException(tags, e);
    } catch (final Throwable t) {
        MetricsUtils.addFailureTagsWithException(tags, t);
        throw t;
    } finally {
        this.registry.timer(JOB_SUBMIT_TIMER_NAME, tags).record(System.nanoTime() - start, TimeUnit.NANOSECONDS);
    }
}
 
Example #10
Source Project: estatio   Author: estatio   File: BadCommandTargets.java    License: Apache License 2.0 6 votes vote down vote up
@Action(semantics = SemanticsOf.SAFE, restrictTo = RestrictTo.PROTOTYPING)
public List<BadTarget> findBadCommandTargets() {

    Set<String> badObjectTypes = Sets.newTreeSet();

    List<Map<String, Object>> rows = isisJdoSupport
            .executeSql("select distinct(substring(target, 1, charindex(':', target)-1)) as objectType from isiscommand.Command order by 1");
    for (Map<String, Object> row : rows) {
        String targetStr = (String) row.get("objectType");
        addIfBad(badObjectTypes, targetStr);
    }

    return Lists.newArrayList(
            FluentIterable.from(badObjectTypes)
                          .transform(x -> new BadTarget(x))
                          .toList());
}
 
Example #11
Source Project: localization_nifi   Author: wangrenlei   File: StandardNiFiServiceFacade.java    License: Apache License 2.0 6 votes vote down vote up
@Override
public ControllerServiceEntity updateControllerService(final Revision revision, final ControllerServiceDTO controllerServiceDTO) {
    // get the component, ensure we have access to it, and perform the update request
    final ControllerServiceNode controllerService = controllerServiceDAO.getControllerService(controllerServiceDTO.getId());
    final RevisionUpdate<ControllerServiceDTO> snapshot = updateComponent(revision,
            controllerService,
            () -> controllerServiceDAO.updateControllerService(controllerServiceDTO),
            cs -> {
                final ControllerServiceDTO dto = dtoFactory.createControllerServiceDto(cs);
                final ControllerServiceReference ref = controllerService.getReferences();
                final ControllerServiceReferencingComponentsEntity referencingComponentsEntity =
                        createControllerServiceReferencingComponentsEntity(ref, Sets.newHashSet(controllerService.getIdentifier()));
                dto.setReferencingComponents(referencingComponentsEntity.getControllerServiceReferencingComponents());
                return dto;
            });

    final PermissionsDTO permissions = dtoFactory.createPermissionsDto(controllerService);
    final List<BulletinDTO> bulletins = dtoFactory.createBulletinDtos(bulletinRepository.findBulletinsForSource(controllerServiceDTO.getId()));
    final List<BulletinEntity> bulletinEntities = bulletins.stream().map(bulletin -> entityFactory.createBulletinEntity(bulletin, permissions.getCanRead())).collect(Collectors.toList());
    return entityFactory.createControllerServiceEntity(snapshot.getComponent(), dtoFactory.createRevisionDTO(snapshot.getLastModification()), permissions, bulletinEntities);
}
 
Example #12
Source Project: tajo   Author: apache   File: TestLogicalPlanner.java    License: Apache License 2.0 6 votes vote down vote up
@Test
public final void testGenerateCuboids() {
  Column [] columns = new Column[3];

  columns[0] = new Column("col1", Type.INT4);
  columns[1] = new Column("col2", Type.INT8);
  columns[2] = new Column("col3", Type.FLOAT4);

  List<Column[]> cube = LogicalPlanner.generateCuboids(columns);
  assertEquals(((int)Math.pow(2, numCubeColumns)), cube.size());

  Set<Set<Column>> cuboids = Sets.newHashSet();
  for (Column [] cols : cube) {
    cuboids.add(Sets.newHashSet(cols));
  }

  for (Set<Column> result : testGenerateCuboidsResult) {
    assertTrue(cuboids.contains(result));
  }
}
 
Example #13
Source Project: cuba   Author: cuba-platform   File: DataContextImpl.java    License: Apache License 2.0 6 votes vote down vote up
@Override
public EntitySet merge(Collection<? extends Entity> entities) {
    checkNotNullArgument(entities, "entity collection is null");

    List<Entity> managedList = new ArrayList<>(entities.size());
    disableListeners = true;
    try {
        Set<Entity> merged = Sets.newIdentityHashSet();

        for (Entity entity : entities) {
            Entity managed = internalMerge(entity, merged, true);
            managedList.add(managed);
        }
    } finally {
        disableListeners = false;
    }
    return EntitySet.of(managedList);
}
 
Example #14
Source Project: fdb-record-layer   Author: FoundationDB   File: ImplementTypeFilterRule.java    License: Apache License 2.0 6 votes vote down vote up
@Override
public void onMatch(@Nonnull PlannerRuleCall call) {
    final LogicalTypeFilterExpression typeFilter = call.get(root);
    final RecordQueryPlan child = call.get(childMatcher);

    Set<String> childRecordTypes = RecordTypesProperty.evaluate(call.getContext(), child);
    Set<String> filterRecordTypes = Sets.newHashSet(typeFilter.getRecordTypes());
    if (filterRecordTypes.containsAll(childRecordTypes)) {
        // type filter is completely redundant, so remove it entirely
        call.yield(call.ref(child));
    } else {
        // otherwise, keep a filter on record types which the child might produce and are included in the filter
        Set<String> unsatisfiedTypeFilters = Sets.intersection(filterRecordTypes, childRecordTypes);
        call.yield(GroupExpressionRef.of(new RecordQueryTypeFilterPlan(child, unsatisfiedTypeFilters)));
    }
}
 
Example #15
Source Project: qmq   Author: qunarcorp   File: TagPullMessageFilterTest.java    License: Apache License 2.0 6 votes vote down vote up
/**
 * 测试请求 and tag 与消息 tag 不一致的情况
 */
@Test
public void testAndTagNotMatches() throws Exception {
	String tag1 = "tag1";
	String tag2 = "tag2";
	String tag3 = "tag3";
	PullRequest request = TestToolBox.createDefaultPullRequest();
	request.setFilters(Lists.newArrayList(new TagPullFilter(TagType.AND, Sets.newHashSet(tag1, tag2, tag3))));
	BaseMessage message = TestToolBox.createDefaultMessage();
	message.addTag(tag1);
	message.addTag(tag2);
	Buffer buffer = TestToolBox.messageToBuffer(message);
	TagPullMessageFilter filter = Mockito.spy(TagPullMessageFilter.class);
	boolean match = filter.match(request, buffer);
	assertFalse(match);
}
 
Example #16
Source Project: astor   Author: SpoonLabs   File: RuntimeTypeCheck.java    License: GNU General Public License v2.0 6 votes vote down vote up
private void visitFunction(NodeTraversal t, Node n) {
  FunctionType funType = n.getJSType().toMaybeFunctionType();
  if (funType != null && !funType.isConstructor()) {
    return;
  }

  Node nodeToInsertAfter = findNodeToInsertAfter(n);

  nodeToInsertAfter = addMarker(funType, nodeToInsertAfter, null);

  TreeSet<ObjectType> stuff = Sets.newTreeSet(ALPHA);
  Iterables.addAll(stuff, funType.getAllImplementedInterfaces());
  for (ObjectType interfaceType : stuff) {
    nodeToInsertAfter =
        addMarker(funType, nodeToInsertAfter, interfaceType);
  }
}
 
Example #17
Source Project: airpal   Author: airbnb   File: AllowAllToken.java    License: Apache License 2.0 6 votes vote down vote up
public AllowAllToken(String host,
        boolean rememberMe,
        String userName,
        Iterable<String> groups,
        String defaultSchema,
        Duration queryTimeout,
        String accessLevel)
{
    this.host = host;
    this.rememberMe = rememberMe;
    this.userName = userName;
    this.groups = Sets.newHashSet(groups);
    this.defaultSchema = defaultSchema;
    this.queryTimeout = queryTimeout;
    this.accessLevel = accessLevel;
}
 
Example #18
Source Project: molgenis   Author: molgenis   File: LocalizationPopulator.java    License: GNU Lesser General Public License v3.0 6 votes vote down vote up
private void updateNamespace(
    AllPropertiesMessageSource source, String namespace, Set<String> messageIds) {
  Map<String, L10nString> toUpdate =
      localizationService
          .getExistingMessages(namespace, messageIds)
          .collect(toMap(L10nString::getMessageID, identity()));
  Map<String, L10nString> toAdd =
      Sets.difference(messageIds, toUpdate.keySet()).stream()
          .map(msgId -> createL10nString(namespace, msgId))
          .collect(toMap(L10nString::getMessageID, identity()));
  Map<String, L10nString> all =
      Maps.asMap(messageIds, messageID -> toUpdate.getOrDefault(messageID, toAdd.get(messageID)));
  all.forEach(
      (messageID, l10nString) -> updateFromSource(source, namespace, messageID, l10nString));
  localizationService.store(toUpdate.values(), toAdd.values());
}
 
Example #19
Source Project: flow-platform-x   Author: FlowCI   File: AgentControllerTest.java    License: Apache License 2.0 6 votes vote down vote up
@Test
public void should_update_agent_resource() throws Throwable {
    Agent agent = createAgent("hello.agent", Sets.newHashSet("test"), StatusCode.OK);

    Agent.Resource resource = new Agent.Resource()
            .setCpu(1)
            .setFreeDisk(2)
            .setTotalDisk(5)
            .setFreeMemory(4)
            .setTotalMemory(20);

    ResponseMessage message = mockMvcHelper.expectSuccessAndReturnClass(post("/agents/resource")
            .header(AgentAuth.HeaderAgentToken, agent.getToken())
            .content(objectMapper.writeValueAsBytes(resource))
            .contentType(MediaType.APPLICATION_JSON), ResponseMessage.class);

    Assert.assertEquals(StatusCode.OK, message.getCode());
}
 
Example #20
Source Project: cm_ext   Author: cloudera   File: UniqueFieldValidatorImpl.java    License: Apache License 2.0 6 votes vote down vote up
@Override
public boolean isValid(Collection<?> list, ConstraintValidatorContext context) {
  if (list != null) {
    Set<Object> seenSoFar = Sets.newHashSet();
    for (Object obj : list) {
      Object value = propertyValue(obj, uniqueField.value());
      if ((value == null) && this.skipNulls) {
        continue;
      }
      if (seenSoFar.contains(value)) {
        addViolation(context, uniqueField.value());
        return false;
      }
      seenSoFar.add(value);
    }
  }
  return true;
}
 
Example #21
Source Project: xtext-xtend   Author: eclipse   File: XtendCompile.java    License: Eclipse Public License 2.0 5 votes vote down vote up
@SuppressWarnings("deprecation")
protected List<String> getClassPath() {
	Set<String> classPath = Sets.newLinkedHashSet();
	classPath.add(project.getBuild().getSourceDirectory());
	try {
		classPath.addAll(project.getCompileClasspathElements());
	} catch (DependencyResolutionRequiredException e) {
		throw new WrappedException(e);
	}
	addDependencies(classPath, project.getCompileArtifacts());
	classPath.remove(project.getBuild().getOutputDirectory());
	return newArrayList(filter(classPath, FILE_EXISTS));
}
 
Example #22
Source Project: tools   Author: spdx   File: SpdxComparer.java    License: Apache License 2.0 5 votes vote down vote up
/**
 * Returns true if two arrays of SPDX elements contain equivalent elements
 * @param elementsA
 * @param elementsB
 * @return
 */
public static boolean elementsEquivalent(RdfModelObject[] elementsA,
		RdfModelObject[] elementsB) {
	if (elementsA == null) {
		return elementsB == null;
	}
	if (elementsB == null) {
		return false;
	}
	if (elementsA.length != elementsB.length) {
		return false;
	}
	Set<Integer> matchedIndexes = Sets.newHashSet();
	for (int i = 0; i < elementsA.length; i++) {
		boolean found = false;
		for (int j = 0; j < elementsB.length; j++) {
			if (!matchedIndexes.contains(j) &&
					elementsA[i].equivalent(elementsB[j])) {
				found = true;
				matchedIndexes.add(j);
				break;
			}
		}
		if (!found) {
			return false;
		}
	}
	return true;
}
 
Example #23
Source Project: emodb   Author: bazaarvoice   File: DistributedTableSerializer.java    License: Apache License 2.0 5 votes vote down vote up
/**
 * Loads the table data from the path and writes it to the output stream.  Returns the set of UUIDs associated with
 * the table, just like {@link #loadAndSerialize(long, java.io.OutputStream)}.
 */
private Set<Long> loadFromNode(String path, OutputStream out)
        throws KeeperException.NoNodeException {
    try {
        // Get the cached representation of this table
        byte[] data = _curator.getData().forPath(path);
        DataInputStream in = new DataInputStream(new ByteArrayInputStream(data));
        int uuidCountOrExceptionCode = in.readInt();

        // A negative first integer indicates an exception condition.
        if (uuidCountOrExceptionCode < 0) {
            String exceptionJson = new String(data, 4, data.length - 4, Charsets.UTF_8);
            switch (uuidCountOrExceptionCode) {
                case UNKNOWN_TABLE:
                    throw JsonHelper.fromJson(exceptionJson, UnknownTableException.class);
                case DROPPED_TABLE:
                    throw JsonHelper.fromJson(exceptionJson, DroppedTableException.class);
            }
        }

        // Load the UUIDs for this table
        Set<Long> uuids = Sets.newHashSet();
        for (int i=0; i < uuidCountOrExceptionCode; i++) {
            uuids.add(in.readLong());
        }

        // Copy the remaining bytes as the content
        ByteStreams.copy(in, out);
        return uuids;
    } catch (Throwable t) {
        Throwables.propagateIfInstanceOf(t, KeeperException.NoNodeException.class);
        throw Throwables.propagate(t);
   }
}
 
Example #24
Source Project: envelope   Author: cloudera-labs   File: TestStepUtils.java    License: Apache License 2.0 5 votes vote down vote up
@Test
public void testHasNoStreamingStep() {
  Set<Step> steps = Sets.newHashSet();
  BatchStep step1 = new BatchStep("step1");
  BatchStep step2 = new BatchStep("step2");
  step1.configure(ConfigFactory.empty());
  step2.configure(ConfigFactory.empty());
  steps.add(step1);
  steps.add(step2);
  
  assertFalse(StepUtils.hasStreamingStep(steps));
}
 
Example #25
Source Project: buck   Author: facebook   File: Flavors.java    License: Apache License 2.0 5 votes vote down vote up
public static Predicate<BuildTarget> containsFlavors(FlavorDomain<?> domain) {
  return input -> {
    ImmutableSet<Flavor> flavorSet =
        Sets.intersection(domain.getFlavors(), input.getFlavors().getSet()).immutableCopy();
    return !flavorSet.isEmpty();
  };
}
 
Example #26
Source Project: cm_ext   Author: cloudera   File: MetricDefinitionFixture.java    License: Apache License 2.0 5 votes vote down vote up
public MetricDefinitionFixture() {
  serviceMetricNames = Sets.newHashSet();
  serviceMetrics = Lists.newArrayList();
  rolesMetricsNames = Maps.newHashMap();
  rolesMetrics = Maps.newHashMap();
  entitiesMetricsNames = Maps.newHashMap();
  additionalServiceEntityTypesMetrics = Maps.newHashMap();
}
 
Example #27
Source Project: guice-validator   Author: xvik   File: GroupsTest.java    License: MIT License 5 votes vote down vote up
@Test
public void testNoContext() throws Exception {
    // ok
    service.noContext(new Model(null, null, "sample"));
    Assert.assertTrue(service.lastCallGroups.length == 0);

    try {
        // default group
        service.noContext(new Model(null, null, null));
    } catch (ConstraintViolationException ex) {
        Set<String> props = PropFunction.convert(ex.getConstraintViolations());
        Assert.assertEquals(1, props.size());
        Assert.assertEquals(Sets.newHashSet("def"), props);
    }
}
 
Example #28
Source Project: emodb   Author: bazaarvoice   File: PartitionedLeaderServiceTest.java    License: Apache License 2.0 5 votes vote down vote up
private Set<Integer> getLeadPartitions(PartitionedLeaderService partitionedLeaderService) {
    Set<Integer> leadPartitions = Sets.newLinkedHashSet();

    for (Integer partition : partitionedLeaderService.getLeadingPartitions()) {
        // Don't just take the service's word for it; double check that the service is actually in the execution body
        Service uncastDelegate = partitionedLeaderService.getPartitionLeaderServices().get(partition).getCurrentDelegateService().orNull();
        TestLeaderService delegate = uncastDelegate != null && uncastDelegate instanceof TestLeaderService ?
                (TestLeaderService) uncastDelegate : null;
        if (delegate != null && delegate.inBody) {
            leadPartitions.add(delegate.partition);
        }
    }

    return leadPartitions;
}
 
Example #29
Source Project: kaif   Author: kaif-open   File: VoteServiceImplTest.java    License: Apache License 2.0 5 votes vote down vote up
@Test
public void listArticleVoters() throws Exception {
  assertEquals(0, service.listArticleVoters(voter, Collections.emptyList()).size());

  service.voteArticle(UP, articleId, voter, EMPTY, 100);
  Article a2 = savedArticle(zoneInfo, savedAccountCitizen("author2"), "title vote");
  service.voteArticle(UP, a2.getArticleId(), voter, EMPTY, 200);

  Set<FlakeId> actual = service.listArticleVoters(voter, asList(articleId, a2.getArticleId()))
      .stream()
      .map(ArticleVoter::getArticleId)
      .collect(Collectors.toSet());
  assertEquals(Sets.newHashSet(articleId, a2.getArticleId()), actual);
}
 
Example #30
@Test(timeOut = 30_000)
public void testTransactionStartedBeforeNonOverlapFenceCommits() throws Exception {

    long startTsTx1 = tsoClient.getNewStartTimestamp().get();

    tsoClient.getFence(7).get();

    try {
        tsoClient.commit(startTsTx1, Sets.newHashSet(c1, c2)).get();
    } catch (ExecutionException ee) {
        Assert.fail("TX should successfully commit");        }
}