+
+ test_add_file_data(test, test->ctx->device, conf_file, buf, size);
+}
+
+void test_add_file_data(struct parser_test *test, struct discover_device *dev,
+ const char *filename, const void *data, int size)
+{
+ struct test_file *file;
+
+ file = talloc_zero(test, struct test_file);
+ file->type = TEST_FILE;
+ file->dev = dev;
+ file->name = filename;
+ file->data = talloc_memdup(test, data, size);
+ file->size = size;
+ list_add(&test->files, &file->list);
+}
+
+void test_add_dir(struct parser_test *test, struct discover_device *dev,
+ const char *dirname)
+{
+ struct test_file *file;
+
+ file = talloc_zero(test, struct test_file);
+ file->type = TEST_DIR;
+ file->dev = dev;
+ file->name = dirname;
+ /* Pick a non-zero size for directories so that "[ -s <dir
+ * path> ]" sees that the file has non-zero size. */
+ file->size = 1;
+ list_add(&test->files, &file->list);
+}
+
+void test_set_event_source(struct parser_test *test)
+{
+ test->ctx->event = talloc_zero(test->ctx, struct event);
+}
+
+void test_set_event_param(struct event *event, const char *name,
+ const char *value)
+{
+ event_set_param(event, name, value);
+}
+
+void test_set_event_device(struct event *event, const char *dev)
+{
+ event->device = talloc_strdup(event, dev);
+}
+
+int parser_request_file(struct discover_context *ctx,
+ struct discover_device *dev, const char *filename,
+ char **buf, int *len)
+{
+ struct parser_test *test = ctx->test_data;
+ struct test_file *file;
+ char *tmp;
+
+ list_for_each_entry(&test->files, file, list) {
+ if (file->dev != dev)
+ continue;
+ if (strcmp(file->name, filename))
+ continue;
+ if (file->type != TEST_FILE)
+ continue;
+
+ /* the read_file() interface always adds a trailing null
+ * for string-safety; do the same here */
+ tmp = talloc_array(test, char, file->size + 1);
+ memcpy(tmp, file->data, file->size);
+ tmp[file->size] = '\0';
+ *buf = tmp;
+ *len = file->size;
+ return 0;
+ }
+
+ return -1;
+}
+
+int parser_stat_path(struct discover_context *ctx,
+ struct discover_device *dev, const char *path,
+ struct stat *statbuf)
+{
+ struct parser_test *test = ctx->test_data;
+ struct test_file *file;
+
+ list_for_each_entry(&test->files, file, list) {
+ if (file->dev != dev)
+ continue;
+ if (strcmp(file->name, path))
+ continue;
+
+ statbuf->st_size = (off_t)file->size;
+ switch (file->type) {
+ case TEST_FILE:
+ statbuf->st_mode = S_IFREG;
+ break;
+ case TEST_DIR:
+ statbuf->st_mode = S_IFDIR;
+ break;
+ default:
+ fprintf(stderr, "%s: bad test file mode %d!", __func__,
+ file->type);
+ exit(EXIT_FAILURE);
+ }
+
+ return 0;
+ }
+
+ return -1;
+}
+
+int parser_replace_file(struct discover_context *ctx,
+ struct discover_device *dev, const char *filename,
+ char *buf, int len)
+{
+ struct parser_test *test = ctx->test_data;
+ struct test_file *f, *file;
+
+ list_for_each_entry(&test->files, f, list) {
+ if (f->dev != dev)
+ continue;
+ if (strcmp(f->name, filename))
+ continue;
+
+ file = f;
+ break;
+ }
+
+ if (!file) {
+ file = talloc_zero(test, struct test_file);
+ file->dev = dev;
+ file->name = filename;
+ list_add(&test->files, &file->list);
+ }
+
+ file->data = talloc_memdup(test, buf, len);
+ file->size = len;
+ return 0;
+}
+
+struct load_url_result *load_url_async(void *ctx, struct pb_url *url,
+ load_url_complete async_cb, void *async_data,
+ waiter_cb stdout_cb, void *stdout_data)
+{
+ struct conf_context *conf = async_data;
+ struct parser_test *test = conf->dc->test_data;
+ struct load_url_result *result;
+ char tmp[] = "/tmp/pb-XXXXXX";
+ ssize_t rc = -1, sz = 0;
+ struct test_file *file;
+ int fd;
+
+ /* Ignore the stdout callback for tests */
+ (void)stdout_cb;
+ (void)stdout_data;
+
+ fd = mkstemp(tmp);
+
+ if (fd < 0)
+ return NULL;
+
+ /* Some parsers will expect to need to read a file, so write the
+ * specified file to a temporary file */
+ list_for_each_entry(&test->files, file, list) {
+ if (file->dev)
+ continue;
+
+ if (strcmp(file->name, url->full))
+ continue;
+
+ while (sz < file->size) {
+ rc = write(fd, file->data, file->size);
+ if (rc < 0) {
+ fprintf(stderr,
+ "Failed to write to tmpfile, %m\n");
+ break;
+ }
+ sz += rc;
+ }
+ break;
+ }
+
+ close(fd);
+
+ result = talloc_zero(ctx, struct load_url_result);
+ if (!result)
+ return NULL;
+
+ result->local = talloc_strdup(result, tmp);
+ result->url = url;
+ if (rc < 0)
+ result->status = LOAD_ERROR;
+ else
+ result->status = result->local ? LOAD_OK : LOAD_ERROR;
+ result->cleanup_local = true;
+
+ async_cb(result, conf);
+
+ return result;