Vá một hàm bằng cách sử dụng Mock

Question

Tôi có một bộ kiểm tra (sử dụng mũi, không phải là không có), và tôi muốn vá một hàm để trả về một chuỗi giá trị cụ thể cho mọi thử nghiệm trong lớp thử nghiệm. Nỗ lực đầu tiên của tôi, sử dụng một ví dụ đơn giản, là:

@patch('time.clock', MagicMock(side_effects=[1, 2])) 
class Tests: 
    def test_1(self): 
     assert time.clock() == 1 
     assert time.clock() == 2 

    def test_2(self): 
     assert time.clock() == 1 
     assert time.clock() == 2 

Tuy nhiên, trường hợp MagicMock chỉ được tạo ra một lần, vì vậy thử nghiệm thứ hai thất bại khi các tác dụng phụ chạy ra ngoài. Tôi có thể vá từng phương pháp thử nghiệm một cách riêng biệt, nhưng tôi không thực sự muốn sao chép trang trí bản vá trên tất cả chúng (có nhiều thử nghiệm hơn trong ví dụ này!) Cách khác tôi có thể làm là tạo bản vá trong mã thiết lập như sau:

class Tests: 
    def setup(self): 
     self.old_clock = time.clock 
     time.clock = MagicMock(side_effects=[1, 2]) 

    def teardown(self): 
     time.clock = self.old_clock 

    def test_1(self): 
     assert time.clock() == 1 
     assert time.clock() == 2 

    def test_2(self): 
     assert time.clock() == 1 
     assert time.clock() == 2 

Tuy nhiên, lưu và khôi phục định nghĩa ban đầu có vẻ như một cái gì đó mà Mock sẽ có thể thực hiện tự động. Có cách nào khác để làm điều này mà tôi đang thiếu? Hay là ví dụ cuối cùng của tôi là cách tốt nhất để làm điều này?

Answer 1

a = (x for x in [1,2]) 

x = lambda : next(a) 

x() 

Out: 1

x() 

Out: 2

Đặt câu trả lời của bạn vào một danh sách. Thay đổi X cho tên bạn muốn.

Answer 2

Bạn chỉ nên áp dụng các bản vá cho mọi kiểm tra, thay vì áp dụng nó vào trong lớp:

class Tests: 
    @patch('time.clock', MagicMock(side_effects=[1, 2])) 
    def test_1(self): 
     assert time.clock() == 1 
     assert time.clock() == 2 

    @patch('time.clock', MagicMock(side_effects=[1, 2])) 
    def test_2(self): 
     assert time.clock() == 1 
     assert time.clock() == 2