private void SetContextFromModeName(IPageBuilderContext context, string modeName)
        {
            context.ModeName().Returns(modeName);

            if (modeName == "Live")
            {
                context.IsLiveMode.Returns(true);
                context.IsLivePreviewMode.Returns(false);
                context.IsEditMode.Returns(false);
                context.Mode.Returns(PageBuilderMode.Live);

                return;
            }

            if (modeName == "LivePreview")
            {
                context.IsLiveMode.Returns(false);
                context.IsLivePreviewMode.Returns(true);
                context.IsEditMode.Returns(false);
                context.Mode.Returns(PageBuilderMode.LivePreview);

                return;
            }

            if (modeName == "Edit")
            {
                context.IsLiveMode.Returns(false);
                context.IsLivePreviewMode.Returns(false);
                context.IsEditMode.Returns(true);
                context.Mode.Returns(PageBuilderMode.Edit);

                return;
            }
        }
 public PageBuilderModeTagHelper(IPageBuilderContext pageBuilderContext) =>
 this.pageBuilderContext = pageBuilderContext ?? throw new ArgumentNullException(nameof(pageBuilderContext));