patch_socket.py

import sys


class TestTriedToAccessNetwork(BaseException):
    pass

if 'test' in sys.argv:
    import socket

    class PatchedSocket(socket.socket):
        def __init__(self, *args, **kwargs):
            super(PatchedSocket, self).__init__(*args, **kwargs)

        def connect(self, address):
            print 'address = %s' % str(address)
            if len(address) == 2:
                addr, _ = address
            elif len(address) == 4:
                addr, _, _, _ = address
            if addr and (addr != '127.0.0.1' and addr != '::1'):
                #print "FAIL: %s" % addr
                #traceback.print_stack()
                raise TestTriedToAccessNetwork("Tried connecting to external resource: %s" % addr)
            super(PatchedSocket, self).connect(address)

    socket.socket = PatchedSocket