@Test public void testDeleteDomain() throws Exception { int custId = createCustomer("testDeleteDomain"); int policyId = createMailRoutingPolicy(custId, "testMailRoutingPolicy", true); List<String> custDomains = createCustomerDomains(custId, policyId); CustomerDomainDAO dao = createDao(); Connection conn = null; try { conn = getDBConnection(); List<CustomerDomain> domains = dao.getDomainsForCustomer(custId, conn); assertEquals("Wrong number of domains returned.", custDomains.size(), domains.size()); for (String dom : custDomains) { assertTrue("Failed to delete domain " + dom, dao.deleteDomain(custId, dom, conn)); } domains = dao.getDomainsForCustomer(custId, conn); assertEquals("Wrong number of domains returned.", 0, domains.size()); } finally { if (conn != null) { conn.close(); } } }