import GameKit
from PyObjCTools.TestSupport import TestCase


class TestGKChallengeEventHandlerHelper(GameKit.NSObject):
    def shouldShowBannerForLocallyReceivedChallenge_(self, a):
        return 1

    def shouldShowBannerForLocallyCompletedChallenge_(self, a):
        return 1

    def shouldShowBannerForRemotelyCompletedChallenge_(self, a):
        return 1


class TestGKChallengeEventHandler(TestCase):
    def testProtocols(self):
        self.assertProtocolExists("GKChallengeEventHandlerDelegate")

    def testProtocolMethods(self):
        self.assertResultIsBOOL(
            TestGKChallengeEventHandlerHelper.shouldShowBannerForLocallyReceivedChallenge_
        )
        self.assertResultIsBOOL(
            TestGKChallengeEventHandlerHelper.shouldShowBannerForLocallyCompletedChallenge_
        )
        self.assertResultIsBOOL(
            TestGKChallengeEventHandlerHelper.shouldShowBannerForRemotelyCompletedChallenge_
        )
