mocking una connessione socket in Python

Question

Sto provando a scrivere unit test per una classe in python. La classe apre una presa TCP su init. Sto cercando di deriderlo in modo tale da poter affermare che la connessione viene chiamata con i valori corretti, ma ovviamente non si verifica effettivamente nei test unitari. Ho stancato MagicMock, patch, ecc. Ma non ho trovato una soluzione.

La mia classe finora assomiglia a questo

import socket 

class MyClass(object): 

    def __init__(self): 
     self.tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 
     self.tcp_socket.connect('0.0.0.0', '6767') 

Answer 1

Se si desidera solo per affermare che connect si chiama correttamente, si tratta di un semplice

import mock 
import socket 

class MyClass(object): 

    def __init__(self): 
     self.tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) 
     self.tcp_socket.connect('0.0.0.0', '6767') 

with mock.patch('socket.socket'): 
    c = MyClass() 
    c.tcp_socket.connect.assert_called_with('0.0.0.0', '6767') 

Se si deve importare un modulo prima di accedere a MyClass, è necessario regolare leggermente la patch:

from mymodule import MyClass 
import mock 

with mock.patch('mymodule.socket.socket'): 
    c = MyClass() 
    c.tcp_socket.connect.assert_called_with('0.0.0.0', '6767')