@@ -368,6 +368,7 @@ static struct vimvar
368368 {VV_NAME("option_new", VAR_STRING), VV_RO},
369369 {VV_NAME("option_old", VAR_STRING), VV_RO},
370370 {VV_NAME("option_type", VAR_STRING), VV_RO},
371+ {VV_NAME("errors", VAR_LIST), 0},
371372};
372373
373374/* shorthand */
@@ -472,6 +473,9 @@ static void f_argc __ARGS((typval_T *argvars, typval_T *rettv));
472473static void f_argidx __ARGS((typval_T *argvars, typval_T *rettv));
473474static void f_arglistid __ARGS((typval_T *argvars, typval_T *rettv));
474475static void f_argv __ARGS((typval_T *argvars, typval_T *rettv));
476+ static void f_assertEqual __ARGS((typval_T *argvars, typval_T *rettv));
477+ static void f_assertFalse __ARGS((typval_T *argvars, typval_T *rettv));
478+ static void f_assertTrue __ARGS((typval_T *argvars, typval_T *rettv));
475479#ifdef FEAT_FLOAT
476480static void f_asin __ARGS((typval_T *argvars, typval_T *rettv));
477481static void f_atan __ARGS((typval_T *argvars, typval_T *rettv));
@@ -8068,6 +8072,9 @@ static struct fst
80688072 {"argidx", 0, 0, f_argidx},
80698073 {"arglistid", 0, 2, f_arglistid},
80708074 {"argv", 0, 1, f_argv},
8075+ {"assertEqual", 2, 3, f_assertEqual},
8076+ {"assertFalse", 1, 2, f_assertFalse},
8077+ {"assertTrue", 1, 2, f_assertTrue},
80718078#ifdef FEAT_FLOAT
80728079 {"asin", 1, 1, f_asin}, /* WJMc */
80738080 {"atan", 1, 1, f_atan},
@@ -9124,6 +9131,113 @@ f_argv(argvars, rettv)
91249131 alist_name(&ARGLIST[idx]), -1);
91259132}
91269133
9134+ static void assertError __ARGS((garray_T *gap));
9135+ static void prepareForAssertError __ARGS((garray_T*gap));
9136+ static void assertBool __ARGS((typval_T *argvars, int isTrue));
9137+
9138+ /*
9139+ * Add an assert error to v:errors.
9140+ */
9141+ static void
9142+ assertError(gap)
9143+ garray_T *gap;
9144+ {
9145+ struct vimvar *vp = &vimvars[VV_ERRORS];
9146+
9147+ if (vp->vv_type != VAR_LIST || vimvars[VV_ERRORS].vv_list == NULL)
9148+ /* Make sure v:errors is a list. */
9149+ set_vim_var_list(VV_ERRORS, list_alloc());
9150+ list_append_string(vimvars[VV_ERRORS].vv_list, gap->ga_data, gap->ga_len);
9151+ }
9152+
9153+ static void
9154+ prepareForAssertError(gap)
9155+ garray_T *gap;
9156+ {
9157+ char buf[NUMBUFLEN];
9158+
9159+ ga_init2(gap, 1, 100);
9160+ ga_concat(gap, sourcing_name);
9161+ sprintf(buf, " line %ld", (long)sourcing_lnum);
9162+ ga_concat(gap, (char_u *)buf);
9163+ }
9164+
9165+ /*
9166+ * "assertEqual(expected, actual[, msg])" function
9167+ */
9168+ static void
9169+ f_assertEqual(argvars, rettv)
9170+ typval_T *argvars;
9171+ typval_T *rettv UNUSED;
9172+ {
9173+ garray_T ga;
9174+ char_u *tofree;
9175+ char_u numbuf[NUMBUFLEN];
9176+
9177+ if (!tv_equal(&argvars[0], &argvars[1], FALSE, FALSE))
9178+ {
9179+ prepareForAssertError(&ga);
9180+ ga_concat(&ga, (char_u *)": Expected ");
9181+ ga_concat(&ga, tv2string(&argvars[0], &tofree, numbuf, 0));
9182+ vim_free(tofree);
9183+ ga_concat(&ga, (char_u *)" but got ");
9184+ ga_concat(&ga, tv2string(&argvars[1], &tofree, numbuf, 0));
9185+ vim_free(tofree);
9186+ assertError(&ga);
9187+ ga_clear(&ga);
9188+ }
9189+ }
9190+
9191+ static void
9192+ assertBool(argvars, isTrue)
9193+ typval_T *argvars;
9194+ int isTrue;
9195+ {
9196+ int error = FALSE;
9197+ garray_T ga;
9198+ char_u *tofree;
9199+ char_u numbuf[NUMBUFLEN];
9200+
9201+ if (argvars[0].v_type != VAR_NUMBER
9202+ || (get_tv_number_chk(&argvars[0], &error) == 0) == isTrue
9203+ || error)
9204+ {
9205+ prepareForAssertError(&ga);
9206+ ga_concat(&ga, (char_u *)": Expected ");
9207+ if (isTrue)
9208+ ga_concat(&ga, (char_u *)"True ");
9209+ else
9210+ ga_concat(&ga, (char_u *)"False ");
9211+ ga_concat(&ga, (char_u *)"but got ");
9212+ ga_concat(&ga, tv2string(&argvars[0], &tofree, numbuf, 0));
9213+ vim_free(tofree);
9214+ assertError(&ga);
9215+ ga_clear(&ga);
9216+ }
9217+ }
9218+
9219+ /*
9220+ * "assertFalse(actual[, msg])" function
9221+ */
9222+ static void
9223+ f_assertFalse(argvars, rettv)
9224+ typval_T *argvars;
9225+ typval_T *rettv UNUSED;
9226+ {
9227+ assertBool(argvars, FALSE);
9228+ }
9229+
9230+ /*
9231+ * "assertTrue(actual[, msg])" function
9232+ */
9233+ static void
9234+ f_assertTrue(argvars, rettv)
9235+ typval_T *argvars;
9236+ typval_T *rettv UNUSED;
9237+ {
9238+ assertBool(argvars, TRUE);
9239+ }
9240+
91279241#ifdef FEAT_FLOAT
91289242/*
91299243 * "asin()" function
0 commit comments