from PyObjCTools.TestSupport import *
from SystemConfiguration import *
import socket

from test_scnetwork import resolver_available

class TestSCNetworkReachability (TestCase):
    def testTypes(self):
        self.assertIsInstance(SCNetworkReachabilityRef, objc.objc_class)

    @onlyIf(resolver_available(), "No DNS resolver available")
    def testFunctions(self):
        self.assertResultIsCFRetained(SCNetworkReachabilityCreateWithAddressPair)
        v = SCNetworkReachabilityCreateWithAddressPair(None, 
                ('0.0.0.0', 20990),
                ('www.python.org', 80))


        sd = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        sd.listen(5)

        self.assertResultIsCFRetained(SCNetworkReachabilityCreateWithAddress)
        ref = v = SCNetworkReachabilityCreateWithAddress(None, sd.getsockname())
        self.assertIsInstance(v, SCNetworkReachabilityRef)

        self.assertResultIsCFRetained(SCNetworkReachabilityCreateWithName)
        v = SCNetworkReachabilityCreateWithName(None, b'www.python.org')
        self.assertIsInstance(v, SCNetworkReachabilityRef)

        v = SCNetworkReachabilityGetTypeID()
        self.assertIsInstance(v, (int, long))

        self.assertResultIsBOOL(SCNetworkReachabilityGetFlags)
        v, fl = SCNetworkReachabilityGetFlags(ref, None)
        self.assertTrue(v)
        self.assertIsInstance(fl, (int, long))


        l = []
        def callout(ref, flags, ctx):
            l.append([ref, flags, ctx])
            print flags
        ctx = object()
        v = SCNetworkReachabilitySetCallback(ref, callout, ctx)
        self.assertTrue(v is True)


        rl = CFRunLoopGetCurrent()
        self.assertResultIsBOOL(SCNetworkReachabilityScheduleWithRunLoop)
        r = SCNetworkReachabilityScheduleWithRunLoop(ref, rl, kCFRunLoopCommonModes)

        sd.close()
        CFRunLoopRunInMode(kCFRunLoopDefaultMode, 1.0, False)

        self.assertResultIsBOOL(SCNetworkReachabilityUnscheduleFromRunLoop)
        r = SCNetworkReachabilityUnscheduleFromRunLoop(ref, rl, kCFRunLoopCommonModes)

        








if __name__ == "__main__":
    main()