Examples with CountObjectsResponse com.zimbra.soap.admin.message.CountObjectsResponse used on opensource projects

Example 1 with CountObjectsResponse

use of com.zimbra.soap.admin.message.CountObjectsResponse in project zm-mailbox by Zimbra.

the class TestCountObjects method testDomainAdmin.

@Test
public void testDomainAdmin() throws Exception {
    try {
        SoapTransport transport = SoapTest.authAdmin(DOMAIN_ADMIN_USER_EMAIL, PASSWORD);
        // count domains
        try {
            CountObjectsRequest req = new CountObjectsRequest(CountObjectsType.domain);
            CountObjectsResponse resp = SoapTest.invokeJaxb(transport, req);
            fail("should not be able to count domains");
        } catch (SoapFaultException e) {
            assertEquals("should not be able to count domains", ServiceException.PERM_DENIED, e.getCode());
        }
        // count COSes
        try {
            CountObjectsRequest req = new CountObjectsRequest(CountObjectsType.cos);
            CountObjectsResponse resp = SoapTest.invokeJaxb(transport, req);
            fail("should not be able to count COSes");
        } catch (SoapFaultException e) {
            assertEquals("should not be able to count COSes", ServiceException.PERM_DENIED, e.getCode());
        }
        // count user accounts
        CountObjectsRequest req = new CountObjectsRequest(CountObjectsType.userAccount);
        CountObjectsResponse resp = SoapTest.invokeJaxb(transport, req);
        assertTrue("should have at least one account", resp.getNum() > 0);
        assertEquals("object type in response should be 'userAccount'", "userAccount", resp.getType());
        // count accounts
        req = new CountObjectsRequest(CountObjectsType.account);
        resp = SoapTest.invokeJaxb(transport, req);
        assertTrue("should have at least one account", resp.getNum() > 0);
        assertEquals("object type in response should be 'account'", "account", resp.getType());
        // count distribution lists
        req = new CountObjectsRequest(CountObjectsType.dl);
        resp = SoapTest.invokeJaxb(transport, req);
        assertTrue("should have at least one distribution list", resp.getNum() > 0);
        assertEquals("object type in response should be 'dl'", "dl", resp.getType());
        // count aliases without a domain filter
        req = new CountObjectsRequest(CountObjectsType.alias);
        resp = SoapTest.invokeJaxb(transport, req);
        assertTrue("should have at least one alias", resp.getNum() > 0);
        assertEquals("object type in response should be 'alias'", "alias", resp.getType());
        // count user accounts with domain filter
        req = new CountObjectsRequest(CountObjectsType.userAccount);
        req.addDomain(new DomainSelector(DomainBy.name, DOMAIN_NAME));
        resp = SoapTest.invokeJaxb(transport, req);
        assertTrue("should have at least one userAccount", resp.getNum() > 0);
        assertEquals("object type in response should be 'userAccount'", "userAccount", resp.getType());
        // count accounts with domain filter
        req = new CountObjectsRequest(CountObjectsType.account, new DomainSelector(DomainBy.name, DOMAIN_NAME));
        resp = SoapTest.invokeJaxb(transport, req);
        assertTrue("should have at least one account", resp.getNum() > 0);
        assertEquals("object type in response should be 'account'", "account", resp.getType());
        // count DLs with domain filter
        req = new CountObjectsRequest(CountObjectsType.dl, new DomainSelector(DomainBy.name, DOMAIN_NAME));
        resp = SoapTest.invokeJaxb(transport, req);
        assertTrue("should have at least one distribution list", resp.getNum() > 0);
        assertEquals("object type in response should be 'dl'", "dl", resp.getType());
        // count aliases with domain filter
        req = new CountObjectsRequest(CountObjectsType.alias, new DomainSelector(DomainBy.name, DOMAIN_NAME));
        resp = SoapTest.invokeJaxb(transport, req);
        assertTrue("should have at least one alias", resp.getNum() > 0);
        assertEquals("object type in response should be 'alias'", "alias", resp.getType());
    } catch (Exception e) {
        fail(e.getLocalizedMessage());
    }
}

Example 2 with CountObjectsResponse

use of com.zimbra.soap.admin.message.CountObjectsResponse in project zm-mailbox by Zimbra.

the class TestCountObjects method testSuperAdmin.

