Function TestUserFunctions

db/functions/function_test.go:175–192

Unit test for JS user functions.

(t *testing.T)

// Unit test for JS user functions.
func TestUserFunctions(t *testing.T) {
	// base.SetUpTestLogging(t, base.LevelDebug, base.KeyAll)
	db, ctx := setupTestDBWithFunctions(t, &kUserFunctionConfig)
	defer db.Close(ctx)

	assert.NotNil(t, db.Options.UserFunctions)
	assert.NotNil(t, db.Options.UserFunctions.Definitions["square"])

	// First run the tests as an admin:
	t.Run("AsAdmin", func(t *testing.T) { testUserFunctionsAsAdmin(t, ctx, db) })

	// Now create a user and make it current:
	db.SetUser(addUserAlice(t, db))
	assert.True(t, db.User().RoleNames().Contains("hero"))

	// Repeat the tests as user "alice":
	t.Run("AsUser", func(t *testing.T) { testUserFunctionsAsUser(t, ctx, db) })
}

// User function tests that work the same for admin and non-admin user:
func testUserFunctionsCommon(t *testing.T, ctx context.Context, db *db.Database) {