Apply suggestions from code review

Co-authored-by: Eastman <andrueastman@users.noreply.github.com>
microsoft · Jan 10, 2024 · e7e8304 · e7e8304
1 parent ef81f9c
commit e7e8304
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/Kiota.Builder/CodeDOM/CodeConstant.cs b/src/Kiota.Builder/CodeDOM/CodeConstant.cs
@@ -49,6 +49,7 @@ public string? UriTemplate
             Name = $"{codeClass.Name.ToFirstCharacterLowerCase()}UriTemplate",
             Kind = CodeConstantKind.UriTemplate,
             UriTemplate = urlTemplateProperty.DefaultValue,
+            OriginalCodeElement = codeClass
         };
     }
     public static CodeConstant? FromRequestBuilderToNavigationMetadata(CodeClass codeClass)

diff --git a/src/Kiota.Builder/CodeDOM/CodeInterface.cs b/src/Kiota.Builder/CodeDOM/CodeInterface.cs
@@ -49,7 +49,7 @@ CodeMethodKind.IndexerBackwardCompatibility or
 
         if (codeClass.Usings.ToArray() is { Length: > 0 } usings)
         {
-            foreach (var usingToCopy in usings.Where(static x => x.Declaration is not null && x.Declaration.TypeDefinition is CodeInterface or CodeClass { Kind: CodeClassKind.RequestBuilder }))
+            foreach (var usingToCopy in usings.Where(static x => x.Declaration?.TypeDefinition is CodeInterface or CodeClass { Kind: CodeClassKind.RequestBuilder }))
             {
                 usingToCopy.IsErasable = true;
             }