@@static const char* samlResponse =

"</script>"

"</html>";

static constexpr Seconds actionAbsenceTimeout = 300_ms;

static constexpr Seconds actionDoneTimeout = 1000_ms;

namespace {

class ScopedNotificationCenterObserveOnce {

public:

    ScopedNotificationCenterObserveOnce() = default;

    ScopedNotificationCenterObserveOnce(NSNotificationName name, NSObject* object)

    {

        m_observer = [m_center addObserverForName:name object:object queue:nil usingBlock:[this] (NSNotification *) {

            [m_center removeObserver:m_observer.get()];

            m_observer = nullptr;

            m_didObserve = true;

            m_center = nullptr;

        }];

    }

    ~ScopedNotificationCenterObserveOnce()

    {

        if (!m_observer)

            return;

        [m_center removeObserver: m_observer.get()];

    }

    ScopedNotificationCenterObserveOnce(ScopedNotificationCenterObserveOnce&&) = default;

    ScopedNotificationCenterObserveOnce& operator=(ScopedNotificationCenterObserveOnce&&) = default;

    bool waitFor(Seconds timeout)

    {

        return TestWebKitAPI::Util::runFor(&m_didObserve, timeout);

    }

    operator bool() const { return m_didObserve; }

private:

    NSNotificationCenter * __weak m_center = [NSNotificationCenter defaultCenter];

    RetainPtr<NSObject> m_observer;

    bool m_didObserve { false };

};

}

@interface TestSOAuthorizationBasicDelegate : NSObject <WKNavigationDelegate>

@end

@@TEST(SOAuthorizationRedirect, InterceptionSucceedSuppressActiveSession)

{

    resetState();

    SWIZZLE_SOAUTH(PAL::getSOAuthorizationClass());

959

999

    RetainPtr<NSURL> testURL = [[NSBundle mainBundle] URLForResource:@"simple" withExtension:@"html" subdirectory:@"TestWebKitAPI.resources"];

    auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:CGRectMake(0, 0, 320, 500)]);

@@TEST(SOAuthorizationRedirect, DismissUIDuringMiniaturization)

    }];

    Util::run(&uiShowed);

1450 bool didEndSheet = false;

14511490 auto *hostWindow = [webView hostWindow];

    auto observer = adoptNS([[NSNotificationCenter defaultCenter] addObserverForName:NSWindowDidEndSheetNotification object:hostWindow queue:nil usingBlock:[&didEndSheet] (NSNotification *) {

        didEndSheet = true;

    }]);

    ScopedNotificationCenterObserveOnce didEndSheet { NSWindowDidEndSheetNotification, hostWindow };

    ScopedNotificationCenterObserveOnce didMiniaturize { NSWindowDidMiniaturizeNotification, hostWindow };

14551494 [hostWindow miniaturize:hostWindow];

1495 EXPECT_TRUE(didMiniaturize.waitFor(actionDoneTimeout));

14561496

1457 // Dimiss the UI

1497 // Dimiss the UI while it is miniaturized.

14581498 [gDelegate authorizationDidCancel:gAuthorization];

1459 EXPECT_FALSE(didEndSheet);

14601499

1461 // UI is only dimissed after the hostWindow is deminimized.

    // This is the condition that the test tests: it is expected that the sheet is not ended because the view is miniaturized.

    EXPECT_FALSE(didEndSheet.waitFor(actionAbsenceTimeout));

    ScopedNotificationCenterObserveOnce didDeminiaturize { NSWindowDidDeminiaturizeNotification, hostWindow };

14621504 [hostWindow deminiaturize:hostWindow];

1463 EXPECT_TRUE(didEndSheet);

    EXPECT_TRUE(didDeminiaturize.waitFor(actionDoneTimeout));

    EXPECT_TRUE(didEndSheet.waitFor(actionDoneTimeout));

}

TEST(SOAuthorizationRedirect, DismissUIDuringHiding)

@@TEST(SOAuthorizationRedirect, DismissUIDuringHiding)

        EXPECT_TRUE(success);

    }];

    Util::run(&uiShowed);

1533 auto *hostWindow = [webView hostWindow];

14891534

    bool didEndSheet = false;

    auto observer = adoptNS([[NSNotificationCenter defaultCenter] addObserverForName:NSWindowDidEndSheetNotification object:[webView hostWindow] queue:nil usingBlock:[&didEndSheet] (NSNotification *) {

        didEndSheet = true;

    }]);

1535 ScopedNotificationCenterObserveOnce didEndSheet { NSWindowDidEndSheetNotification, hostWindow };

    [NSApp hide:NSApp];

    // Wait for hiding to finish.

    while (!NSApp.hidden)

        Util::spinRunLoop();

14971543 // Dimiss the UI

14981544 [gDelegate authorizationDidCancel:gAuthorization];

1499 EXPECT_FALSE(didEndSheet);

    // This is the condition that the test tests: it is expected that the sheet is not ended because the view is hidden.

    EXPECT_FALSE(didEndSheet.waitFor(actionAbsenceTimeout));

    // UI is only dimissed after the hostApp is unhidden.

    [NSApp unhide:NSApp];

1503 EXPECT_TRUE(didEndSheet);

    while (NSApp.hidden)

        Util::spinRunLoop();

    EXPECT_TRUE(didEndSheet.waitFor(actionDoneTimeout));

}

TEST(SOAuthorizationRedirect, DismissUIDuringMiniaturizationThenAnother)

