From 3a8987967ef412ee50e748edf8b25299de9bcba3 Mon Sep 17 00:00:00 2001
From: Ed Hartnett <edwardjameshartnett@gmail.com>
Date: Tue, 5 Jun 2018 11:53:59 -0600
Subject: [PATCH] improving testing of user-defined formats

---
 nc_test4/tst_udf.c | 30 ++++++++++++++++++++++++++++++
 1 file changed, 30 insertions(+)

diff --git a/nc_test4/tst_udf.c b/nc_test4/tst_udf.c
index b50ed5dae..4ff0f262c 100644
--- a/nc_test4/tst_udf.c
+++ b/nc_test4/tst_udf.c
@@ -154,11 +154,41 @@ NC_NOTNC4_set_var_chunk_cache,
 NC_NOTNC4_get_var_chunk_cache
 };
 
+#define NUM_UDFS 2
+
 int
 main(int argc, char **argv)
 {
    printf("\n*** Testing user-defined formats.\n");
    printf("*** testing simple user-defined format...");
+   {
+      int ncid;
+      int mode[NUM_UDFS] = {NC_UDF0, NC_UDF1};
+      int i;
+      
+      /* Create an empty file to play with. */
+      if (nc_create(FILE_NAME, 0, &ncid)) ERR;
+      if (nc_close(ncid)) ERR;
+
+      /* Test all available user-defined format slots. */
+      for (i = 0; i < NUM_UDFS; i++)
+      {
+         /* Add our user defined format. */
+         if (nc_def_user_format(mode[i], &tst_dispatcher, NULL)) ERR;
+         
+         /* Open file with our defined functions. */
+         if (nc_open(FILE_NAME, mode[i], &ncid)) ERR;
+         if (nc_close(ncid)) ERR;
+         
+         /* Open file again and abort, which is the same as closing it. */
+         if (nc_open(FILE_NAME, mode[i], &ncid)) ERR;
+         if (nc_inq_format(ncid, NULL) != TEST_VAL_42) ERR;
+         if (nc_inq_format_extended(ncid, NULL, NULL) != TEST_VAL_42) ERR;
+         if (nc_abort(ncid) != TEST_VAL_42) ERR;
+      }
+   }
+   SUMMARIZE_ERR;
+   printf("*** testing user-defined format with magic number...");
    {
       int ncid;
       /* Create an empty file to play with. */