def test_close(self): conn = Connection("localhost", 1234) conn.stream = Mock() self.assertEqual(conn.closing, False) conn.close() self.assertEqual(conn.closing, True) conn.stream.close.assert_called_once_with()