feat:custom login service add environment (#659)

masastack · Jul 25, 2023 · b6e29fb · b6e29fb
1 parent 8d12e4e
commit b6e29fb
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/...ngBlocks/StackSdks/Auth/Masa.BuildingBlocks.StackSdks.Auth/Service/ICustomLoginService.cs b/...ngBlocks/StackSdks/Auth/Masa.BuildingBlocks.StackSdks.Auth/Service/ICustomLoginService.cs
@@ -5,5 +5,5 @@ namespace Masa.BuildingBlocks.StackSdks.Auth.Service;
 
 public interface ICustomLoginService
 {
-    Task<CustomLoginModel?> GetCustomLoginByClientIdAsync(string clientId);
+    Task<CustomLoginModel?> GetCustomLoginByClientIdAsync(string environment, string clientId);
 }
diff --git a/src/Contrib/StackSdks/Masa.Contrib.StackSdks.Auth/Service/CustomLoginService.cs b/src/Contrib/StackSdks/Masa.Contrib.StackSdks.Auth/Service/CustomLoginService.cs
@@ -12,9 +12,9 @@ public CustomLoginService(ICaller caller)
         _caller = caller;
     }
 
-    public async Task<CustomLoginModel?> GetCustomLoginByClientIdAsync(string clientId)
+    public async Task<CustomLoginModel?> GetCustomLoginByClientIdAsync(string environment, string clientId)
     {
         var requestUri = $"api/sso/customLogin/getByClientId";
-        return await _caller.GetAsync<object, CustomLoginModel>(requestUri, new { clientId });
+        return await _caller.GetAsync<object, CustomLoginModel>(requestUri, new { environment, clientId });
     }
 }
diff --git a/src/Contrib/StackSdks/Tests/Masa.Contrib.StackSdks.Auth.Tests/CustomLoginServiceTest.cs b/src/Contrib/StackSdks/Tests/Masa.Contrib.StackSdks.Auth.Tests/CustomLoginServiceTest.cs
@@ -15,7 +15,7 @@ public async Task TestGetCustomLoginByClientIdAsync()
         var caller = new Mock<ICaller>();
         caller.Setup(provider => provider.GetAsync<object, CustomLoginModel>(requestUri, It.IsAny<object>(), default)).ReturnsAsync(data).Verifiable();
         var customLoginService = new CustomLoginService(caller.Object);
-        var result = await customLoginService.GetCustomLoginByClientIdAsync(clientId);
+        var result = await customLoginService.GetCustomLoginByClientIdAsync("test", clientId);
         caller.Verify(provider => provider.GetAsync<object, CustomLoginModel>(requestUri, It.IsAny<object>(), default), Times.Once);
     }
 }