From adc7184d9824b0606950e97bbaf761b535e5f938 Mon Sep 17 00:00:00 2001
From: Ajay Bhargav Baaskaran <ajbaaska@microsoft.com>
Date: Fri, 23 Mar 2018 12:34:00 -0700
Subject: [PATCH] Register missed directives in RazorPageGenerator

---
 src/RazorPageGenerator/Program.cs | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/RazorPageGenerator/Program.cs b/src/RazorPageGenerator/Program.cs
index f3ae757886..49a3154c90 100644
--- a/src/RazorPageGenerator/Program.cs
+++ b/src/RazorPageGenerator/Program.cs
@@ -7,6 +7,7 @@ using System.IO;
 using System.Linq;
 using System.Text;
 using Microsoft.AspNetCore.Razor.Language;
+using Microsoft.AspNetCore.Razor.Language.Extensions;
 
 namespace RazorPageGenerator
 {
@@ -60,6 +61,10 @@ Examples:
                         @class.Modifiers.Add("internal");
                     });
 
+                FunctionsDirective.Register(builder);
+                InheritsDirective.Register(builder);
+                SectionDirective.Register(builder);
+
                 builder.Features.Add(new SuppressChecksumOptionsFeature());
                 builder.Features.Add(new SuppressMetadataAttributesFeature());