]> git.saurik.com Git - apple/icu.git/blobdiff - icuSources/test/cintltst/cintltst.c
ICU-531.30.tar.gz
[apple/icu.git] / icuSources / test / cintltst / cintltst.c
index f99885a8a7caf7bf71a84581cd9f7f073e1872c8..1448e0633eff96c81d5fba21b6c560f480bf6743 100644 (file)
@@ -1,6 +1,6 @@
 /********************************************************************
  * COPYRIGHT:
- * Copyright (c) 1997-2003, International Business Machines Corporation and
+ * Copyright (c) 1997-2013, International Business Machines Corporation and
  * others. All Rights Reserved.
  ********************************************************************/
 /********************************************************************************
 #include <string.h>
 #include "unicode/utypes.h"
 #include "unicode/putil.h"
-
+#include "cstring.h"
 #include "cintltst.h"
-#include "umutex.h"
-
+#include "uassert.h"
+#include "cmemory.h"
 #include "unicode/uchar.h"
 #include "unicode/ustring.h"
 #include "unicode/ucnv.h"
 #include "unicode/ures.h"
 #include "unicode/uclean.h"
+#include "unicode/ucal.h"
+#include "uoptions.h"
+#include "putilimp.h" /* for uprv_getRawUTCtime() */
+#ifdef URES_DEBUG
+#include "uresimp.h" /* for ures_dumpCacheContents() */
+#endif
 
 #ifdef XP_MAC_CONSOLE
 #   include <console.h>
 #endif
 
+#define CTST_MAX_ALLOC 8192
+/* Array used as a queue */
+static void * ctst_allocated_stuff[CTST_MAX_ALLOC] = {0};
+static int ctst_allocated = 0;
+static UBool ctst_free = FALSE;
+static int ctst_allocated_total = 0;
+
 #define CTST_LEAK_CHECK 1
+
 #ifdef CTST_LEAK_CHECK
-U_CFUNC void ctst_freeAll(void);
-U_CFUNC void ctst_init(void);
+static void ctst_freeAll(void);
 #endif
 
 static char* _testDataPath=NULL;
@@ -47,21 +60,7 @@ static char* _testDataPath=NULL;
  */
 void ctest_setICU_DATA(void);
 
-static UBool gMutexInitialized = FALSE;
 
-static void TestMutex(void) {
-    if (!gMutexInitialized) {
-        log_verbose("*** Failure! The global mutex was not initialized.\n"
-                "*** Make sure the right linker was used.\n");
-    }
-}
-
-U_CFUNC void addSetup(TestNode** root);
-
-void addSetup(TestNode** root)
-{
-    addTest(root, &TestMutex,    "setup/TestMutex");
-}
 
 #if UCONFIG_NO_LEGACY_CONVERSION
 #   define TRY_CNV_1 "iso-8859-1"
@@ -71,70 +70,80 @@ void addSetup(TestNode** root)
 #   define TRY_CNV_2 "sjis"
 #endif
 