@Test
public void testSuperAdmin() {
    try {
        SoapTransport transport = TestUtil.getAdminSoapTransport();
        // count domains
        CountObjectsRequest req = new CountObjectsRequest(CountObjectsType.domain);
        CountObjectsResponse resp = SoapTest.invokeJaxb(transport, req);
        assertTrue("should have at least one domain", resp.getNum() > 0);
        // count user accounts
        req = new CountObjectsRequest(CountObjectsType.userAccount);
        resp = SoapTest.invokeJaxb(transport, req);
        assertTrue("should have at least one userAccount", resp.getNum() > 0);
        // count accounts
        req = new CountObjectsRequest(CountObjectsType.account);
        resp = SoapTest.invokeJaxb(transport, req);
        assertTrue("should have at least one account", resp.getNum() > 0);
        // count DLs
        req = new CountObjectsRequest(CountObjectsType.dl);
        resp = SoapTest.invokeJaxb(transport, req);
        assertTrue("should have at least one distribution list", resp.getNum() > 0);
        // count aliases
        req = new CountObjectsRequest(CountObjectsType.alias);
        resp = SoapTest.invokeJaxb(transport, req);
        assertTrue("should have at least one alias", resp.getNum() > 0);
        // count COSes
        req = new CountObjectsRequest(CountObjectsType.cos);
        resp = SoapTest.invokeJaxb(transport, req);
        assertTrue("should have at least one COS", resp.getNum() > 0);
    } catch (Exception e) {
        fail(e.getLocalizedMessage());
    }
}

Example 3 with CountObjectsResponse

use of com.zimbra.soap.admin.message.CountObjectsResponse in project zm-mailbox by Zimbra.

the class TestCountObjects method testRogueAdmin.

@Test
public void testRogueAdmin() throws Exception {
    try {
        SoapTransport transport = SoapTest.authAdmin(ROGUE_ADMIN_USER_EMAIL, PASSWORD);
        // count domains
        try {
            CountObjectsRequest req = new CountObjectsRequest(CountObjectsType.domain);
            CountObjectsResponse resp = SoapTest.invokeJaxb(transport, req);
            fail("should not be able to count domains");
        } catch (SoapFaultException e) {
            assertEquals(ServiceException.PERM_DENIED, e.getCode());
        }
        // count user accounts
        try {
            CountObjectsRequest req = new CountObjectsRequest(CountObjectsType.userAccount, new DomainSelector(DomainBy.name, DOMAIN_NAME));
            CountObjectsResponse resp = SoapTest.invokeJaxb(transport, req);
            fail("should not be able to count accounts");
        } catch (SoapFaultException e) {
            assertEquals(ServiceException.PERM_DENIED, e.getCode());
        }
        // count accounts
        try {
            CountObjectsRequest req = new CountObjectsRequest(CountObjectsType.account, new DomainSelector(DomainBy.name, DOMAIN_NAME));
            CountObjectsResponse resp = SoapTest.invokeJaxb(transport, req);
            fail("should not be able to count accounts");
        } catch (SoapFaultException e) {
            assertEquals(ServiceException.PERM_DENIED, e.getCode());
        }
        // count distribution lists
        try {
            CountObjectsRequest req = new CountObjectsRequest(CountObjectsType.dl);
            req.addDomain(new DomainSelector(DomainBy.name, DOMAIN_NAME));
            CountObjectsResponse resp = SoapTest.invokeJaxb(transport, req);
            assertTrue("should have at least one distribution list", resp.getNum() > 0);
        } catch (ServiceException e) {
            fail("should not be throwing exception here " + e.getLocalizedMessage());
        }
        // count aliases
        try {
            CountObjectsRequest req = new CountObjectsRequest(CountObjectsType.alias, new DomainSelector(DomainBy.name, DOMAIN_NAME));
            CountObjectsResponse resp = SoapTest.invokeJaxb(transport, req);
            fail("should not be able to count aliases");
        } catch (SoapFaultException e) {
            assertEquals(ServiceException.PERM_DENIED, e.getCode());
        }
    } catch (Exception e) {
        fail(e.getLocalizedMessage());
    }
}

Example 4 with CountObjectsResponse

use of com.zimbra.soap.admin.message.CountObjectsResponse in project zm-mailbox by Zimbra.

the class CountObjects method handle.

