X-Git-Url: http://git.meshlink.io/?a=blobdiff_plain;ds=sidebyside;f=test%2Fstorage-policy.c;fp=test%2Fstorage-policy.c;h=f9593c8e13ec7e8747f3d91b1e62012747ef4541;hb=8a12cc2a26a66f6f50f79a2a12912ff5e991707d;hp=0000000000000000000000000000000000000000;hpb=3ed9219f9f7a7b0f71843aa1ca05b5d3cf5cf016;p=meshlink

diff --git a/test/storage-policy.c b/test/storage-policy.c
new file mode 100644
index 00000000..f9593c8e
--- /dev/null
+++ b/test/storage-policy.c
@@ -0,0 +1,154 @@
+#ifdef NDEBUG
+#undef NDEBUG
+#endif
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <sys/time.h>
+#include <assert.h>
+
+#include "meshlink.h"
+#include "utils.h"
+
+int main(void) {
+	meshlink_set_log_cb(NULL, MESHLINK_DEBUG, log_cb);
+
+	meshlink_handle_t *mesh1;
+	meshlink_handle_t *mesh2;
+
+	// Open two instances
+
+	assert(meshlink_destroy("storage-policy_conf.1"));
+	assert(meshlink_destroy("storage-policy_conf.2"));
+
+	mesh1 = meshlink_open("storage-policy_conf.1", "foo", "storage-policy", DEV_CLASS_BACKBONE);
+	mesh2 = meshlink_open("storage-policy_conf.2", "bar", "storage-policy", DEV_CLASS_BACKBONE);
+	assert(mesh1);
+	assert(mesh2);
+	meshlink_enable_discovery(mesh1, false);
+	meshlink_enable_discovery(mesh2, false);
+	meshlink_set_storage_policy(mesh1, MESHLINK_STORAGE_DISABLED);
+	meshlink_set_storage_policy(mesh2, MESHLINK_STORAGE_DISABLED);
+
+	// Exchange data
+
+	char *export1 = meshlink_export(mesh1);
+	char *export2 = meshlink_export(mesh2);
+
+	assert(export1);
+	assert(export2);
+
+	assert(meshlink_import(mesh1, export2));
+	assert(meshlink_import(mesh2, export1));
+
+	// Check that they know each other
+
+	assert(meshlink_get_node(mesh1, "bar"));
+	assert(meshlink_get_node(mesh2, "foo"));
+
+	start_meshlink_pair(mesh1, mesh2);
+
+	// Close the instances and reopen them.
+
+	close_meshlink_pair(mesh1, mesh2);
+
+	mesh1 = meshlink_open("storage-policy_conf.1", "foo", "storage-policy", DEV_CLASS_BACKBONE);
+	mesh2 = meshlink_open("storage-policy_conf.2", "bar", "storage-policy", DEV_CLASS_BACKBONE);
+	assert(mesh1);
+	assert(mesh2);
+	meshlink_enable_discovery(mesh1, false);
+	meshlink_enable_discovery(mesh2, false);
+	meshlink_set_storage_policy(mesh1, MESHLINK_STORAGE_KEYS_ONLY);
+	meshlink_set_storage_policy(mesh2, MESHLINK_STORAGE_KEYS_ONLY);
+
+	// Check that the nodes no longer know each other
+
+	assert(!meshlink_get_node(mesh1, "bar"));
+	assert(!meshlink_get_node(mesh2, "foo"));
+
+	// Exchange data again
+
+	assert(meshlink_import(mesh1, export2));
+	assert(meshlink_import(mesh2, export1));
+
+	free(export1);
+	free(export2);
+
+	// Close the instances and reopen them.
+
+	close_meshlink_pair(mesh1, mesh2);
+
+	mesh1 = meshlink_open("storage-policy_conf.1", "foo", "storage-policy", DEV_CLASS_BACKBONE);
+	mesh2 = meshlink_open("storage-policy_conf.2", "bar", "storage-policy", DEV_CLASS_BACKBONE);
+	assert(mesh1);
+	assert(mesh2);
+	meshlink_enable_discovery(mesh1, false);
+	meshlink_enable_discovery(mesh2, false);
+	meshlink_set_storage_policy(mesh1, MESHLINK_STORAGE_KEYS_ONLY);
+	meshlink_set_storage_policy(mesh2, MESHLINK_STORAGE_KEYS_ONLY);
+
+	// Check that the nodes know each other
+
+	assert(meshlink_get_node(mesh1, "bar"));
+	assert(meshlink_get_node(mesh2, "foo"));
+
+	// Check that we update reachability
+
+	time_t last_reachable;
+	time_t last_unreachable;
+	assert(!meshlink_get_node_reachability(mesh1, meshlink_get_node(mesh1, "bar"), &last_reachable, &last_unreachable));
+	assert(!last_reachable);
+	assert(!last_unreachable);
+
+	start_meshlink_pair(mesh1, mesh2);
+	stop_meshlink_pair(mesh1, mesh2);
+
+	assert(!meshlink_get_node_reachability(mesh1, meshlink_get_node(mesh1, "bar"), &last_reachable, &last_unreachable));
+	assert(last_reachable);
+	assert(last_unreachable);
+
+	// But have not stored it
+
+	close_meshlink_pair(mesh1, mesh2);
+
+	mesh1 = meshlink_open("storage-policy_conf.1", "foo", "storage-policy", DEV_CLASS_BACKBONE);
+	mesh2 = meshlink_open("storage-policy_conf.2", "bar", "storage-policy", DEV_CLASS_BACKBONE);
+	assert(mesh1);
+	assert(mesh2);
+	meshlink_enable_discovery(mesh1, false);
+	meshlink_enable_discovery(mesh2, false);
+	meshlink_set_storage_policy(mesh1, MESHLINK_STORAGE_KEYS_ONLY);
+	meshlink_set_storage_policy(mesh2, MESHLINK_STORAGE_KEYS_ONLY);
+
+	assert(meshlink_get_node(mesh1, "bar"));
+	assert(meshlink_get_node(mesh2, "foo"));
+
+	assert(!meshlink_get_node_reachability(mesh1, meshlink_get_node(mesh1, "bar"), &last_reachable, &last_unreachable));
+	assert(!last_reachable);
+	assert(!last_unreachable);
+
+	// Check that if we change back to STORAGE_ENABLED right before closing, pending changes are still saved
+
+	start_meshlink_pair(mesh1, mesh2);
+	stop_meshlink_pair(mesh1, mesh2);
+
+	meshlink_set_storage_policy(mesh1, MESHLINK_STORAGE_ENABLED);
+	meshlink_set_storage_policy(mesh2, MESHLINK_STORAGE_ENABLED);
+
+	close_meshlink_pair(mesh1, mesh2);
+
+	mesh1 = meshlink_open("storage-policy_conf.1", "foo", "storage-policy", DEV_CLASS_BACKBONE);
+	mesh2 = meshlink_open("storage-policy_conf.2", "bar", "storage-policy", DEV_CLASS_BACKBONE);
+	assert(mesh1);
+	assert(mesh2);
+	meshlink_enable_discovery(mesh1, false);
+	meshlink_enable_discovery(mesh2, false);
+
+	assert(!meshlink_get_node_reachability(mesh1, meshlink_get_node(mesh1, "bar"), &last_reachable, &last_unreachable));
+	assert(last_reachable);
+	assert(last_unreachable);
+
+	// That's it.
+
+	close_meshlink_pair(mesh1, mesh2);
+}