1 files changed, 249 insertions, 65 deletions

diff --git a/test/test_env.c b/test/test_env.c
index 69b69061..d2c80343 100644
--- a/test/test_env.c
+++ b/test/test_env.c
@@ -13,6 +13,8 @@
 #include <assert.h>
 #include <string.h>
 
+#define NS_EG "http://example.org/"
+
 static void
 test_copy(void)
 {
@@ -54,113 +56,295 @@ test_comparison(void)
   serd_env_free(env);
 }
 
+static void
+test_null(void)
+{
+  SerdNode* const eg = serd_new_uri(serd_string(NS_EG));
+
+  // "Copying" NULL returns null
+  assert(!serd_env_copy(NULL));
+
+  // Accessors are tolerant to a NULL env for convenience
+  assert(!serd_env_base_uri(NULL));
+  assert(!serd_env_expand(NULL, NULL));
+  assert(!serd_env_qualify(NULL, eg));
+
+  // Only null is equal to null
+  assert(serd_env_equals(NULL, NULL));
+
+  serd_node_free(eg);
+}
+
 static SerdStatus
 count_prefixes(void* handle, const SerdEvent* event)
 {
-  if (event->type == SERD_PREFIX) {
-    ++*(int*)handle;
-  }
+  *(int*)handle += event->type == SERD_PREFIX;
 
   return SERD_SUCCESS;
 }
 
 static void
-test_env(void)
+test_base_uri(void)
 {
-  const SerdStringView prefix = serd_string("eg.2");
-  const SerdStringView eg     = serd_string("http://example.org/");
-
-  SerdNode* hello = serd_new_string(serd_string("hello\""));
-  SerdNode* rel   = serd_new_uri(serd_string("rel"));
-  SerdNode* foo_u = serd_new_uri(serd_string("http://example.org/foo"));
-  SerdNode* foo_c = serd_new_curie(serd_string("eg.2:foo"));
-  SerdNode* b     = serd_new_curie(serd_string("invalid"));
-
-  SerdEnv* env = serd_env_new(serd_empty_string());
-
-  serd_env_set_prefix(env, prefix, eg);
-
-  assert(!serd_env_base_uri(NULL));
-  assert(!serd_env_expand(NULL, NULL));
-  assert(!serd_env_expand(NULL, foo_c));
-  assert(!serd_env_expand(env, NULL));
-  assert(!serd_env_qualify(NULL, NULL));
-  assert(!serd_env_qualify(env, NULL));
-  assert(!serd_env_qualify(NULL, foo_u));
+  SerdEnv* const  env = serd_env_new(serd_empty_string());
+  SerdNode* const eg  = serd_new_uri(serd_string(NS_EG));
 
+  // Test that invalid calls work as expected
   assert(!serd_env_base_uri(env));
   assert(!serd_env_set_base_uri(env, serd_empty_string()));
   assert(!serd_env_base_uri(env));
 
-  serd_env_set_prefix(
-    env, serd_string("eg.2"), serd_string("http://example.org/"));
-
+  // Try setting a relative prefix with no base URI
   assert(serd_env_set_prefix(env, serd_string("eg.3"), serd_string("rel")) ==
          SERD_BAD_ARG);
 
-  SerdNode* xnode = serd_env_expand(env, hello);
-  assert(!xnode);
-
-  assert(!serd_env_expand(env, b));
-  assert(!serd_env_expand(env, hello));
+  // Set a valid base URI
+  assert(!serd_env_set_base_uri(env, serd_node_string_view(eg)));
+  assert(serd_node_equals(serd_env_base_uri(env), eg));
 
+  // Reset the base URI
   assert(!serd_env_set_base_uri(env, serd_empty_string()));
+  assert(!serd_env_base_uri(env));
 
-  SerdNode* xu = serd_env_expand(env, foo_c);
-  assert(!strcmp(serd_node_string(xu), "http://example.org/foo"));
-  serd_node_free(xu);
+  serd_env_free(env);
+  serd_node_free(eg);
+}
 
