DBUnit: Composite Unique Key

[Sam Mascarenhas]

Hello,

Trying to insert into a database table with DBunit however, since the table doesn't explisitly define a primary key. The table defines a unique composite key. DBunit fails while execution with a "org.dbunit.dataset.NoPrimaryKeyException"... googled on this one and came across the following helpful site - http://dbunit.wikidot.com/noprimarykeytable...

With reference to the sample provided on this site... the issue is that it only supports defining a single primary key field.......Similarly, how do we provide it with a composite key...

Please advise.

Sam

[Dragan Stankovic]

I've bumped on this question while having the same problem. Didn't find much resources about this so I decided to share what I eventually came up with. I've found some old post from 2005 (http://www.systemmobile.com/?p=102) which gave me the idea that worked. Take a look at the following code sample which shows how to do it for composite keys:

		// ....
		IDatabaseConnection connection = new DatabaseConnection(db.getConnection());
		DatabaseConfig config = connection.getConfig();
		config.setProperty(DatabaseConfig.PROPERTY_DATATYPE_FACTORY, new MySqlDataTypeFactory());
		config.setProperty(DatabaseConfig.PROPERTY_METADATA_HANDLER, new MySqlMetadataHandler());
		config.setProperty(DatabaseConfig.PROPERTY_ESCAPE_PATTERN, "`?`");
		config.setProperty(DatabaseConfig.PROPERTY_PRIMARY_KEY_FILTER, new IColumnFilter() {

			Map<String, List<String>> tablePrimaryKeyMap = new HashMap<String, List<String>>();
			{
				tablePrimaryKeyMap.put("table_with_composite1",
						Arrays.asList(new String[] { "composite_part1", "composite_part2" }));
				tablePrimaryKeyMap.put("table_with_composite2",
						Arrays.asList(new String[] { "other_part1", "other_part2", "other_part3" }));
				// ...
			}

			@Override
			public boolean accept(String tableName, Column column) {
				if (tablePrimaryKeyMap.containsKey(tableName)) {
					return tablePrimaryKeyMap.get(tableName).contains(column.getColumnName());
				} else {
					// you can add default handling if for example all
					// other table ids are named as 'id'
					return column.getColumnName().equalsIgnoreCase("id");
				}
			}
		});

Hope that this helps (at least to someone who bumps as I did).
Edit: I wrote an article with more details as reminder on my blog. Here it is dbunit with composite primary keys