Unit_test単体テストが知らない別のメソッドを呼び出すメソッド

Question

別のメソッドを呼び出す単純なメソッドをどのようにテストしますか？私は現在wssenderメソッドをテストしようとしています。

in worker.py 
---------------- 
class Worker(self): 
    def __init__(self, ws) 
     self.ws = ws 

    def wssender(self,str): 
     newstr = '<br>{0}'.format(str) 
     self.ws.sendMessage(newstr.encode()) 

とテストコード

in unit_test.py 
----------------- 
class SimpleTest(unittest.TestCase): 
    def test_wssender(self): 
     msg = 'test send message' 
     wss=worker.Worker 
     wss.wssender(wss, msg) 
     expected = "<br>test send message" 
     self.assertEqual(<something>, expected) 

2つの問題があります。このテストを実行するときに私は得る。私はこのケースでテストすることはよく分からないので

AttributeError: type object 'Worker' has no attribute 'ws' 

と wssenderは何も返しません（WSは、Webソケットです）。

Answer 1

どうやってですか？

import unittest 
from unittest.mock import MagicMock 

class Worker(object): 
    def __init__(self, ws): 
     self.ws = ws 

    def wssender(self, str): 
     newstr = '<br>{0}'.format(str) 
     self.ws.sendMessage(newstr.encode()) 

class WorkerTests(unittest.TestCase): 
    def test_wssender(self): 
     # Arrange 
     ws = MagicMock() 
     ws.sendMessage = MagicMock() 
     worker = Worker(ws) 

     # Act 
     worker.wssender('test send message') 

     # Assert 
     ws.sendMessage.assert_called_once_with(b'<br>test send message') 

if __name__ == '__main__': 
    unittest.main() 

出力

. 
---------------------------------------------------------------------- 
Ran 1 test in 0.000s 

OK 

Answer 2

私は "クラス労働者（自己）" タイプミスであり、それは本当に "クラスの労働者（オブジェクト）"（またはその他の基底クラス）だと仮定します。

ws = WhateverWsIsSupposedToBe() 
wss=worker.Worker(ws) 
wss.wssender(msg) 

全く関係のないもののwssender()は、メソッドの良い名前ではありません - senderは正午で、方法はアクションです：あなたが欲しい -

WRT /最初の問題は、あなたのテストでWorkerをインスタンス化するのを忘れ動詞を使用する必要がありますので、wssend()（またはちょうどsend() FWIW）が良いでしょう。

2番目の問題では、いくつかのオプションがあります。何をテストすることはメッセージのフォーマットである場合は、ちょうど別の方法でそれを分割：

class Worker(object): 
    def __init__(self, ws) 
     self.ws = ws 

    def prepare(self, msg): 
     return '<br>{0}'.format(msg).encode(...) 

    def send(self, msg): 
     self.ws.sendMessage(self.prepare(msg)) 

を今、あなたが孤立してWorker.prepare(msg)をテストすることができます。

Worker.send(msg)がself.ws.sendMessage(...)を呼び出して（期待される引数で呼び出す）ことをテストしたい場合は、wsをモックする必要があります。

class MockWs(object): 
    def __init__(self): 
     self.msg = None 
    def sendMessage(self, msg): 
     self.msg = msg 

class SimpleTest(unittest.TestCase): 
    def test_send(self): 
     ws = MockWs() 
     msg = 'test send message' 
     wss=worker.Worker(ws) 
     wss.send(msg) 
     expected = "<br>test send message" 
     self.assertEqual(ws.msg, expected) 

をそれとも、mockライブラリ、CF Tacgの答えを使用することができます：あなたはこれを手動で行うことができます。