-  SerdNode* badpre  = serd_new_curie(serd_string("hm:what"));
-  SerdNode* xbadpre = serd_env_expand(env, badpre);
-  assert(!xbadpre);
-  serd_node_free(badpre);
+static void
+test_set_prefix(void)
+{
+  const SerdStringView eg    = serd_string(NS_EG);
+  const SerdStringView name1 = serd_string("eg.1");
+  const SerdStringView name2 = serd_string("eg.2");
+  const SerdStringView rel   = serd_string("rel");
+  const SerdStringView base  = serd_string("http://example.org/");
 
-  SerdNode* xc = serd_env_expand(env, foo_c);
-  assert(serd_node_equals(xc, foo_u));
-  serd_node_free(xc);
+  SerdEnv* const env = serd_env_new(serd_empty_string());
 
-  SerdNode* blank = serd_new_blank(serd_string("b1"));
-  assert(!serd_env_expand(env, blank));
-  serd_node_free(blank);
+  // Set a valid prefix
+  assert(!serd_env_set_prefix(env, name1, eg));
+
+  // Test setting a prefix from a relative URI
+  assert(serd_env_set_prefix(env, name2, rel) == SERD_BAD_ARG);
+  assert(!serd_env_set_base_uri(env, base));
+  assert(!serd_env_set_prefix(env, name2, rel));
+
+  // Test setting a prefix from strings
+  assert(!serd_env_set_prefix(
+    env, serd_string("eg.3"), serd_string("http://example.org/three")));
 
   size_t          n_prefixes = 0;
   SerdSink* const count_prefixes_sink =
     serd_sink_new(&n_prefixes, count_prefixes, NULL);
 
-  serd_env_set_prefix(env, prefix, eg);
   serd_env_write_prefixes(env, count_prefixes_sink);
-  assert(n_prefixes == 1);
+  serd_sink_free(count_prefixes_sink);
+  assert(n_prefixes == 3);
 
-  SerdNode* qualified = serd_env_qualify(env, foo_u);
-  assert(serd_node_equals(qualified, foo_c));
-  serd_node_free(qualified);
+  serd_env_free(env);
+}
 
-  SerdNode* unqualifiable = serd_new_uri(serd_string("http://drobilla.net/"));
-  assert(!serd_env_qualify(env, unqualifiable));
-  serd_node_free(unqualifiable);
+static void
+test_expand_untyped_literal(void)
+{
+  SerdNode* const untyped = serd_new_string(serd_string("data"));
+  SerdEnv* const  env     = serd_env_new(serd_empty_string());
 
-  assert(!serd_env_expand(env, rel));
-  serd_env_set_base_uri(env, serd_string("http://example.org/base/"));
+  assert(!serd_env_expand(env, untyped));
 
-  SerdNode* xrel = serd_env_expand(env, rel);
-  assert(xrel);
-  assert(!strcmp(serd_node_string(xrel), "http://example.org/base/rel"));
-  serd_node_free(xrel);
+  serd_env_free(env);
+  serd_node_free(untyped);
+}
 
-  serd_sink_free(count_prefixes_sink);
-  serd_node_free(b);
-  serd_node_free(foo_c);
-  serd_node_free(foo_u);
+static void
+test_expand_bad_uri_datatype(void)
+{
+  const SerdStringView type = serd_string("Type");
+
+  SerdNode* const typed = serd_new_typed_literal(serd_string("data"), type);
+  SerdEnv* const  env   = serd_env_new(serd_empty_string());
+
+  assert(!serd_env_expand(env, typed));
+
+  serd_env_free(env);
+  serd_node_free(typed);
+}
+
+static void
+test_expand_uri(void)
+{
+  const SerdStringView base = serd_string("http://example.org/b/");
+
+  SerdEnv* const  env       = serd_env_new(base);
+  SerdNode* const rel       = serd_new_uri(serd_string("rel"));
+  SerdNode* const rel_out   = serd_env_expand(env, rel);
+  SerdNode* const empty     = serd_new_uri(serd_empty_string());
+  SerdNode* const empty_out = serd_env_expand(env, empty);
+
+  assert(!strcmp(serd_node_string(rel_out), "http://example.org/b/rel"));
+  assert(!strcmp(serd_node_string(empty_out), "http://example.org/b/"));
+
+  serd_node_free(empty_out);
+  serd_node_free(empty);
+  serd_node_free(rel_out);
   serd_node_free(rel);
-  serd_node_free(hello);
   serd_env_free(env);
 }
 
