migrated context functionality in tests