diff --git a/src/Discord.Net.Rest/BaseDiscordClient.cs b/src/Discord.Net.Rest/BaseDiscordClient.cs index fc938d04d..1837e38c0 100644 --- a/src/Discord.Net.Rest/BaseDiscordClient.cs +++ b/src/Discord.Net.Rest/BaseDiscordClient.cs @@ -65,7 +65,7 @@ namespace Discord.Rest } finally { _stateLock.Release(); } } - private async Task LoginInternalAsync(TokenType tokenType, string token, bool validateToken) + internal virtual async Task LoginInternalAsync(TokenType tokenType, string token, bool validateToken) { if (_isFirstLogin) { @@ -118,7 +118,7 @@ namespace Discord.Rest } finally { _stateLock.Release(); } } - private async Task LogoutInternalAsync() + internal virtual async Task LogoutInternalAsync() { if (LoginState == LoginState.LoggedOut) return; LoginState = LoginState.LoggingOut; diff --git a/src/Discord.Net.WebSocket/DiscordSocketRestClient.cs b/src/Discord.Net.WebSocket/DiscordSocketRestClient.cs index 2c9176d20..5107629a8 100644 --- a/src/Discord.Net.WebSocket/DiscordSocketRestClient.cs +++ b/src/Discord.Net.WebSocket/DiscordSocketRestClient.cs @@ -10,7 +10,11 @@ namespace Discord.WebSocket public new Task LoginAsync(TokenType tokenType, string token, bool validateToken = true) => throw new NotSupportedException("The Socket REST wrapper cannot be used to log in or out."); + internal override Task LoginInternalAsync(TokenType tokenType, string token, bool validateToken) + => throw new NotSupportedException("The Socket REST wrapper cannot be used to log in or out."); public new Task LogoutAsync() => throw new NotSupportedException("The Socket REST wrapper cannot be used to log in or out."); + internal override Task LogoutInternalAsync() + => throw new NotSupportedException("The Socket REST wrapper cannot be used to log in or out."); } }