+static int gOrigArgc;
+static const char* const * gOrigArgv;
+
 int main(int argc, const char* const argv[])
 {
     int nerrors = 0;
-    int warnOnMissingData = 0;
-    int i;
+    UBool   defaultDataFound;
     TestNode *root;
     const char *warnOrErr = "Failure"; 
+    UDate startTime, endTime;
+    int32_t diffTime;
 
     /* initial check for the default converter */
     UErrorCode errorCode = U_ZERO_ERROR;
     UResourceBundle *rb;
     UConverter *cnv;
 
-    /* This must be tested before using anything! */
-    gMutexInitialized = umtx_isInitialized(NULL);
+    U_MAIN_INIT_ARGS(argc, argv);
 
-    /* Checkargs */
-    for(i=1;i<argc;i++) {
-        if(!strcmp(argv[i],"-w")) {
-            warnOnMissingData = 1;
-            warnOrErr = "Warning";
-        }
-    }
-
-    while (REPEAT_TESTS > 0) {
+    startTime = uprv_getRawUTCtime();
 
-#ifdef CTST_LEAK_CHECK
-        ctst_init();
+    gOrigArgc = argc;
+    gOrigArgv = argv;
+    if (!initArgs(argc, argv, NULL, NULL)) {
+        /* Error already displayed. */
+        return -1;
+    }
+    
+    /* Check whether ICU will initialize without forcing the build data directory into
+     *  the ICU_DATA path.  Success here means either the data dll contains data, or that
+     *  this test program was run with ICU_DATA set externally.  Failure of this check
+     *  is normal when ICU data is not packaged into a shared library.
+     *
+     *  Whether or not this test succeeds, we want to cleanup and reinitialize
+     *  with a data path so that data loading from individual files can be tested.
+     */
+    defaultDataFound = TRUE;
+    u_init(&errorCode);
+    if (U_FAILURE(errorCode)) {
+        fprintf(stderr,
+            "#### Note:  ICU Init without build-specific setDataDirectory() failed. %s\n", u_errorName(errorCode));
+        defaultDataFound = FALSE;
+    }
+    u_cleanup();
+#ifdef URES_DEBUG
+    fprintf(stderr, "After initial u_cleanup: RB cache %s empty.\n", ures_dumpCacheContents()?"WAS NOT":"was");
 #endif
-        /* try opening the data from dll instead of the dat file */
-        cnv = ucnv_open(TRY_CNV_1, &errorCode);
-        if(cnv != 0) {
-            /* ok */
-            ucnv_close(cnv);
-        } else {
-            fprintf(stderr,
-                    "#### WARNING! The converter for " TRY_CNV_1 " cannot be loaded from data dll/so."
-                    "Proceeding to load data from dat file.\n");
-            errorCode = U_ZERO_ERROR;
 
-            ctest_setICU_DATA();
+    while (getTestOption(REPEAT_TESTS_OPTION) > 0) {   /* Loop runs once per complete execution of the tests
+                                  *   used for -r  (repeat) test option.                */
+        if (!initArgs(argc, argv, NULL, NULL)) {
+            /* Error already displayed. */
+            return -1;
         }
+        errorCode = U_ZERO_ERROR;
 
-        /* If no ICU_DATA environment was set, try to fake up one. */
-        /* fprintf(stderr, "u_getDataDirectory() = %s\n", u_getDataDirectory()); */
-
-#ifdef XP_MAC_CONSOLE
-        argc = ccommand((char***)&argv);
-#endif
-
-        cnv  = ucnv_open(NULL, &errorCode);
-        if(cnv != NULL) {
-            /* ok */
-            ucnv_close(cnv);
-        } else {
+        /* Initialize ICU */
+        if (!defaultDataFound) {
+            ctest_setICU_DATA();    /* u_setDataDirectory() must happen Before u_init() */
+        }
+        u_init(&errorCode);
+        if (U_FAILURE(errorCode)) {
             fprintf(stderr,
-                "*** %s! The default converter cannot be opened.\n"
+                "#### ERROR! %s: u_init() failed with status = \"%s\".\n" 
                 "*** Check the ICU_DATA environment variable and \n"
-                "*** check that the data files are present.\n", warnOrErr);
-            if(warnOnMissingData == 0) {
-                fprintf(stderr, "*** Exitting.  Use the '-w' option if data files were\n*** purposely removed, to continue test anyway.\n");
-                return 1;
-            }
+                "*** check that the data files are present.\n", argv[0], u_errorName(errorCode));
+                if(!getTestOption(WARN_ON_MISSING_DATA_OPTION)) {
+                    fprintf(stderr, "*** Exiting.  Use the '-w' option if data files were\n*** purposely removed, to continue test anyway.\n");
+                    u_cleanup();
+                    return 1;
+                }
         }
+        
+
 
         /* try more data */
         cnv = ucnv_open(TRY_CNV_2, &errorCode);
@@ -146,8 +155,9 @@ int main(int argc, const char* const argv[])
                     "*** %s! The converter for " TRY_CNV_2 " cannot be opened.\n"
                     "*** Check the ICU_DATA environment variable and \n"
                     "*** check that the data files are present.\n", warnOrErr);
-            if(warnOnMissingData == 0) {
+            if(!getTestOption(WARN_ON_MISSING_DATA_OPTION)) {
                 fprintf(stderr, "*** Exitting.  Use the '-w' option if data files were\n*** purposely removed, to continue test anyway.\n");
+                u_cleanup();
                 return 1;
             }
         }
@@ -161,40 +171,75 @@ int main(int argc, const char* const argv[])
                     "*** %s! The \"en\" locale resource bundle cannot be opened.\n"
                     "*** Check the ICU_DATA environment variable and \n"
                     "*** check that the data files are present.\n", warnOrErr);
-            if(warnOnMissingData == 0) {
+            if(!getTestOption(WARN_ON_MISSING_DATA_OPTION)) {
                 fprintf(stderr, "*** Exitting.  Use the '-w' option if data files were\n*** purposely removed, to continue test anyway.\n");
+                u_cleanup();
                 return 1;
             }
         }
 
+        errorCode = U_ZERO_ERROR;
+        rb = ures_open(NULL, NULL, &errorCode);
+        if(U_SUCCESS(errorCode)) {
+            /* ok */
+            if (errorCode == U_USING_DEFAULT_WARNING || errorCode == U_USING_FALLBACK_WARNING) {
+                fprintf(stderr,
+                        "#### Note: The default locale %s is not available\n", uloc_getDefault());
+            }
+            ures_close(rb);
+        } else {
+            fprintf(stderr,
+                    "*** %s! Can not open a resource bundle for the default locale %s\n", warnOrErr, uloc_getDefault());
+            if(!getTestOption(WARN_ON_MISSING_DATA_OPTION)) {
+                fprintf(stderr, "*** Exitting.  Use the '-w' option if data files were\n"
+                    "*** purposely removed, to continue test anyway.\n");
+                u_cleanup();
+                return 1;
+            }
+        }
         fprintf(stdout, "Default locale for this run is %s\n", uloc_getDefault());
 
+        /* Build a tree of all tests.   
+         *   Subsequently will be used to find / iterate the tests to run */
         root = NULL;
         addAllTests(&root);
-        nerrors = processArgs(root, argc, argv);
-        if (--REPEAT_TESTS > 0) {
-            printf("Repeating tests %d more time(s)\n", REPEAT_TESTS);
+
+        /*  Tests acutally run HERE.   TODO:  separate command line option parsing & setting from test execution!! */
+        nerrors = runTestRequest(root, argc, argv);
+
+        setTestOption(REPEAT_TESTS_OPTION, DECREMENT_OPTION_VALUE);
+        if (getTestOption(REPEAT_TESTS_OPTION) > 0) {
+            printf("Repeating tests %d more time(s)\n", getTestOption(REPEAT_TESTS_OPTION));
         }
         cleanUpTestTree(root);
+
 #ifdef CTST_LEAK_CHECK
         ctst_freeAll();
-
         /* To check for leaks */
-
         u_cleanup(); /* nuke the hashtable.. so that any still-open cnvs are leaked */
+        
+        if(getTestOption(VERBOSITY_OPTION) && ctst_allocated_total>0) {
+          fprintf(stderr,"ctst_freeAll():  cleaned up after %d allocations (queue of %d)\n", ctst_allocated_total, CTST_MAX_ALLOC);
+        }
+#ifdef URES_DEBUG
+        if(ures_dumpCacheContents()) {
+          fprintf(stderr, "Error: After final u_cleanup, RB cache was not empty.\n");
+          nerrors++;
+        } else {
+          fprintf(stderr,"OK: After final u_cleanup, RB cache was empty.\n");
+        }
+#endif
 #endif
-    }
 
-    if (!gMutexInitialized) {
-        fprintf(stderr,
-            "#### WARNING!\n"
-            "  The global mutex was not initialized during C++ static initialization.\n"
-            "  You must explicitly initialize ICU by calling u_init() before using ICU in multiple threads.\n"
-            "  If you are using ICU in a single threaded application, use of u_init() is recommended,\n"
-            "  but is not required.\n"
-            "#### WARNING!\n"
-            );
-    }
+    }  /* End of loop that repeats the entire test, if requested.  (Normally doesn't loop)  */
+
+    endTime = uprv_getRawUTCtime();
+    diffTime = (int32_t)(endTime - startTime);
+    printf("Elapsed Time: %02d:%02d:%02d.%03d\n",
+        (int)((diffTime%U_MILLIS_PER_DAY)/U_MILLIS_PER_HOUR),
+        (int)((diffTime%U_MILLIS_PER_HOUR)/U_MILLIS_PER_MINUTE),
+        (int)((diffTime%U_MILLIS_PER_MINUTE)/U_MILLIS_PER_SECOND),
+        (int)(diffTime%U_MILLIS_PER_SECOND));
 
     return nerrors ? 1 : 0;
 }
