diff --git a/test/erase_flash.c b/test/erase_flash.c
new file mode 100644
index 0000000..23a536f
--- /dev/null
+++ b/test/erase_flash.c
@@ -0,0 +1,224 @@
+/*
+ * MBox Daemon Test File
+ *
+ * Copyright 2017 IBM
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ */
+
+#include <assert.h>
+#include <stdarg.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <sys/ioctl.h>
+#include <unistd.h>
+
+#include <linux/types.h>
+
+#include "mbox.h"
+#include "mboxd_flash.h"
+
+#include "test/tmpf.h"
+
+static struct tmpf mtd;
+
+void cleanup_mtd(void)
+{
+	tmpf_destroy(&mtd);
+}
+
+char *get_dev_mtd(void)
+{
+	int rc;
+
+	rc = tmpf_init(&mtd, "flashXXXXXX");
+	if (rc < 0)
+		return NULL;
+
+	return strdup(mtd.path);
+}
+
+struct erase_info_user *recorded;
+int n_ioctls;
+
+#define MEM_SIZE 3
+#define ERASE_SIZE 1
+
+int ioctl(int fd, unsigned long request, ...)
+{
+	va_list ap;
+	struct erase_info_user *provided, *alloced;
+
+	if (!(request == MEMERASE || request == MEMGETINFO)) {
+		printf("Uh-oh: ioctl() called with request 0x%08lx\n", request);
+		return -1;
+	}
+
+	switch (request) {
+	case MEMGETINFO:
+	{
+		struct mtd_info_user *info;
+
+		va_start(ap, request);
+		info = va_arg(ap, struct mtd_info_user *);
+		info->size = MEM_SIZE;
+		info->erasesize = ERASE_SIZE;
+		va_end(ap);
+		break;
+	}
+	case MEMERASE:
+		va_start(ap, request);
+		provided = va_arg(ap, struct erase_info_user *);
+
+		n_ioctls++;
+
+		alloced = realloc(recorded, n_ioctls * sizeof(*recorded));
+		if (!alloced)
+			return -1;
+		recorded = alloced;
+
+		memcpy(&recorded[n_ioctls - 1], provided, sizeof(*provided));
+
+		va_end(ap);
+		break;
+	default:
+		break;
+	}
+
+	return 0;
+}
+
+void dump_ioctls(void)
+{
+	int i;
+
+	printf("n_ioctls: %d\n", n_ioctls);
+
+	for (i = 0; i < n_ioctls; i++)
+		printf("%d: start: %d, length %d\n",
+				i, recorded[i].start, recorded[i].length);
+}
+
+int main(void)
+{
+	struct mbox_context context;
+	char data[MEM_SIZE];
+	int rc;
+
+	rc = atexit(cleanup_mtd);
+	if (rc)
+		return rc;
+
+	n_ioctls = 0;
+	recorded = NULL;
+
+	init_flash_dev(&context);
+
+	/* Erase from an unknown state */
+	rc = erase_flash(&context, 0, sizeof(data));
+
+	assert(rc == 0);
+	assert(n_ioctls == 1);
+	assert(recorded[0].start == 0);
+	assert(recorded[0].length == sizeof(data));
+
+	free(recorded);
+	recorded = NULL;
+	n_ioctls = 0;
+
+	/* Erase an erased flash */
+	rc = erase_flash(&context, 0, sizeof(data));
+
+	assert(rc == 0);
+	assert(n_ioctls == 0);
+
+	memset(data, 0xaa, sizeof(data));
+
+	/* Erase written flash */
+	rc = write_flash(&context, 0, data, sizeof(data));
+	assert(rc == 0);
+	rc = erase_flash(&context, 0, sizeof(data));
+
+	assert(rc == 0);
+	assert(n_ioctls == 1);
+	assert(recorded[0].start == 0);
+	assert(recorded[0].length == sizeof(data));
+
+	free(recorded);
+	recorded = NULL;
+	n_ioctls = 0;
+
+	/* Erase the start of flash */
+	rc = write_flash(&context, 0, data, sizeof(data) - 1);
+	assert(rc == 0);
+	rc = erase_flash(&context, 0, sizeof(data));
+
+	assert(rc == 0);
+	assert(n_ioctls == 1);
+	assert(recorded[0].start == 0);
+	assert(recorded[0].length == sizeof(data) - 1);
+
+	free(recorded);
+	recorded = NULL;
+	n_ioctls = 0;
+
+	/* Erase the end of flash */
+	rc = write_flash(&context, 1, data, sizeof(data) - 1);
+	assert(rc == 0);
+	rc = erase_flash(&context, 0, sizeof(data));
+
+	assert(rc == 0);
+	assert(n_ioctls == 1);
+	assert(recorded[0].start == 1);
+	assert(recorded[0].length == sizeof(data) - 1);
+
+	free(recorded);
+	recorded = NULL;
+	n_ioctls = 0;
+
+	/* Erase each end of flash */
+	rc = write_flash(&context, 0, data, 1);
+	rc = write_flash(&context, 2, data, 1);
+	assert(rc == 0);
+	rc = erase_flash(&context, 0, sizeof(data));
+
+	assert(rc == 0);
+	assert(n_ioctls == 2);
+	assert(recorded[0].start == 0);
+	assert(recorded[0].length == 1);
+	assert(recorded[1].start == 2);
+	assert(recorded[1].length == 1);
+
+	free(recorded);
+	recorded = NULL;
+	n_ioctls = 0;
+
+	/* Erase the middle of flash */
+	rc = write_flash(&context, 1, data, 1);
+	assert(rc == 0);
+	rc = erase_flash(&context, 0, sizeof(data));
+
+	assert(rc == 0);
+	assert(n_ioctls == 1);
+	assert(recorded[0].start == 1);
+	assert(recorded[0].length == 1);
+
+	free(recorded);
+	recorded = NULL;
+	n_ioctls = 0;
+
+	free_flash_dev(&context);
+
+	return rc;
+}