X-Git-Url: http://git.ozlabs.org/?p=petitboot;a=blobdiff_plain;f=test%2Fparser%2Futils.c;h=80117937e56f2f598e8bf211a2d3235d8072d26d;hp=b80e0e102025430f4ebaece59540481d44fb5814;hb=66bf929fa97fabb70198455f9eec2e9ccf0661fa;hpb=cc28b17bb59411e6031a8273b784e9d1d3a9490d

diff --git a/test/parser/utils.c b/test/parser/utils.c
index b80e0e1..8011793 100644
--- a/test/parser/utils.c
+++ b/test/parser/utils.c
@@ -301,6 +301,21 @@ void test_hotplug_device(struct parser_test *test, struct discover_device *dev)
 		boot_option_resolve(test->handler, opt);
 }
 
+void test_remove_device(struct parser_test *test, struct discover_device *dev)
+{
+	struct discover_boot_option *opt, *tmp;
+
+	if (dev == test->ctx->device) {
+		list_for_each_entry_safe(&test->ctx->boot_options,
+				opt, tmp, list) {
+			list_remove(&opt->list);
+			talloc_free(opt);
+		}
+	}
+
+	device_handler_remove(test->handler, dev);
+}
+
 struct discover_boot_option *get_boot_option(struct discover_context *ctx,
 		int idx)
 {