使用指定的记录创建DBUnit数据集

Question

我正在使用DBUnit来创建仅包含具有正确ID的记录的数据集。

Set ids = new HashSet();
ids.add(new BigDecimal(34552));
TablesDependencyHelper.getDataset(connection, "MYTABLE", ids);

但我得到的数据集包含该表中的所有记录（+来自某些相关表的记录）。

我也对复合主键感到困惑。例如，我将ID1和ID2列作为主键 - 在这种情况下如何创建数据集（仅限指定的记录）？

Answer 1

一段时间后，我意识到使用DBUnit进行此操作的唯一方法是：

QueryDataSet queryDataSet = new QueryDataSet(connection);
queryDataSet.addTable("MYTABLE", "SELECT * FROM MYTABLE WHERE ID=34552");