+static void
+test_expand_empty_uri_ref(void)
+{
+  const SerdStringView base = serd_string("http://example.org/b/");
+
+  SerdNode* const rel     = serd_new_uri(serd_string("rel"));
+  SerdEnv* const  env     = serd_env_new(base);
+  SerdNode* const rel_out = serd_env_expand(env, rel);
+
+  assert(!strcmp(serd_node_string(rel_out), "http://example.org/b/rel"));
+  serd_node_free(rel_out);
+
+  serd_env_free(env);
+  serd_node_free(rel);
+}
+
+static void
+test_expand_bad_uri(void)
+{
+  SerdNode* const bad_uri = serd_new_uri(serd_string("rel"));
+  SerdEnv* const  env     = serd_env_new(serd_empty_string());
+
+  assert(!serd_env_expand(env, bad_uri));
+
+  serd_env_free(env);
+  serd_node_free(bad_uri);
+}
+
+static void
+test_expand_curie(void)
+{
+  const SerdStringView name = serd_string("eg.1");
+  const SerdStringView eg   = serd_string(NS_EG);
+
+  SerdNode* const curie = serd_new_curie(serd_string("eg.1:foo"));
+  SerdEnv* const  env   = serd_env_new(serd_empty_string());
+
+  assert(!serd_env_set_prefix(env, name, eg));
+
+  SerdNode* const curie_out = serd_env_expand(env, curie);
+  assert(curie_out);
+  assert(!strcmp(serd_node_string(curie_out), "http://example.org/foo"));
+  serd_node_free(curie_out);
+
+  serd_env_free(env);
+  serd_node_free(curie);
+}
+
+static void
+test_expand_bad_curie(void)
+{
+  SerdNode* const curie = serd_new_curie(serd_string("eg.1:foo"));
+  SerdEnv* const  env   = serd_env_new(serd_empty_string());
+
+  assert(!serd_env_expand(env, curie));
+
+  serd_env_free(env);
+  serd_node_free(curie);
+}
+
+static void
+test_expand_blank(void)
+{
+  SerdNode* const blank = serd_new_blank(serd_string("b1"));
+  SerdEnv* const  env   = serd_env_new(serd_empty_string());
+
+  assert(!serd_env_expand(env, blank));
+
+  serd_env_free(env);
+  serd_node_free(blank);
+}
+
+static void
+test_qualify(void)
+{
+  const SerdStringView eg = serd_string(NS_EG);
+
+  SerdNode* const name = serd_new_string(serd_string("eg"));
+  SerdNode* const c1   = serd_new_curie(serd_string("eg:foo"));
+  SerdNode* const u1   = serd_new_uri(serd_string("http://example.org/foo"));
+  SerdNode* const u2   = serd_new_uri(serd_string("http://drobilla.net/bar"));
+
+  SerdEnv* const env = serd_env_new(serd_empty_string());
+
+  assert(!serd_env_set_prefix(env, serd_node_string_view(name), eg));
+
+  assert(!serd_env_expand(env, name));
+
+  SerdNode* const u1_out = serd_env_qualify(env, u1);
+  assert(serd_node_equals(u1_out, c1));
+  serd_node_free(u1_out);
+
+  assert(!serd_env_qualify(env, u2));
+
+  serd_env_free(env);
+  serd_node_free(u2);
+  serd_node_free(u1);
+  serd_node_free(c1);
+  serd_node_free(name);
+}
+
+static void
+test_equals(void)
+{
+  const SerdStringView name1 = serd_string("n1");
+  const SerdStringView base1 = serd_string(NS_EG "b1/");
+  const SerdStringView base2 = serd_string(NS_EG "b2/");
+
+  SerdEnv* const env1 = serd_env_new(base1);
+  SerdEnv* const env2 = serd_env_new(base2);
+
+  assert(!serd_env_equals(env1, NULL));
+  assert(!serd_env_equals(NULL, env1));
+  assert(serd_env_equals(NULL, NULL));
+  assert(!serd_env_equals(env1, env2));
+
+  serd_env_set_base_uri(env2, base1);
+  assert(serd_env_equals(env1, env2));
+
+  assert(!serd_env_set_prefix(env1, name1, serd_string(NS_EG "n1")));
+  assert(!serd_env_equals(env1, env2));
+  assert(!serd_env_set_prefix(env2, name1, serd_string(NS_EG "othern1")));
+  assert(!serd_env_equals(env1, env2));
+  assert(!serd_env_set_prefix(env2, name1, serd_string(NS_EG "n1")));
+  assert(serd_env_equals(env1, env2));
+
+  serd_env_set_base_uri(env2, base2);
+  assert(!serd_env_equals(env1, env2));
+
+  SerdEnv* const env3 = serd_env_copy(env2);
+  assert(serd_env_equals(env3, env2));
+  serd_env_free(env3);
+
+  serd_env_free(env2);
+  serd_env_free(env1);
+}
+
 int
 main(void)
 {
   test_copy();
   test_comparison();
-  test_env();
+  test_null();
+  test_base_uri();
+  test_set_prefix();
+  test_expand_untyped_literal();
+  test_expand_bad_uri_datatype();
+  test_expand_uri();
+  test_expand_empty_uri_ref();
+  test_expand_bad_uri();
+  test_expand_curie();
+  test_expand_bad_curie();
+  test_expand_blank();
+  test_qualify();
+  test_equals();
   return 0;
 }