Question

I have the following Entity:

    @Entity
    @Table
    @NamedQuery(name = Constants.FINDALLFINDERNAME,
                query = Constants.FINDALLQUERY)
    public class Customer extends AuditableEntity {

        ...

        @ElementCollection(fetch = FetchType.EAGER)
        @CollectionTable(name = Constants.PHONES, joinColumns = @JoinColumn(name = Constants.CUSTOMER_ID))
        @Column(name = Constants.CUSTOMER_PHONES, nullable = true)
        private List<Phone> phones;

I want to import the following data into my schema during a test:

    <?xml version='1.0' encoding='UTF-8'?>
    <dataset>
        <CUSTOMER id='101' USERNAME="user1" FIRSTNAME="Mick" LASTNAME="Knutson"/>

        <PHONES AREACODE="415" P_NUMBER="5551212" TYPE="WORK" CUST_ID="101" />
    </dataset>

But I get this error:

    [EL Info]: 2011-01-31 10:57:56.945--ClientSession(30624226)--Communication failure detected when attempting to create transaction on database.  Attempting to retry begin transaction. Error was: Exception [EclipseLink-4002] (Eclipse Persistence Services - 2.3.0.v20110129-r8902): org.eclipse.persistence.exceptions.DatabaseException
    Internal Exception: org.h2.jdbc.JdbcSQLException: The object is already closed [90007-148]
    Error Code: 90007.
    Dataset written

    org.dbunit.dataset.NoPrimaryKeyException: PHONES

Here is my unit test:

    ...

    public class CustomerTest {

        //-----------------------------------------------------------------------//
        // Attributes
        //-----------------------------------------------------------------------//
        private static EntityManagerFactory emf;
        private static EntityManager em;
        private static EntityTransaction tx;

        //-----------------------------------------------------------------------//
        // Lifecycle Methods
        //-----------------------------------------------------------------------//
        @BeforeClass
        public static void initEntityManager() throws Exception {
            emf = Persistence.createEntityManagerFactory(Constants.PERSISTENCEUNIT);
            em = emf.createEntityManager();
        }

        @AfterClass
        public static void closeEntityManager() throws SQLException {
            if (em != null) em.close();
            if (emf != null) emf.close();
        }

        @Before
        public void initTransaction() throws Exception {
            tx = em.getTransaction();
            seedData();
        }

        @After
        public void afterTests() throws Exception {
            dumpData();
        }

        //-----------------------------------------------------------------------//
        // Unit Tests
        //-----------------------------------------------------------------------//

        @Test
        public void test__Create__and__Read_SingleCustomer() throws Exception {
            // Creates an instance of Customer
            Customer customer = CustomerFixture.createSingleCustomer();

            // Persists the Customer to the database
            tx.begin();
            em.persist(customer);
            tx.commit();

            tx.begin();
            assertNotNull("ID should not be null", customer.getId());
            // Retrieves a single Customer from the database
            TypedQuery<Customer> q = em.createNamedQuery(
                    Constants.FINDALLFINDERNAME, Customer.class);
            List<Customer> customers = q.getResultList();
            assertThat(customers.size(), is(4));
            tx.commit();
        }

        @Test
        public void test__DeleteCustomer() throws Exception {
            tx.begin();

            // Uses Sting Based Criteria
            CriteriaBuilder cb = em.getCriteriaBuilder();
            CriteriaQuery<Customer> c = cb.createQuery(Customer.class);
            Root<Customer> cust = c.from(Customer.class);
            c.select(cust)
                    .where(cb.equal(cust.get("username"), "user1"));
            Customer result = em.createQuery(c).getSingleResult();

            em.remove(result);

            // Retrieves all the Customers from the database
            TypedQuery<Customer> q = em.createNamedQuery(
                    Constants.FINDALLFINDERNAME, Customer.class);
            List<Customer> customers = q.getResultList();

            tx.commit();

            assertThat(customers.size(), is(3));
        }                    

        //-----------------------------------------------------------------------//
        // DBUnit Helper Methods
        //-----------------------------------------------------------------------//

        protected void seedData() throws Exception {
            tx.begin();
            Connection connection = em.unwrap(java.sql.Connection.class);

            try {
                IDatabaseConnection dbUnitCon = new DatabaseConnection(connection);

                dbUnitCon.getConfig().setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY,
                        new H2DataTypeFactory());

                IDataSet dataSet = getDataSet();

                DatabaseOperation.REFRESH.execute(dbUnitCon, dataSet);
            } finally {
                tx.commit();
                connection.close();
            }
        }

        protected IDataSet getDataSet() throws Exception {
            return new FlatXmlDataSetBuilder().build(new FileInputStream("./src/test/resources/dataset.xml"));
        }

        protected void dumpData() throws Exception {
            tx.begin();
            Connection connection = em.unwrap(java.sql.Connection.class);

            try {
                IDatabaseConnection dbUnitCon = new DatabaseConnection(connection);
                dbUnitCon.getConfig().setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY,
                        new H2DataTypeFactory());

                IDataSet dataSet = dbUnitCon.createDataSet();

                FlatXmlDataSet.write(dataSet, new FileOutputStream("./target/test-dataset_dump.xml"));
                System.out.println("Dataset written");
            } finally {
                tx.commit();
                connection.close();
            }
        }
    }

Can someone help me to understand how to import CollectionTables with DBUnit?

Answer 1

I can across enter link description here and tried the solution, but modified it a bit:

The property:

        // Set the property by passing the new IColumnFilter
        dbUnitCon.getConfig().setProperty(
                DatabaseConfig.PROPERTY_PRIMARY_KEY_FILTER,
                new NullPrimaryKeyFilter("ID", "ADDRESS_KEY", "P_NUMBER", "HOBBY_NAME"));

then the Filter Class:

    class NullPrimaryKeyFilter implements IColumnFilter {
        private String[] keys = null;

        NullPrimaryKeyFilter(String... keys) {
            this.keys = keys;
        }

        public boolean accept(String tableName, Column column) {
            for(String key: keys){
                if(column.getColumnName().equalsIgnoreCase(key)){
                    return true;
                }
            }
            return false;
        }
    }

Now I can use the filter for all my TABLES in my test.