@@ -238,60 +283,6 @@ static void ctest_appendToDataDirectory(const char *toAppend)
 }
 */
 
-void
-ctest_pathnameInContext( char* fullname, int32_t maxsize, const char* relPath )
-{
-    char mainDirBuffer[1024];
-    char* mainDir = NULL;
-    const char *dataDirectory = ctest_dataOutDir();
-    const char inpSepChar = '|';
-    char* tmp;
-    int32_t lenMainDir;
-    int32_t lenRelPath;
-
-#ifdef XP_MAC
-    Str255 volName;
-    int16_t volNum;
-    OSErr err = GetVol( volName, &volNum );
-    if (err != noErr)
-        volName[0] = 0;
-    mainDir = (char*) &(volName[1]);
-    mainDir[volName[0]] = 0;
-#else
-    if (dataDirectory != NULL) {
-        strcpy(mainDirBuffer, dataDirectory);
-        strcat(mainDirBuffer, ".." U_FILE_SEP_STRING);
-    } else {
-        mainDirBuffer[0]='\0';
-    }
-    mainDir = mainDirBuffer;
-#endif
-
-    lenMainDir = (int32_t)strlen(mainDir);
-    if(lenMainDir > 0 && mainDir[lenMainDir - 1] != U_FILE_SEP_CHAR) {
-        mainDir[lenMainDir++] = U_FILE_SEP_CHAR;
-        mainDir[lenMainDir] = 0;
-    }
-
-    if (relPath[0] == '|')
-        relPath++;
-    lenRelPath = (int32_t)strlen(relPath);
-    if (maxsize < lenMainDir + lenRelPath + 2) {
-        fullname[0] = 0;
-        return;
-    }
-    strcpy(fullname, mainDir);
-    /*strcat(fullname, U_FILE_SEP_STRING);*/
-    strcat(fullname, relPath);
-    strchr(fullname, inpSepChar);
-    tmp = strchr(fullname, inpSepChar);
-    while (tmp) {
-        *tmp = U_FILE_SEP_CHAR;
-        tmp = strchr(tmp+1, inpSepChar);
-    }
-}
-
-
 /* returns the path to icu/source/data */
 const char *  ctest_dataSrcDir()
 {
@@ -324,7 +315,6 @@ const char *  ctest_dataSrcDir()
         char *pBackSlash;
         int i;
 
-        dataSrcDir = p;
         strcpy(p, __FILE__);
         /* We want to back over three '\' chars.                            */
         /*   Only Windows should end up here, so looking for '\' is safe.   */
@@ -340,10 +330,18 @@ const char *  ctest_dataSrcDir()
              *  Now append "source\data" and set the environment
              */
             strcpy(pBackSlash, U_FILE_SEP_STRING "data" U_FILE_SEP_STRING );
+            dataSrcDir = p;
         }
         else {
             /* __FILE__ on MSVC7 does not contain the directory */
-            strcpy(p, ".."U_FILE_SEP_STRING".."U_FILE_SEP_STRING "data" U_FILE_SEP_STRING);
+            FILE *file = fopen(".."U_FILE_SEP_STRING".."U_FILE_SEP_STRING "data" U_FILE_SEP_STRING "Makefile.in", "r");
+            if (file) {
+                fclose(file);
+                dataSrcDir = ".."U_FILE_SEP_STRING".."U_FILE_SEP_STRING "data" U_FILE_SEP_STRING;
+            }
+            else {
+                dataSrcDir = ".."U_FILE_SEP_STRING".."U_FILE_SEP_STRING".."U_FILE_SEP_STRING".."U_FILE_SEP_STRING "data" U_FILE_SEP_STRING;
+            }
         }
     }
 #endif
