How to use

getTagsForObject

method

in

com.ning.billing.util.api.TagUserApi

protected Response getTags(final UUID accountId, final UUID taggedObjectId, final AuditMode auditMode, final boolean includeDeleted, final TenantContext context) throws TagDefinitionApiException {
  final List<Tag> tags = tagUserApi.getTagsForObject(taggedObjectId, getObjectType(), includeDeleted, context);
  final AccountAuditLogsForObjectType tagsAuditLogs = auditUserApi.getAccountAuditLogs(accountId, ObjectType.TAG, auditMode.getLevel(), context);
  final Map<UUID, TagDefinition> tagDefinitionsCache = new HashMap<UUID, TagDefinition>();
  final Collection<TagJson> result = new LinkedList<TagJson>();
  for (final Tag tag : tags) {
    if (tagDefinitionsCache.get(tag.getTagDefinitionId()) == null) {
      tagDefinitionsCache.put(tag.getTagDefinitionId(), tagUserApi.getTagDefinition(tag.getTagDefinitionId(), context));
    }
    final TagDefinition tagDefinition = tagDefinitionsCache.get(tag.getTagDefinitionId());
    final List<AuditLog> auditLogs = tagsAuditLogs.getAuditLogs(tag.getId());
    result.add(new TagJson(tag, tagDefinition, auditLogs));
  }
  return Response.status(Response.Status.OK).entity(result).build();
}

  @Test(groups = "slow")
  public void testAddRemoveWrittenOffTag() throws InvoiceApiException, TagApiException {
    invoiceUserApi.tagInvoiceAsWrittenOff(invoiceId, callContext);

    List<Tag> tags = tagUserApi.getTagsForObject(invoiceId, ObjectType.INVOICE, false, callContext);
    assertEquals(tags.size(), 1);
    assertEquals(tags.get(0).getTagDefinitionId(), ControlTagType.WRITTEN_OFF.getId());

    invoiceUserApi.tagInvoiceAsNotWrittenOff(invoiceId, callContext);
    tags = tagUserApi.getTagsForObject(invoiceId, ObjectType.INVOICE, false, callContext);
    assertEquals(tags.size(), 0);
  }
}