@Override
public Element handle(Element request, Map<String, Object> context) throws ServiceException {
    ZimbraSoapContext zsc = getZimbraSoapContext(context);
    CountObjectsRequest req = JaxbUtil.elementToJaxb(request);
    CountObjectsType countObjectsType = req.getType();
    if (countObjectsType == null) {
        throw ServiceException.INVALID_REQUEST("No type specified", null);
    }
    Provisioning prov = Provisioning.getInstance();
    UCService ucService = null;
    UCServiceSelector ucserviceSelector = req.getUcService();
    if (null != ucserviceSelector) {
        if (!countObjectsType.allowsUCService()) {
            throw ServiceException.INVALID_REQUEST("UCService cannot be specified for type: " + countObjectsType.name(), null);
        }
        String value = ucserviceSelector.getKey();
        ucService = prov.get(Key.UCServiceBy.fromString(ucserviceSelector.getBy().name()), value);
        if (ucService == null) {
            throw AccountServiceException.NO_SUCH_UC_SERVICE(value);
        }
    }
    List<DomainSelector> specifiedDomains = req.getDomains();
    if (!countObjectsType.allowsDomain() && !specifiedDomains.isEmpty()) {
        throw ServiceException.INVALID_REQUEST("domain cannot be specified for type: " + countObjectsType.name(), null);
    }
    long count = 0;
    if (specifiedDomains.isEmpty() && !zsc.getAuthToken().isAdmin() && countObjectsType.allowsDomain() && !countObjectsType.equals(CountObjectsType.domain)) {
        // if a delegated admin is trying to count objects that exist within
        // a domain, count only within this admin's domains
        List<Domain> domains = prov.getAllDomains();
        AdminAccessControl aac = AdminAccessControl.getAdminAccessControl(zsc);
        AdminRight associatedRight = getAssociatedRight(countObjectsType);
        for (Iterator<Domain> it = domains.iterator(); it.hasNext(); ) {
            Domain domain = it.next();
            if (!aac.hasRight(domain, associatedRight)) {
                it.remove();
            }
        }
        count = 0;
        int threshold = DebugConfig.minimumDomainsToUseThreadsForDomainAdminCountObjects;
        if (threshold > 0 && domains.size() >= threshold) {
            // For a large number of domains, counting can be slow.  Do the LDAP queries in parallel.
            // As they all use different bases, they don't interfere with each other much.
            AtomicLong atomicCount = new AtomicLong(0);
            List<Thread> threads = Lists.newArrayList();
            final int chunkSize = (domains.size() / DebugConfig.numberOfThreadsToUseForDomainAdminCountObjects) + 1;
            int lastIndex = domains.size() - 1;
            int begin = 0;
            int end = (lastIndex < chunkSize) ? lastIndex : chunkSize - 1;
            while (end <= lastIndex) {
                threads.add(new Thread(new GetDomainCountsThread(atomicCount, prov, domains.subList(begin, end + 1), countObjectsType, ucService), String.format("%s-CountsForDomains-%d", Thread.currentThread().getName(), threads.size())));
                if (end >= lastIndex) {
                    break;
                }
                begin += chunkSize;
                end += chunkSize;
                if (end > lastIndex) {
                    end = lastIndex;
                }
            }
            for (Thread thread : threads) {
                thread.start();
            }
            for (Thread thread : threads) {
                try {
                    thread.join();
                } catch (InterruptedException e) {
                    ZimbraLog.search.debug("Unexpected exception counting for domain", e);
                }
            }
            count = atomicCount.get();
        } else {
            for (Domain domain : domains) {
                count += prov.countObjects(countObjectsType, domain, ucService);
            }
        }
    } else if (!specifiedDomains.isEmpty() && countObjectsType.allowsDomain()) {
        // count objects within specified domains
        for (DomainSelector specifiedDomain : specifiedDomains) {
            DomainBy by = specifiedDomain.getBy();
            String domValue = specifiedDomain.getKey();
            Domain domain = prov.get(Key.DomainBy.fromString(by.name()), domValue);
            if (domain == null) {
                throw AccountServiceException.NO_SUCH_DOMAIN(domValue);
            }
            checkDomainRight(zsc, domain, getAssociatedRight(countObjectsType));
            count += prov.countObjects(countObjectsType, domain, ucService);
        }
    } else {
        // count objects globally
        this.checkRight(zsc, context, null, getAssociatedRight(countObjectsType));
        count += prov.countObjects(countObjectsType, null, ucService);
    }
    return zsc.jaxbToElement(new CountObjectsResponse(count, countObjectsType.name()));
}