@@ -371,7 +369,7 @@ const char *ctest_dataOutDir()
     */
 #if defined (U_TOPBUILDDIR)
     {
-        dataOutDir = U_TOPBUILDDIR  U_FILE_SEP_STRING "data"U_FILE_SEP_STRING"out"U_FILE_SEP_STRING;
+        dataOutDir = U_TOPBUILDDIR "data"U_FILE_SEP_STRING"out"U_FILE_SEP_STRING;
     }
 #else
 
@@ -384,7 +382,6 @@ const char *ctest_dataOutDir()
         char *pBackSlash;
         int i;
 
-        dataOutDir = p;
         strcpy(p, __FILE__);
         /* We want to back over three '\' chars.                            */
         /*   Only Windows should end up here, so looking for '\' is safe.   */
@@ -400,10 +397,18 @@ const char *ctest_dataOutDir()
              *  Now append "source\data" and set the environment
              */
             strcpy(pBackSlash, U_FILE_SEP_STRING "data" U_FILE_SEP_STRING "out" U_FILE_SEP_STRING);
+            dataOutDir = p;
         }
         else {
             /* __FILE__ on MSVC7 does not contain the directory */
-            strcpy(p, ".."U_FILE_SEP_STRING".."U_FILE_SEP_STRING "data" U_FILE_SEP_STRING "out" U_FILE_SEP_STRING);
+            FILE *file = fopen(".."U_FILE_SEP_STRING".."U_FILE_SEP_STRING "data" U_FILE_SEP_STRING "Makefile.in", "r");
+            if (file) {
+                fclose(file);
+                dataOutDir = ".."U_FILE_SEP_STRING".."U_FILE_SEP_STRING "data" U_FILE_SEP_STRING "out" U_FILE_SEP_STRING;
+            }
+            else {
+                dataOutDir = ".."U_FILE_SEP_STRING".."U_FILE_SEP_STRING".."U_FILE_SEP_STRING".."U_FILE_SEP_STRING "data" U_FILE_SEP_STRING "out" U_FILE_SEP_STRING;
+            }
         }
     }
 #endif
