mirror of
https://github.com/ppy/osu.git
synced 2026-05-21 03:39:53 +08:00
Bartłomiej Dach
369 lines
17 KiB
C#
369 lines
17 KiB
C#
// Copyright (c) ppy Pty Ltd <contact@ppy.sh>. Licensed under the MIT Licence.
|
|
// See the LICENCE file in the repository root for full licence text.
|
|
|
|
using System;
|
|
using System.Linq;
|
|
using System.Net;
|
|
using NUnit.Framework;
|
|
using osu.Framework.Allocation;
|
|
using osu.Framework.Graphics;
|
|
using osu.Framework.Graphics.Containers;
|
|
using osu.Framework.Testing;
|
|
using osu.Game.Configuration;
|
|
using osu.Game.Graphics.Sprites;
|
|
using osu.Game.Graphics.UserInterface;
|
|
using osu.Game.Online;
|
|
using osu.Game.Online.API;
|
|
using osu.Game.Online.API.Requests;
|
|
using osu.Game.Online.API.Requests.Responses;
|
|
using osu.Game.Overlays;
|
|
using osu.Game.Overlays.Login;
|
|
using osu.Game.Overlays.Settings;
|
|
using osu.Game.Tests.Visual.Online;
|
|
using osu.Game.Users;
|
|
using osu.Game.Users.Drawables;
|
|
using osuTK.Input;
|
|
|
|
namespace osu.Game.Tests.Visual.Menus
|
|
{
|
|
[TestFixture]
|
|
public partial class TestSceneLoginOverlay : OsuManualInputManagerTestScene
|
|
{
|
|
private DummyAPIAccess dummyAPI => (DummyAPIAccess)API;
|
|
|
|
private LoginOverlay loginOverlay = null!;
|
|
private OsuConfigManager localConfig = null!;
|
|
|
|
[Cached(typeof(LocalUserStatisticsProvider))]
|
|
private readonly TestSceneUserPanel.TestUserStatisticsProvider statisticsProvider = new TestSceneUserPanel.TestUserStatisticsProvider();
|
|
|
|
[BackgroundDependencyLoader]
|
|
private void load()
|
|
{
|
|
Dependencies.Cache(localConfig = new OsuConfigManager(LocalStorage));
|
|
|
|
Child = loginOverlay = new LoginOverlay
|
|
{
|
|
Anchor = Anchor.Centre,
|
|
Origin = Anchor.Centre,
|
|
};
|
|
}
|
|
|
|
[SetUpSteps]
|
|
public void SetUpSteps()
|
|
{
|
|
AddStep("reset online state", () => localConfig.SetValue(OsuSetting.UserOnlineStatus, UserStatus.Online));
|
|
AddStep("show login overlay", () => loginOverlay.Show());
|
|
}
|
|
|
|
[Test]
|
|
public void TestLoginSuccess_EmailVerification()
|
|
{
|
|
AddStep("logout", () => API.Logout());
|
|
assertAPIState(APIState.Offline);
|
|
|
|
AddStep("enter password", () => loginOverlay.ChildrenOfType<OsuPasswordTextBox>().First().Text = "password");
|
|
AddStep("submit", () => loginOverlay.ChildrenOfType<OsuButton>().First(b => b.Text.ToString() == "Sign in").TriggerClick());
|
|
|
|
assertAPIState(APIState.RequiresSecondFactorAuth);
|
|
AddUntilStep("wait for second factor auth form", () => loginOverlay.ChildrenOfType<SecondFactorAuthForm>().SingleOrDefault(), () => Is.Not.Null);
|
|
|
|
AddStep("set up verification handling", () => dummyAPI.HandleRequest = req =>
|
|
{
|
|
switch (req)
|
|
{
|
|
case VerifySessionRequest verifySessionRequest:
|
|
if (verifySessionRequest.VerificationKey == "88800088")
|
|
verifySessionRequest.TriggerSuccess();
|
|
else
|
|
verifySessionRequest.TriggerFailure(new WebException());
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
});
|
|
|
|
AddStep("enter code", () => loginOverlay.ChildrenOfType<OsuTextBox>().First().Text = "88800088");
|
|
assertAPIState(APIState.Online);
|
|
assertDropdownState(UserAction.Online);
|
|
|
|
AddStep("set failing", () => { dummyAPI.SetState(APIState.Failing); });
|
|
AddStep("return to online", () => { dummyAPI.SetState(APIState.Online); });
|
|
|
|
AddStep("clear handler", () => dummyAPI.HandleRequest = null);
|
|
|
|
assertDropdownState(UserAction.Online);
|
|
AddStep("change user state", () => localConfig.SetValue(OsuSetting.UserOnlineStatus, UserStatus.DoNotDisturb));
|
|
assertDropdownState(UserAction.DoNotDisturb);
|
|
}
|
|
|
|
[Test]
|
|
public void TestLoginSuccess_TOTPVerification()
|
|
{
|
|
AddStep("logout", () => API.Logout());
|
|
assertAPIState(APIState.Offline);
|
|
AddStep("expect totp verification", () => dummyAPI.SessionVerificationMethod = SessionVerificationMethod.TimedOneTimePassword);
|
|
|
|
AddStep("enter password", () => loginOverlay.ChildrenOfType<OsuPasswordTextBox>().First().Text = "password");
|
|
AddStep("submit", () => loginOverlay.ChildrenOfType<OsuButton>().First(b => b.Text.ToString() == "Sign in").TriggerClick());
|
|
|
|
assertAPIState(APIState.RequiresSecondFactorAuth);
|
|
AddUntilStep("wait for second factor auth form", () => loginOverlay.ChildrenOfType<SecondFactorAuthForm>().SingleOrDefault(), () => Is.Not.Null);
|
|
|
|
AddStep("set up verification handling", () => dummyAPI.HandleRequest = req =>
|
|
{
|
|
switch (req)
|
|
{
|
|
case VerifySessionRequest verifySessionRequest:
|
|
if (verifySessionRequest.VerificationKey == "012345")
|
|
verifySessionRequest.TriggerSuccess();
|
|
else
|
|
verifySessionRequest.TriggerFailure(new WebException());
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
});
|
|
|
|
AddStep("enter code", () => loginOverlay.ChildrenOfType<OsuTextBox>().First().Text = "012345");
|
|
assertAPIState(APIState.Online);
|
|
assertDropdownState(UserAction.Online);
|
|
|
|
AddStep("set failing", () => { dummyAPI.SetState(APIState.Failing); });
|
|
AddStep("return to online", () => { dummyAPI.SetState(APIState.Online); });
|
|
|
|
AddStep("clear handler", () => dummyAPI.HandleRequest = null);
|
|
|
|
assertDropdownState(UserAction.Online);
|
|
AddStep("change user state", () => localConfig.SetValue(OsuSetting.UserOnlineStatus, UserStatus.DoNotDisturb));
|
|
assertDropdownState(UserAction.DoNotDisturb);
|
|
}
|
|
|
|
[Test]
|
|
public void TestLoginSuccess_TOTPVerification_FallbackToEmail()
|
|
{
|
|
AddStep("logout", () => API.Logout());
|
|
assertAPIState(APIState.Offline);
|
|
AddStep("expect totp verification", () => dummyAPI.SessionVerificationMethod = SessionVerificationMethod.TimedOneTimePassword);
|
|
|
|
AddStep("enter password", () => loginOverlay.ChildrenOfType<OsuPasswordTextBox>().First().Text = "password");
|
|
AddStep("submit", () => loginOverlay.ChildrenOfType<OsuButton>().First(b => b.Text.ToString() == "Sign in").TriggerClick());
|
|
|
|
assertAPIState(APIState.RequiresSecondFactorAuth);
|
|
AddUntilStep("wait for second factor auth form", () => loginOverlay.ChildrenOfType<SecondFactorAuthForm>().SingleOrDefault(), () => Is.Not.Null);
|
|
|
|
AddStep("set up verification handling", () => dummyAPI.HandleRequest = req =>
|
|
{
|
|
switch (req)
|
|
{
|
|
case VerifySessionRequest verifySessionRequest:
|
|
if (verifySessionRequest.VerificationKey == "deadbeef")
|
|
verifySessionRequest.TriggerSuccess();
|
|
else
|
|
verifySessionRequest.TriggerFailure(new WebException());
|
|
return true;
|
|
|
|
case VerificationMailFallbackRequest verificationMailFallbackRequest:
|
|
verificationMailFallbackRequest.TriggerSuccess();
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
});
|
|
|
|
AddStep("request fallback to email", () =>
|
|
{
|
|
InputManager.MoveMouseTo(this.ChildrenOfType<OsuSpriteText>().Single(t => t.Text.ToString().Contains("email", StringComparison.InvariantCultureIgnoreCase)));
|
|
InputManager.Click(MouseButton.Left);
|
|
});
|
|
|
|
AddStep("enter code", () => loginOverlay.ChildrenOfType<OsuTextBox>().First().Text = "deadbeef");
|
|
assertAPIState(APIState.Online);
|
|
assertDropdownState(UserAction.Online);
|
|
|
|
AddStep("set failing", () => { dummyAPI.SetState(APIState.Failing); });
|
|
AddStep("return to online", () => { dummyAPI.SetState(APIState.Online); });
|
|
|
|
AddStep("clear handler", () => dummyAPI.HandleRequest = null);
|
|
|
|
assertDropdownState(UserAction.Online);
|
|
AddStep("change user state", () => localConfig.SetValue(OsuSetting.UserOnlineStatus, UserStatus.DoNotDisturb));
|
|
assertDropdownState(UserAction.DoNotDisturb);
|
|
}
|
|
|
|
[Test]
|
|
public void TestLoginSuccess_TOTPVerification_TurnedOffMidwayThrough()
|
|
{
|
|
bool firstAttemptHandled = false;
|
|
|
|
AddStep("logout", () => API.Logout());
|
|
assertAPIState(APIState.Offline);
|
|
AddStep("expect totp verification", () => dummyAPI.SessionVerificationMethod = SessionVerificationMethod.TimedOneTimePassword);
|
|
|
|
AddStep("enter password", () => loginOverlay.ChildrenOfType<OsuPasswordTextBox>().First().Text = "password");
|
|
AddStep("submit", () => loginOverlay.ChildrenOfType<OsuButton>().First(b => b.Text.ToString() == "Sign in").TriggerClick());
|
|
|
|
assertAPIState(APIState.RequiresSecondFactorAuth);
|
|
AddUntilStep("wait for second factor auth form", () => loginOverlay.ChildrenOfType<SecondFactorAuthForm>().SingleOrDefault(), () => Is.Not.Null);
|
|
|
|
AddStep("set up verification handling", () => dummyAPI.HandleRequest = req =>
|
|
{
|
|
switch (req)
|
|
{
|
|
case VerifySessionRequest verifySessionRequest:
|
|
verifySessionRequest.RequiredVerificationMethod = SessionVerificationMethod.EmailMessage;
|
|
verifySessionRequest.TriggerFailure(new WebException());
|
|
firstAttemptHandled = true;
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
});
|
|
|
|
AddStep("enter code", () => loginOverlay.ChildrenOfType<OsuTextBox>().First().Text = "123456");
|
|
AddUntilStep("first verification attempt handled", () => firstAttemptHandled);
|
|
assertAPIState(APIState.RequiresSecondFactorAuth);
|
|
|
|
AddStep("set up verification handling", () => dummyAPI.HandleRequest = req =>
|
|
{
|
|
switch (req)
|
|
{
|
|
case VerifySessionRequest verifySessionRequest:
|
|
if (verifySessionRequest.VerificationKey == "deadbeef")
|
|
verifySessionRequest.TriggerSuccess();
|
|
else
|
|
verifySessionRequest.TriggerFailure(new WebException());
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
});
|
|
AddStep("enter code", () => loginOverlay.ChildrenOfType<OsuTextBox>().First().Text = "deadbeef");
|
|
assertAPIState(APIState.Online);
|
|
assertDropdownState(UserAction.Online);
|
|
}
|
|
|
|
private void assertDropdownState(UserAction state)
|
|
{
|
|
AddAssert($"dropdown state is {state}", () => loginOverlay.ChildrenOfType<UserDropdown>().First().Current.Value, () => Is.EqualTo(state));
|
|
}
|
|
|
|
private void assertAPIState(APIState expected) =>
|
|
AddUntilStep($"login state is {expected}", () => API.State.Value, () => Is.EqualTo(expected));
|
|
|
|
[Test]
|
|
public void TestVerificationFailure()
|
|
{
|
|
bool verificationHandled = false;
|
|
AddStep("reset flag", () => verificationHandled = false);
|
|
AddStep("logout", () => API.Logout());
|
|
assertAPIState(APIState.Offline);
|
|
|
|
AddStep("enter password", () => loginOverlay.ChildrenOfType<OsuPasswordTextBox>().First().Text = "password");
|
|
AddStep("submit", () => loginOverlay.ChildrenOfType<OsuButton>().First(b => b.Text.ToString() == "Sign in").TriggerClick());
|
|
|
|
assertAPIState(APIState.RequiresSecondFactorAuth);
|
|
AddUntilStep("wait for second factor auth form", () => loginOverlay.ChildrenOfType<SecondFactorAuthForm>().SingleOrDefault(), () => Is.Not.Null);
|
|
|
|
AddStep("set up verification handling", () => dummyAPI.HandleRequest = req =>
|
|
{
|
|
switch (req)
|
|
{
|
|
case VerifySessionRequest verifySessionRequest:
|
|
if (verifySessionRequest.VerificationKey == "88800088")
|
|
verifySessionRequest.TriggerSuccess();
|
|
else
|
|
verifySessionRequest.TriggerFailure(new WebException());
|
|
verificationHandled = true;
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
});
|
|
AddStep("enter code", () => loginOverlay.ChildrenOfType<OsuTextBox>().First().Text = "abcdefgh");
|
|
AddUntilStep("wait for verification handled", () => verificationHandled);
|
|
assertAPIState(APIState.RequiresSecondFactorAuth);
|
|
AddStep("clear handler", () => dummyAPI.HandleRequest = null);
|
|
}
|
|
|
|
[Test]
|
|
public void TestLoginFailure()
|
|
{
|
|
AddStep("logout", () =>
|
|
{
|
|
API.Logout();
|
|
((DummyAPIAccess)API).FailNextLogin();
|
|
});
|
|
|
|
AddStep("enter password", () => loginOverlay.ChildrenOfType<OsuPasswordTextBox>().First().Text = "password");
|
|
AddStep("submit", () => loginOverlay.ChildrenOfType<OsuButton>().First(b => b.Text.ToString() == "Sign in").TriggerClick());
|
|
}
|
|
|
|
[Test]
|
|
public void TestLoginConnecting()
|
|
{
|
|
AddStep("logout", () =>
|
|
{
|
|
API.Logout();
|
|
((DummyAPIAccess)API).PauseOnConnectingNextLogin();
|
|
});
|
|
|
|
AddStep("enter password", () => loginOverlay.ChildrenOfType<OsuPasswordTextBox>().First().Text = "password");
|
|
AddStep("submit", () => loginOverlay.ChildrenOfType<OsuButton>().First(b => b.Text.ToString() == "Sign in").TriggerClick());
|
|
}
|
|
|
|
[Test]
|
|
public void TestClickingOnFlagClosesOverlay()
|
|
{
|
|
AddStep("logout", () => API.Logout());
|
|
AddStep("enter password", () => loginOverlay.ChildrenOfType<OsuPasswordTextBox>().First().Text = "password");
|
|
AddStep("submit", () => loginOverlay.ChildrenOfType<OsuButton>().First(b => b.Text.ToString() == "Sign in").TriggerClick());
|
|
|
|
assertAPIState(APIState.RequiresSecondFactorAuth);
|
|
AddUntilStep("wait for second factor auth form", () => loginOverlay.ChildrenOfType<SecondFactorAuthForm>().SingleOrDefault(), () => Is.Not.Null);
|
|
|
|
AddStep("enter code", () => loginOverlay.ChildrenOfType<OsuTextBox>().First().Text = "88800088");
|
|
assertAPIState(APIState.Online);
|
|
|
|
AddStep("feed statistics", () => statisticsProvider.UpdateStatistics(new UserStatistics(), Ruleset.Value));
|
|
AddStep("click on flag", () =>
|
|
{
|
|
InputManager.MoveMouseTo(loginOverlay.ChildrenOfType<UpdateableFlag>().First());
|
|
InputManager.Click(MouseButton.Left);
|
|
});
|
|
AddAssert("login overlay is hidden", () => loginOverlay.State.Value == Visibility.Hidden);
|
|
}
|
|
|
|
[Test]
|
|
public void TestUncheckingRememberUsernameClearsIt()
|
|
{
|
|
AddStep("logout", () => API.Logout());
|
|
AddStep("set username", () => localConfig.SetValue(OsuSetting.Username, "test_user"));
|
|
AddStep("set remember password", () => localConfig.SetValue(OsuSetting.SavePassword, true));
|
|
AddStep("uncheck remember username", () =>
|
|
{
|
|
InputManager.MoveMouseTo(loginOverlay.ChildrenOfType<SettingsCheckbox>().First());
|
|
InputManager.Click(MouseButton.Left);
|
|
});
|
|
AddAssert("remember username off", () => localConfig.Get<bool>(OsuSetting.SaveUsername), () => Is.False);
|
|
AddAssert("remember password off", () => localConfig.Get<bool>(OsuSetting.SavePassword), () => Is.False);
|
|
AddAssert("username cleared", () => localConfig.Get<string>(OsuSetting.Username), () => Is.Empty);
|
|
}
|
|
|
|
[Test]
|
|
public void TestUncheckingRememberPasswordClearsToken()
|
|
{
|
|
AddStep("logout", () => API.Logout());
|
|
AddStep("set token", () => localConfig.SetValue(OsuSetting.Token, "test_token"));
|
|
AddStep("set remember password", () => localConfig.SetValue(OsuSetting.SavePassword, true));
|
|
AddStep("uncheck remember token", () =>
|
|
{
|
|
InputManager.MoveMouseTo(loginOverlay.ChildrenOfType<SettingsCheckbox>().Last());
|
|
InputManager.Click(MouseButton.Left);
|
|
});
|
|
AddAssert("remember password off", () => localConfig.Get<bool>(OsuSetting.SavePassword), () => Is.False);
|
|
AddAssert("token cleared", () => localConfig.Get<string>(OsuSetting.Token), () => Is.Empty);
|
|
}
|
|
}
|
|
}
|