Skip to content

Commit d6c674c

Browse files
Fix the get mockable client methods to be private (#50978)
* fix the issue * mockable client should be public
1 parent 37d949f commit d6c674c

File tree

3 files changed

+7
-4
lines changed

3 files changed

+7
-4
lines changed

eng/packages/http-client-csharp-mgmt/generator/Azure.Generator.Management/src/ManagementOutputLibrary.cs

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,18 +81,21 @@ private IReadOnlyList<TypeProvider> BuildExtensions(IReadOnlyList<ResourceClient
8181
scopeCandidates[resource.ResourceScope].Add(resource);
8282
}
8383

84+
var mockableArmClientResource = new MockableArmClientProvider(typeof(ArmClient), resources);
8485
var mockableResources = new List<MockableResourceProvider>(scopeCandidates.Count)
8586
{
8687
// add the arm client mockable resource
87-
new MockableArmClientProvider(typeof(ArmClient), resources)
88+
mockableArmClientResource
8889
};
90+
ManagementClientGenerator.Instance.AddTypeToKeep(mockableArmClientResource.Name);
8991

9092
foreach (var (scope, candidates) in scopeCandidates)
9193
{
9294
if (candidates.Count > 0)
9395
{
9496
var mockableExtension = new MockableResourceProvider(_scopeToTypes[scope], candidates);
9597
mockableResources.Add(mockableExtension);
98+
ManagementClientGenerator.Instance.AddTypeToKeep(mockableExtension.Name);
9699
}
97100
}
98101

eng/packages/http-client-csharp-mgmt/generator/Azure.Generator.Management/src/Providers/ExtensionProvider.cs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ private MethodProvider BuildGetCachedClientMethod(MockableResourceProvider mocka
6464
var methodSignature = new MethodSignature(
6565
$"Get{mockableResource.Name}",
6666
null,
67-
MethodSignatureModifiers.Public | MethodSignatureModifiers.Static,
67+
MethodSignatureModifiers.Private | MethodSignatureModifiers.Static,
6868
mockableResource.Type,
6969
null,
7070
[parameter]);

eng/packages/http-client-csharp-mgmt/generator/TestProjects/Local/Mgmt-TypeSpec/src/Generated/Extensions/MgmtTypeSpecExtensions.cs

Lines changed: 2 additions & 2 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)