test: Add sequence_numbers

Change-Id: Ida6123bf27445f0fb65099fbf6d9520344d33f10
Signed-off-by: Andrew Jeffery <andrew@aj.id.au>
diff --git a/Makefile.am b/Makefile.am
index 735efd8..700a330 100644
--- a/Makefile.am
+++ b/Makefile.am
@@ -89,6 +89,9 @@
 test_read_window_cycle_SOURCES = test/read_window_cycle.c \
 			     $(TEST_MBOX_SRCS) $(TEST_MOCK_SRCS)
 
+test_sequence_numbers_SOURCES = test/sequence_numbers.c \
+			     $(TEST_MBOX_SRCS) $(TEST_MOCK_SRCS)
+
 check_PROGRAMS = test/sanity \
 		 test/copy_flash \
 		 test/erase_flash \
@@ -113,6 +116,7 @@
 		 test/read_window_mark_write_erased \
 		 test/write_window_dirty_erase \
 		 test/invalid_command \
-		 test/read_window_cycle
+		 test/read_window_cycle \
+		 test/sequence_numbers
 
 TESTS = $(check_PROGRAMS)
diff --git a/test/sequence_numbers.c b/test/sequence_numbers.c
new file mode 100644
index 0000000..be95d87
--- /dev/null
+++ b/test/sequence_numbers.c
@@ -0,0 +1,107 @@
+/*
+ * 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 "mbox.h"
+#include "mboxd_msg.h"
+
+#include "test/mbox.h"
+#include "test/system.h"
+
+static const uint8_t get_mbox_info[] = {
+	0x02, 0x00, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00,
+	0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
+};
+
+static const uint8_t get_mbox_info_response[] = {
+	0x02, 0x00, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00,
+	0x00, 0x00, 0x00, 0x00, 0x00, 0x01,
+};
+
+static const uint8_t get_flash_info0[] = {
+	0x03, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+	0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
+};
+
+static const uint8_t get_flash_info_response0[] = {
+	0x03, 0x01, 0x03, 0x00, 0x01, 0x00, 0x00, 0x00,
+	0x00, 0x00, 0x00, 0x00, 0x00, 0x01
+};
+
+static const uint8_t get_flash_info1[] = {
+	0x03, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
+	0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00
+};
+
+static const uint8_t get_flash_info_response1[] = {
+	0x03, 0x02, 0x03, 0x00, 0x01, 0x00, 0x00, 0x00,
+	0x00, 0x00, 0x00, 0x00, 0x00, 0x01
+};
+
+#define MEM_SIZE	3
+#define ERASE_SIZE	1
+#define N_WINDOWS	1
+#define WINDOW_SIZE	1
+
+int main(void)
+{
+	struct mbox_context *ctx;
+	int rc;
+
+	system_set_reserved_size(MEM_SIZE);
+	system_set_mtd_sizes(MEM_SIZE, ERASE_SIZE);
+
+	ctx = mbox_create_test_context(N_WINDOWS, WINDOW_SIZE);
+
+	/* Consecutive GET_MBOX_INFOs can use "invalid" sequence numbers */
+	rc = mbox_command_dispatch(ctx, get_mbox_info, sizeof(get_mbox_info));
+	assert(rc == 1);
+
+	rc = mbox_cmp(ctx, get_mbox_info_response,
+			sizeof(get_mbox_info_response));
+	assert(rc == 0);
+
+	rc = mbox_command_dispatch(ctx, get_mbox_info, sizeof(get_mbox_info));
+	assert(rc == 1);
+
+	/* Other commands must use valid sequence numbers */
+	rc = mbox_command_dispatch(ctx, get_flash_info0,
+			sizeof(get_flash_info0));
+	assert(rc == 1);
+
+	rc = mbox_cmp(ctx, get_flash_info_response0,
+			sizeof(get_flash_info_response0));
+	assert(rc == 0);
+
+	rc = mbox_command_dispatch(ctx, get_flash_info0,
+			sizeof(get_flash_info0));
+	assert(rc == 8);
+
+	/* Retry with a "valid" sequence number */
+	rc = mbox_command_dispatch(ctx, get_flash_info1,
+			sizeof(get_flash_info1));
+	assert(rc == 1);
+
+	rc = mbox_cmp(ctx, get_flash_info_response1,
+			sizeof(get_flash_info_response1));
+	assert(rc == 0);
+
+	return rc;
+}