@@ -422,11 +427,46 @@ const char *ctest_dataOutDir()
  */
 void ctest_setICU_DATA() {
 
-
     /* No location for the data dir was identifiable.
      *   Add other fallbacks for the test data location here if the need arises
      */
-    u_setDataDirectory(ctest_dataOutDir());
+    if (getenv("ICU_DATA") == NULL) {
+        /* If ICU_DATA isn't set, set it to the usual location */
+        u_setDataDirectory(ctest_dataOutDir());
+    }
+}
+
+/*  These tests do cleanup and reinitialize ICU in the course of their operation.
+ *    The ICU data directory must be preserved across these operations.
+ *    Here is a helper function to assist with that.
+ */
+static char *safeGetICUDataDirectory() {
+    const char *dataDir = u_getDataDirectory();  /* Returned string vanashes with u_cleanup */
+    char *retStr = NULL;
+    if (dataDir != NULL) {
+        retStr = (char *)malloc(strlen(dataDir)+1);
+        strcpy(retStr, dataDir);
+    }
+    return retStr;
+}
+
+UBool ctest_resetICU() {
+    UErrorCode   status = U_ZERO_ERROR;
+    char         *dataDir = safeGetICUDataDirectory();
+
+    u_cleanup();
+    if (!initArgs(gOrigArgc, gOrigArgv, NULL, NULL)) {
+        /* Error already displayed. */
+        return FALSE;
+    }
+    u_setDataDirectory(dataDir);
+    free(dataDir);
+    u_init(&status);
+    if (U_FAILURE(status)) {
+        log_err_status(status, "u_init failed with %s\n", u_errorName(status));
+        return FALSE;
+    }
+    return TRUE;
 }
 
 UChar* CharsToUChars(const char* str) {
@@ -461,7 +501,7 @@ char *aescstrdup(const UChar* unichars,int32_t length){
     const void *p;
     UErrorCode errorCode = U_ZERO_ERROR;
 #if U_CHARSET_FAMILY==U_EBCDIC_FAMILY
-#   ifdef OS390
+#   if U_PLATFORM == U_PF_OS390
         static const char convName[] = "ibm-1047";
 #   else
         static const char convName[] = "ibm-37";
@@ -476,7 +516,7 @@ char *aescstrdup(const UChar* unichars,int32_t length){
     newString = (char*)ctst_malloc ( sizeof(char) * 8 * (length +1));
     target = newString;
     targetLimit = newString+sizeof(char) * 8 * (length +1);
-    ucnv_setFromUCallBack(conv, UCNV_FROM_U_CALLBACK_ESCAPE, UCNV_ESCAPE_JAVA, &cb, &p, &errorCode);
+    ucnv_setFromUCallBack(conv, UCNV_FROM_U_CALLBACK_ESCAPE, UCNV_ESCAPE_C, &cb, &p, &errorCode);
     ucnv_fromUnicode(conv,&target,targetLimit, &unichars, (UChar*)(unichars+length),NULL,TRUE,&errorCode);
     ucnv_close(conv);
     *target = '\0';
@@ -484,12 +524,18 @@ char *aescstrdup(const UChar* unichars,int32_t length){
 }
 
 const char* loadTestData(UErrorCode* err){
-    const char*      directory=NULL;
-    UResourceBundle* test =NULL;
-    char* tdpath=NULL;
-    const char* tdrelativepath = ".."U_FILE_SEP_STRING".."U_FILE_SEP_STRING"test"U_FILE_SEP_STRING"testdata"U_FILE_SEP_STRING"out"U_FILE_SEP_STRING;
     if( _testDataPath == NULL){
+        const char*      directory=NULL;
+        UResourceBundle* test =NULL;
+        char* tdpath=NULL;
+        const char* tdrelativepath;
+#if defined (U_TOPBUILDDIR)
+        tdrelativepath = "test"U_FILE_SEP_STRING"testdata"U_FILE_SEP_STRING"out"U_FILE_SEP_STRING;
+        directory = U_TOPBUILDDIR;
+#else
+        tdrelativepath = ".."U_FILE_SEP_STRING".."U_FILE_SEP_STRING"test"U_FILE_SEP_STRING"testdata"U_FILE_SEP_STRING"out"U_FILE_SEP_STRING;
         directory= ctest_dataOutDir();
+#endif
 
         tdpath = (char*) ctst_malloc(sizeof(char) *(( strlen(directory) * strlen(tdrelativepath)) + 10));
 
@@ -510,7 +556,7 @@ const char* loadTestData(UErrorCode* err){
         /* Fall back did not succeed either so return */
         if(U_FAILURE(*err)){
             *err = U_FILE_ACCESS_ERROR;
-            log_err("Could not load testtypes.res in testdata bundle with path %s - %s\n", tdpath, u_errorName(*err));
+            log_data_err("Could not load testtypes.res in testdata bundle with path %s - %s\n", tdpath, u_errorName(*err));
             return "";
         }
         ures_close(test);
@@ -520,20 +566,62 @@ const char* loadTestData(UErrorCode* err){
     return _testDataPath;
 }
 
-#define CTST_MAX_ALLOC 10000
-/* Array used as a queue */
-static void * ctst_allocated_stuff[CTST_MAX_ALLOC];
-static int ctst_allocated = 0;
-static UBool ctst_free = FALSE;
+#define CTEST_MAX_TIMEZONE_SIZE 256
+static UChar gOriginalTimeZone[CTEST_MAX_TIMEZONE_SIZE] = {0};
 
-void ctst_init(void) {
-    int i;
-    for(i=0; i<CTST_MAX_ALLOC; i++) {
-        ctst_allocated_stuff[i] = NULL;
+/**
+ * Call this once to get a consistent timezone. Use ctest_resetTimeZone to set it back to the original value.
+ * @param optionalTimeZone Set this to a requested timezone.
+ *      Set to NULL to use the standard test timezone (Pacific Time)
+ */
+U_CFUNC void ctest_setTimeZone(const char *optionalTimeZone, UErrorCode *status) {
+#if !UCONFIG_NO_FORMATTING
+    UChar zoneID[CTEST_MAX_TIMEZONE_SIZE];
+
+    if (optionalTimeZone == NULL) {
+        optionalTimeZone = "America/Los_Angeles";
+    }
+    if (gOriginalTimeZone[0]) {
+        log_data_err("*** Error: time zone saved twice. New value will be %s (Are you missing data?)\n",
+               optionalTimeZone);
     }
+    ucal_getDefaultTimeZone(gOriginalTimeZone, CTEST_MAX_TIMEZONE_SIZE, status);
+    if (U_FAILURE(*status)) {
+        log_err("*** Error: Failed to save default time zone: %s\n",
+               u_errorName(*status));
+        *status = U_ZERO_ERROR;
+    }
+
+    u_uastrncpy(zoneID, optionalTimeZone, CTEST_MAX_TIMEZONE_SIZE-1);
+    zoneID[CTEST_MAX_TIMEZONE_SIZE-1] = 0;
+    ucal_setDefaultTimeZone(zoneID, status);
+    if (U_FAILURE(*status)) {
+        log_err("*** Error: Failed to set default time zone to \"%s\": %s\n",
+               optionalTimeZone, u_errorName(*status));
+    }
+#endif
 }
 
+/**
+ * Call this once get back the original timezone
+ */
+U_CFUNC void ctest_resetTimeZone(void) {
+#if !UCONFIG_NO_FORMATTING
+    UErrorCode status = U_ZERO_ERROR;
+
+    ucal_setDefaultTimeZone(gOriginalTimeZone, &status);
+    if (U_FAILURE(status)) {
+        log_err("*** Error: Failed to reset default time zone: %s\n",
+               u_errorName(status));
+    }
+    /* Set to an empty state */
+    gOriginalTimeZone[0] = 0;
+#endif
+}
+
+
 void *ctst_malloc(size_t size) {
+  ctst_allocated_total++;
     if(ctst_allocated >= CTST_MAX_ALLOC - 1) {
         ctst_allocated = 0;
         ctst_free = TRUE;
@@ -545,17 +633,70 @@ void *ctst_malloc(size_t size) {
 }
 
 #ifdef CTST_LEAK_CHECK
-void ctst_freeAll() {
+static void ctst_freeAll() {
     int i;
-    if(ctst_free == 0) {
+    if(ctst_free == FALSE) { /* only free up to the allocated mark */
         for(i=0; i<ctst_allocated; i++) {
             free(ctst_allocated_stuff[i]);
+            ctst_allocated_stuff[i] = NULL;
         }
-    } else {
+    } else { /* free all */
         for(i=0; i<CTST_MAX_ALLOC; i++) {
             free(ctst_allocated_stuff[i]);
+            ctst_allocated_stuff[i] = NULL;
         }
     }
+    ctst_allocated = 0;
     _testDataPath=NULL;
 }
+
+#define VERBOSE_ASSERTIONS
+
+U_CFUNC UBool assertSuccessCheck(const char* msg, UErrorCode* ec, UBool possibleDataError) {
+    U_ASSERT(ec!=NULL);
+    if (U_FAILURE(*ec)) {
+        if (possibleDataError) {
+            log_data_err("FAIL: %s (%s)\n", msg, u_errorName(*ec));
+        } else {
+            log_err_status(*ec, "FAIL: %s (%s)\n", msg, u_errorName(*ec));
+        }
+        return FALSE;
+    }
+    return TRUE;
+}
+
+U_CFUNC UBool assertSuccess(const char* msg, UErrorCode* ec) {
+    U_ASSERT(ec!=NULL);
+    return assertSuccessCheck(msg, ec, FALSE);
+}
+
+/* if 'condition' is a UBool, the compiler complains bitterly about
+   expressions like 'a > 0' which it evaluates as int */
+U_CFUNC UBool assertTrue(const char* msg, int /*not UBool*/ condition) { 
+    if (!condition) {
+        log_err("FAIL: assertTrue() failed: %s\n", msg);
+    }
+#ifdef VERBOSE_ASSERTIONS
+    else {
+        log_verbose("Ok: %s\n", msg);
+    }
+#endif
+    return (UBool)condition;   
+}
+
+U_CFUNC UBool assertEquals(const char* message, const char* expected,
+                           const char* actual) {
+    if (uprv_strcmp(expected, actual) != 0) {
+        log_err("FAIL: %s; got \"%s\"; expected \"%s\"\n",
+                message, actual, expected);
+        return FALSE;
+    }
+#ifdef VERBOSE_ASSERTIONS
+    else {
+        log_verbose("Ok: %s; got \"%s\"\n", message, actual);
+    }
+#endif
+    return TRUE;
+}
+
 #endif