@@TEST(SOAuthorizationRedirect, DismissUIDuringMiniaturizationThenAnother)

        EXPECT_TRUE(success);

    }];

    Util::run(&uiShowed);

1529

1530 bool didEndSheet = false;

15311580 auto *hostWindow = [webView hostWindow];

    auto observer = adoptNS([[NSNotificationCenter defaultCenter] addObserverForName:NSWindowDidEndSheetNotification object:hostWindow queue:nil usingBlock:[&didEndSheet] (NSNotification *) {

        didEndSheet = true;

    }]);

    ScopedNotificationCenterObserveOnce didEndSheet { NSWindowDidEndSheetNotification, hostWindow };

    ScopedNotificationCenterObserveOnce didMiniaturize { NSWindowDidMiniaturizeNotification, hostWindow };

15351584 [hostWindow miniaturize:hostWindow];

1585 didMiniaturize.waitFor(actionDoneTimeout);

    // Dimiss the UI

    [gDelegate authorizationDidCancel:gAuthorization];

1539 EXPECT_FALSE(didEndSheet);

    // This is the condition that the test tests: it is expected that the sheet is not ended because the view is miniaturized.

    EXPECT_FALSE(didEndSheet.waitFor(actionAbsenceTimeout));

    // Load another AppSSO request.

    authorizationPerformed = false;

@@TEST(SOAuthorizationRedirect, DismissUIDuringMiniaturizationThenAnother)

    Util::run(&authorizationPerformed);

    EXPECT_TRUE(policyForAppSSOPerformed);

1548 // UI is only dimissed after the hostApp is unhidden.

    // This is the condition that the test tests: it is expected that the sheet is not ended because the view is miniaturized.

    EXPECT_FALSE(didEndSheet.waitFor(actionAbsenceTimeout));

    // UI is only dimissed after the window is deminiaturized.

    ScopedNotificationCenterObserveOnce didDeminiaturize { NSWindowDidDeminiaturizeNotification, hostWindow };

15491605 [hostWindow deminiaturize:hostWindow];

1550 EXPECT_TRUE(didEndSheet);

    EXPECT_TRUE(didDeminiaturize.waitFor(actionDoneTimeout));

    EXPECT_TRUE(didEndSheet.waitFor(actionDoneTimeout));

}

TEST(SOAuthorizationRedirect, DismissUIDuringHidingThenAnother)

@@TEST(SOAuthorizationRedirect, DismissUIDuringHidingThenAnother)

    }];

    Util::run(&uiShowed);

    bool didEndSheet = false;

    auto observer = adoptNS([[NSNotificationCenter defaultCenter] addObserverForName:NSWindowDidEndSheetNotification object:[webView hostWindow] queue:nil usingBlock:[&didEndSheet] (NSNotification *) {

        didEndSheet = true;

    }]);

1635 ScopedNotificationCenterObserveOnce didEndSheet { NSWindowDidEndSheetNotification, [webView hostWindow] };

15811636

15821637 [NSApp hide:NSApp];

1638 while (!NSApp.hidden)

1639 Util::spinRunLoop();

    // Dimiss the UI

    [gDelegate authorizationDidCancel:gAuthorization];

1586 EXPECT_FALSE(didEndSheet);

    // This is the condition that the test tests: it is expected that the sheet is not ended because the view is hidden.

    EXPECT_FALSE(didEndSheet.waitFor(actionAbsenceTimeout));

    // Load another AppSSO request.

    authorizationPerformed = false;

@@TEST(SOAuthorizationRedirect, DismissUIDuringHidingThenAnother)

    Util::run(&authorizationPerformed);

    EXPECT_TRUE(policyForAppSSOPerformed);

1595 // UI is only dimissed after the hostApp is unhidden.

15961654 [NSApp unhide:NSApp];

1597 EXPECT_TRUE(didEndSheet);

    while (NSApp.hidden)

        Util::spinRunLoop();

    EXPECT_TRUE(didEndSheet.waitFor(actionDoneTimeout));

}

#endif

@@TEST(SOAuthorizationPopUp, InterceptionSucceedSuppressActiveSession)

    EXPECT_TRUE(policyForAppSSOPerformed);

    // Suppress the last active session.

2010

2071

    auto configuration = adoptNS([webView.get().configuration copy]);

    auto messageHandler = adoptNS([[TestSOAuthorizationScriptMessageHandler alloc] initWithExpectation:@[@"Hello.", @"WindowClosed."]]);

    [[configuration userContentController] addScriptMessageHandler:messageHandler.get() name:@"testHandler"];

@@TEST(SOAuthorizationSubFrame, InterceptionError)

    EXPECT_TRUE(policyForAppSSOPerformed);

    [messageHandler extendExpectations:@[@"null", @"SOAuthorizationDidCancel", @""]];

2270

2331

    [gDelegate authorization:gAuthorization didCompleteWithError:adoptNS([[NSError alloc] initWithDomain:NSCocoaErrorDomain code:0 userInfo:nil]).get()];

    Util::run(&allMessagesReceived);

    // Make sure we don't load the request of the iframe to the main frame.

@@TEST(SOAuthorizationSubFrame, InterceptionSuccessTwice)

        policyForAppSSOPerformed = false;

        [messageHandler resetExpectations:@[@"http://www.example.com", @"SOAuthorizationDidStart"]];

2430

2491

        [webView loadHTMLString:testHtml baseURL:nil];

        Util::run(&allMessagesReceived);

COMMIT_MESSAGE

Tools/TestWebKitAPI/Tests/WebKitCocoa/SOAuthorizationTests.mm