Fri, 16 Jan 2015 18:13:44 +0100
Integrate suggestion from review to improve consistency with existing code.
michael@0 | 1 | |
michael@0 | 2 | <!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN" |
michael@0 | 3 | "http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd"> |
michael@0 | 4 | |
michael@0 | 5 | |
michael@0 | 6 | <html xmlns="http://www.w3.org/1999/xhtml"> |
michael@0 | 7 | <head> |
michael@0 | 8 | <meta http-equiv="Content-Type" content="text/html; charset=utf-8" /> |
michael@0 | 9 | |
michael@0 | 10 | <title>mocksignature — Mock 0.8.1alpha1 documentation</title> |
michael@0 | 11 | |
michael@0 | 12 | <link rel="stylesheet" href="_static/nature.css" type="text/css" /> |
michael@0 | 13 | <link rel="stylesheet" href="_static/pygments.css" type="text/css" /> |
michael@0 | 14 | |
michael@0 | 15 | <script type="text/javascript"> |
michael@0 | 16 | var DOCUMENTATION_OPTIONS = { |
michael@0 | 17 | URL_ROOT: '', |
michael@0 | 18 | VERSION: '0.8.1alpha1', |
michael@0 | 19 | COLLAPSE_INDEX: false, |
michael@0 | 20 | FILE_SUFFIX: '.html', |
michael@0 | 21 | HAS_SOURCE: true |
michael@0 | 22 | }; |
michael@0 | 23 | </script> |
michael@0 | 24 | <script type="text/javascript" src="_static/jquery.js"></script> |
michael@0 | 25 | <script type="text/javascript" src="_static/underscore.js"></script> |
michael@0 | 26 | <script type="text/javascript" src="_static/doctools.js"></script> |
michael@0 | 27 | <link rel="top" title="Mock 0.8.1alpha1 documentation" href="index.html" /> |
michael@0 | 28 | <link rel="next" title="Getting Started with Mock" href="getting-started.html" /> |
michael@0 | 29 | <link rel="prev" title="Mocking Magic Methods" href="magicmock.html" /> |
michael@0 | 30 | </head> |
michael@0 | 31 | <body> |
michael@0 | 32 | <div class="related"> |
michael@0 | 33 | <h3>Navigation</h3> |
michael@0 | 34 | <ul> |
michael@0 | 35 | <li class="right" style="margin-right: 10px"> |
michael@0 | 36 | <a href="genindex.html" title="General Index" |
michael@0 | 37 | accesskey="I">index</a></li> |
michael@0 | 38 | <li class="right" > |
michael@0 | 39 | <a href="getting-started.html" title="Getting Started with Mock" |
michael@0 | 40 | accesskey="N">next</a> |</li> |
michael@0 | 41 | <li class="right" > |
michael@0 | 42 | <a href="magicmock.html" title="Mocking Magic Methods" |
michael@0 | 43 | accesskey="P">previous</a> |</li> |
michael@0 | 44 | <li><a href="index.html">Mock 0.8.1alpha1 documentation</a> »</li> |
michael@0 | 45 | </ul> |
michael@0 | 46 | </div> |
michael@0 | 47 | |
michael@0 | 48 | <div class="document"> |
michael@0 | 49 | <div class="documentwrapper"> |
michael@0 | 50 | <div class="bodywrapper"> |
michael@0 | 51 | <div class="body"> |
michael@0 | 52 | |
michael@0 | 53 | <div class="section" id="mocksignature"> |
michael@0 | 54 | <h1>mocksignature<a class="headerlink" href="#mocksignature" title="Permalink to this headline">¶</a></h1> |
michael@0 | 55 | <div class="admonition note"> |
michael@0 | 56 | <p class="first admonition-title">Note</p> |
michael@0 | 57 | <p class="last"><a class="reference internal" href="helpers.html#auto-speccing"><em>Autospeccing</em></a>, added in mock 0.8, is a more advanced version of |
michael@0 | 58 | <cite>mocksignature</cite> and can be used for many of the same use cases.</p> |
michael@0 | 59 | </div> |
michael@0 | 60 | <p>A problem with using mock objects to replace real objects in your tests is that |
michael@0 | 61 | <a class="reference internal" href="mock.html#mock.Mock" title="mock.Mock"><tt class="xref py py-class docutils literal"><span class="pre">Mock</span></tt></a> can be <em>too</em> flexible. Your code can treat the mock objects in |
michael@0 | 62 | any way and you have to manually check that they were called correctly. If your |
michael@0 | 63 | code calls functions or methods with the wrong number of arguments then mocks |
michael@0 | 64 | don’t complain.</p> |
michael@0 | 65 | <p>The solution to this is <cite>mocksignature</cite>, which creates functions with the |
michael@0 | 66 | same signature as the original, but delegating to a mock. You can interrogate |
michael@0 | 67 | the mock in the usual way to check it has been called with the <em>right</em> |
michael@0 | 68 | arguments, but if it is called with the wrong number of arguments it will |
michael@0 | 69 | raise a <cite>TypeError</cite> in the same way your production code would.</p> |
michael@0 | 70 | <p>Another advantage is that your mocked objects are real functions, which can |
michael@0 | 71 | be useful when your code uses |
michael@0 | 72 | <a class="reference external" href="http://docs.python.org/library/inspect.html">inspect</a> or depends on |
michael@0 | 73 | functions being function objects.</p> |
michael@0 | 74 | <dl class="function"> |
michael@0 | 75 | <dt id="mock.mocksignature"> |
michael@0 | 76 | <tt class="descname">mocksignature</tt><big>(</big><em>func</em>, <em>mock=None</em>, <em>skipfirst=False</em><big>)</big><a class="headerlink" href="#mock.mocksignature" title="Permalink to this definition">¶</a></dt> |
michael@0 | 77 | <dd><p>Create a new function with the same signature as <cite>func</cite> that delegates |
michael@0 | 78 | to <cite>mock</cite>. If <cite>skipfirst</cite> is True the first argument is skipped, useful |
michael@0 | 79 | for methods where <cite>self</cite> needs to be omitted from the new function.</p> |
michael@0 | 80 | <p>If you don’t pass in a <cite>mock</cite> then one will be created for you.</p> |
michael@0 | 81 | <p>Functions returned by <cite>mocksignature</cite> have many of the same attributes |
michael@0 | 82 | and assert methods as a mock object.</p> |
michael@0 | 83 | <p>The mock is set as the <cite>mock</cite> attribute of the returned function for easy |
michael@0 | 84 | access.</p> |
michael@0 | 85 | <p><cite>mocksignature</cite> can also be used with classes. It copies the signature of |
michael@0 | 86 | the <cite>__init__</cite> method.</p> |
michael@0 | 87 | <p>When used with callable objects (instances) it copies the signature of the |
michael@0 | 88 | <cite>__call__</cite> method.</p> |
michael@0 | 89 | </dd></dl> |
michael@0 | 90 | |
michael@0 | 91 | <p><cite>mocksignature</cite> will work out if it is mocking the signature of a method on |
michael@0 | 92 | an instance or a method on a class and do the “right thing” with the <cite>self</cite> |
michael@0 | 93 | argument in both cases.</p> |
michael@0 | 94 | <p>Because of a limitation in the way that arguments are collected by functions |
michael@0 | 95 | created by <cite>mocksignature</cite> they are <em>always</em> passed as positional arguments |
michael@0 | 96 | (including defaults) and not keyword arguments.</p> |
michael@0 | 97 | <div class="section" id="mocksignature-api"> |
michael@0 | 98 | <h2>mocksignature api<a class="headerlink" href="#mocksignature-api" title="Permalink to this headline">¶</a></h2> |
michael@0 | 99 | <p>Although the objects returned by <cite>mocksignature</cite> api are real function objects, |
michael@0 | 100 | they have much of the same api as the <a class="reference internal" href="mock.html#mock.Mock" title="mock.Mock"><tt class="xref py py-class docutils literal"><span class="pre">Mock</span></tt></a> class. This includes the |
michael@0 | 101 | assert methods:</p> |
michael@0 | 102 | <div class="highlight-python"><div class="highlight"><pre><span class="gp">>>> </span><span class="k">def</span> <span class="nf">func</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="n">b</span><span class="p">,</span> <span class="n">c</span><span class="p">):</span> |
michael@0 | 103 | <span class="gp">... </span> <span class="k">pass</span> |
michael@0 | 104 | <span class="gp">...</span> |
michael@0 | 105 | <span class="gp">>>> </span><span class="n">func2</span> <span class="o">=</span> <span class="n">mocksignature</span><span class="p">(</span><span class="n">func</span><span class="p">)</span> |
michael@0 | 106 | <span class="gp">>>> </span><span class="n">func2</span><span class="o">.</span><span class="n">called</span> |
michael@0 | 107 | <span class="go">False</span> |
michael@0 | 108 | <span class="gp">>>> </span><span class="n">func2</span><span class="o">.</span><span class="n">return_value</span> <span class="o">=</span> <span class="mi">3</span> |
michael@0 | 109 | <span class="gp">>>> </span><span class="n">func2</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">,</span> <span class="mi">3</span><span class="p">)</span> |
michael@0 | 110 | <span class="go">3</span> |
michael@0 | 111 | <span class="gp">>>> </span><span class="n">func2</span><span class="o">.</span><span class="n">called</span> |
michael@0 | 112 | <span class="go">True</span> |
michael@0 | 113 | <span class="gp">>>> </span><span class="n">func2</span><span class="o">.</span><span class="n">assert_called_once_with</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">,</span> <span class="mi">3</span><span class="p">)</span> |
michael@0 | 114 | <span class="gp">>>> </span><span class="n">func2</span><span class="o">.</span><span class="n">assert_called_with</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">,</span> <span class="mi">4</span><span class="p">)</span> |
michael@0 | 115 | <span class="gt">Traceback (most recent call last):</span> |
michael@0 | 116 | <span class="c">...</span> |
michael@0 | 117 | <span class="gr">AssertionError: Expected call</span>: <span class="n">mock(1, 2, 4)</span> |
michael@0 | 118 | <span class="go">Actual call: mock(1, 2, 3)</span> |
michael@0 | 119 | <span class="gp">>>> </span><span class="n">func2</span><span class="o">.</span><span class="n">call_count</span> |
michael@0 | 120 | <span class="go">1</span> |
michael@0 | 121 | <span class="gp">>>> </span><span class="n">func2</span><span class="o">.</span><span class="n">side_effect</span> <span class="o">=</span> <span class="ne">IndexError</span> |
michael@0 | 122 | <span class="gp">>>> </span><span class="n">func2</span><span class="p">(</span><span class="mi">4</span><span class="p">,</span> <span class="mi">5</span><span class="p">,</span> <span class="mi">6</span><span class="p">)</span> |
michael@0 | 123 | <span class="gt">Traceback (most recent call last):</span> |
michael@0 | 124 | <span class="c">...</span> |
michael@0 | 125 | <span class="gr">IndexError</span> |
michael@0 | 126 | </pre></div> |
michael@0 | 127 | </div> |
michael@0 | 128 | <p>The mock object that is being delegated to is available as the <cite>mock</cite> attribute |
michael@0 | 129 | of the function created by <cite>mocksignature</cite>.</p> |
michael@0 | 130 | <div class="highlight-python"><div class="highlight"><pre><span class="gp">>>> </span><span class="n">func2</span><span class="o">.</span><span class="n">mock</span><span class="o">.</span><span class="n">mock_calls</span> |
michael@0 | 131 | <span class="go">[call(1, 2, 3), call(4, 5, 6)]</span> |
michael@0 | 132 | </pre></div> |
michael@0 | 133 | </div> |
michael@0 | 134 | <p>The methods and attributes available on functions returned by <cite>mocksignature</cite> |
michael@0 | 135 | are:</p> |
michael@0 | 136 | <blockquote> |
michael@0 | 137 | <div><a class="reference internal" href="mock.html#mock.Mock.assert_any_call" title="mock.Mock.assert_any_call"><tt class="xref py py-meth docutils literal"><span class="pre">assert_any_call()</span></tt></a>, <a class="reference internal" href="mock.html#mock.Mock.assert_called_once_with" title="mock.Mock.assert_called_once_with"><tt class="xref py py-meth docutils literal"><span class="pre">assert_called_once_with()</span></tt></a>, |
michael@0 | 138 | <a class="reference internal" href="mock.html#mock.Mock.assert_called_with" title="mock.Mock.assert_called_with"><tt class="xref py py-meth docutils literal"><span class="pre">assert_called_with()</span></tt></a>, <a class="reference internal" href="mock.html#mock.Mock.assert_has_calls" title="mock.Mock.assert_has_calls"><tt class="xref py py-meth docutils literal"><span class="pre">assert_has_calls()</span></tt></a>, |
michael@0 | 139 | <a class="reference internal" href="mock.html#mock.Mock.call_args" title="mock.Mock.call_args"><tt class="xref py py-attr docutils literal"><span class="pre">call_args</span></tt></a>, <a class="reference internal" href="mock.html#mock.Mock.call_args_list" title="mock.Mock.call_args_list"><tt class="xref py py-attr docutils literal"><span class="pre">call_args_list</span></tt></a>, |
michael@0 | 140 | <a class="reference internal" href="mock.html#mock.Mock.call_count" title="mock.Mock.call_count"><tt class="xref py py-attr docutils literal"><span class="pre">call_count</span></tt></a>, <a class="reference internal" href="mock.html#mock.Mock.called" title="mock.Mock.called"><tt class="xref py py-attr docutils literal"><span class="pre">called</span></tt></a>, |
michael@0 | 141 | <a class="reference internal" href="mock.html#mock.Mock.method_calls" title="mock.Mock.method_calls"><tt class="xref py py-attr docutils literal"><span class="pre">method_calls</span></tt></a>, <cite>mock</cite>, <a class="reference internal" href="mock.html#mock.Mock.mock_calls" title="mock.Mock.mock_calls"><tt class="xref py py-attr docutils literal"><span class="pre">mock_calls</span></tt></a>, |
michael@0 | 142 | <a class="reference internal" href="mock.html#mock.Mock.reset_mock" title="mock.Mock.reset_mock"><tt class="xref py py-meth docutils literal"><span class="pre">reset_mock()</span></tt></a>, <a class="reference internal" href="mock.html#mock.Mock.return_value" title="mock.Mock.return_value"><tt class="xref py py-attr docutils literal"><span class="pre">return_value</span></tt></a>, and |
michael@0 | 143 | <a class="reference internal" href="mock.html#mock.Mock.side_effect" title="mock.Mock.side_effect"><tt class="xref py py-attr docutils literal"><span class="pre">side_effect</span></tt></a>.</div></blockquote> |
michael@0 | 144 | </div> |
michael@0 | 145 | <div class="section" id="example-use"> |
michael@0 | 146 | <h2>Example use<a class="headerlink" href="#example-use" title="Permalink to this headline">¶</a></h2> |
michael@0 | 147 | <div class="section" id="basic-use"> |
michael@0 | 148 | <h3>Basic use<a class="headerlink" href="#basic-use" title="Permalink to this headline">¶</a></h3> |
michael@0 | 149 | <div class="highlight-python"><div class="highlight"><pre><span class="gp">>>> </span><span class="k">def</span> <span class="nf">function</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="n">b</span><span class="p">,</span> <span class="n">c</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span> |
michael@0 | 150 | <span class="gp">... </span> <span class="k">pass</span> |
michael@0 | 151 | <span class="gp">...</span> |
michael@0 | 152 | <span class="gp">>>> </span><span class="n">mock</span> <span class="o">=</span> <span class="n">Mock</span><span class="p">()</span> |
michael@0 | 153 | <span class="gp">>>> </span><span class="n">function</span> <span class="o">=</span> <span class="n">mocksignature</span><span class="p">(</span><span class="n">function</span><span class="p">,</span> <span class="n">mock</span><span class="p">)</span> |
michael@0 | 154 | <span class="gp">>>> </span><span class="n">function</span><span class="p">()</span> |
michael@0 | 155 | <span class="gt">Traceback (most recent call last):</span> |
michael@0 | 156 | <span class="c">...</span> |
michael@0 | 157 | <span class="gr">TypeError</span>: <span class="n"><lambda>() takes at least 2 arguments (0 given)</span> |
michael@0 | 158 | <span class="gp">>>> </span><span class="n">function</span><span class="o">.</span><span class="n">return_value</span> <span class="o">=</span> <span class="s">'some value'</span> |
michael@0 | 159 | <span class="gp">>>> </span><span class="n">function</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">,</span> <span class="s">'foo'</span><span class="p">)</span> |
michael@0 | 160 | <span class="go">'some value'</span> |
michael@0 | 161 | <span class="gp">>>> </span><span class="n">function</span><span class="o">.</span><span class="n">assert_called_with</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">,</span> <span class="s">'foo'</span><span class="p">)</span> |
michael@0 | 162 | </pre></div> |
michael@0 | 163 | </div> |
michael@0 | 164 | </div> |
michael@0 | 165 | <div class="section" id="keyword-arguments"> |
michael@0 | 166 | <h3>Keyword arguments<a class="headerlink" href="#keyword-arguments" title="Permalink to this headline">¶</a></h3> |
michael@0 | 167 | <p>Note that arguments to functions created by <cite>mocksignature</cite> are always passed |
michael@0 | 168 | in to the underlying mock by position even when called with keywords:</p> |
michael@0 | 169 | <div class="highlight-python"><div class="highlight"><pre><span class="gp">>>> </span><span class="k">def</span> <span class="nf">function</span><span class="p">(</span><span class="n">a</span><span class="p">,</span> <span class="n">b</span><span class="p">,</span> <span class="n">c</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span> |
michael@0 | 170 | <span class="gp">... </span> <span class="k">pass</span> |
michael@0 | 171 | <span class="gp">...</span> |
michael@0 | 172 | <span class="gp">>>> </span><span class="n">function</span> <span class="o">=</span> <span class="n">mocksignature</span><span class="p">(</span><span class="n">function</span><span class="p">)</span> |
michael@0 | 173 | <span class="gp">>>> </span><span class="n">function</span><span class="o">.</span><span class="n">return_value</span> <span class="o">=</span> <span class="bp">None</span> |
michael@0 | 174 | <span class="gp">>>> </span><span class="n">function</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">)</span> |
michael@0 | 175 | <span class="gp">>>> </span><span class="n">function</span><span class="o">.</span><span class="n">assert_called_with</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">,</span> <span class="bp">None</span><span class="p">)</span> |
michael@0 | 176 | </pre></div> |
michael@0 | 177 | </div> |
michael@0 | 178 | </div> |
michael@0 | 179 | <div class="section" id="mocking-methods-and-self"> |
michael@0 | 180 | <h3>Mocking methods and self<a class="headerlink" href="#mocking-methods-and-self" title="Permalink to this headline">¶</a></h3> |
michael@0 | 181 | <p>When you use <cite>mocksignature</cite> to replace a method on a class then <cite>self</cite> |
michael@0 | 182 | will be included in the method signature - and you will need to include |
michael@0 | 183 | the instance when you do your asserts.</p> |
michael@0 | 184 | <p>As a curious factor of the way Python (2) wraps methods fetched from a class, |
michael@0 | 185 | we can <em>get</em> the <cite>return_value</cite> from a function set on a class, but we can’t |
michael@0 | 186 | set it. We have to do this through the exposed <cite>mock</cite> attribute instead:</p> |
michael@0 | 187 | <div class="highlight-python"><div class="highlight"><pre><span class="gp">>>> </span><span class="k">class</span> <span class="nc">SomeClass</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span> |
michael@0 | 188 | <span class="gp">... </span> <span class="k">def</span> <span class="nf">method</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">a</span><span class="p">,</span> <span class="n">b</span><span class="p">,</span> <span class="n">c</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span> |
michael@0 | 189 | <span class="gp">... </span> <span class="k">pass</span> |
michael@0 | 190 | <span class="gp">...</span> |
michael@0 | 191 | <span class="gp">>>> </span><span class="n">SomeClass</span><span class="o">.</span><span class="n">method</span> <span class="o">=</span> <span class="n">mocksignature</span><span class="p">(</span><span class="n">SomeClass</span><span class="o">.</span><span class="n">method</span><span class="p">)</span> |
michael@0 | 192 | <span class="gp">>>> </span><span class="n">SomeClass</span><span class="o">.</span><span class="n">method</span><span class="o">.</span><span class="n">mock</span><span class="o">.</span><span class="n">return_value</span> <span class="o">=</span> <span class="bp">None</span> |
michael@0 | 193 | <span class="gp">>>> </span><span class="n">instance</span> <span class="o">=</span> <span class="n">SomeClass</span><span class="p">()</span> |
michael@0 | 194 | <span class="gp">>>> </span><span class="n">instance</span><span class="o">.</span><span class="n">method</span><span class="p">()</span> |
michael@0 | 195 | <span class="gt">Traceback (most recent call last):</span> |
michael@0 | 196 | <span class="c">...</span> |
michael@0 | 197 | <span class="gr">TypeError</span>: <span class="n"><lambda>() takes at least 4 arguments (1 given)</span> |
michael@0 | 198 | <span class="gp">>>> </span><span class="n">instance</span><span class="o">.</span><span class="n">method</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">,</span> <span class="mi">3</span><span class="p">)</span> |
michael@0 | 199 | <span class="gp">>>> </span><span class="n">instance</span><span class="o">.</span><span class="n">method</span><span class="o">.</span><span class="n">assert_called_with</span><span class="p">(</span><span class="n">instance</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">,</span> <span class="mi">3</span><span class="p">)</span> |
michael@0 | 200 | </pre></div> |
michael@0 | 201 | </div> |
michael@0 | 202 | <p>When you use <cite>mocksignature</cite> on instance methods <cite>self</cite> isn’t included (and we |
michael@0 | 203 | can set the <cite>return_value</cite> etc directly):</p> |
michael@0 | 204 | <div class="highlight-python"><div class="highlight"><pre><span class="gp">>>> </span><span class="k">class</span> <span class="nc">SomeClass</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span> |
michael@0 | 205 | <span class="gp">... </span> <span class="k">def</span> <span class="nf">method</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">a</span><span class="p">,</span> <span class="n">b</span><span class="p">,</span> <span class="n">c</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span> |
michael@0 | 206 | <span class="gp">... </span> <span class="k">pass</span> |
michael@0 | 207 | <span class="gp">...</span> |
michael@0 | 208 | <span class="gp">>>> </span><span class="n">instance</span> <span class="o">=</span> <span class="n">SomeClass</span><span class="p">()</span> |
michael@0 | 209 | <span class="gp">>>> </span><span class="n">instance</span><span class="o">.</span><span class="n">method</span> <span class="o">=</span> <span class="n">mocksignature</span><span class="p">(</span><span class="n">instance</span><span class="o">.</span><span class="n">method</span><span class="p">)</span> |
michael@0 | 210 | <span class="gp">>>> </span><span class="n">instance</span><span class="o">.</span><span class="n">method</span><span class="o">.</span><span class="n">return_value</span> <span class="o">=</span> <span class="bp">None</span> |
michael@0 | 211 | <span class="gp">>>> </span><span class="n">instance</span><span class="o">.</span><span class="n">method</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">,</span> <span class="mi">3</span><span class="p">)</span> |
michael@0 | 212 | <span class="gp">>>> </span><span class="n">instance</span><span class="o">.</span><span class="n">method</span><span class="o">.</span><span class="n">assert_called_with</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">,</span> <span class="mi">3</span><span class="p">)</span> |
michael@0 | 213 | </pre></div> |
michael@0 | 214 | </div> |
michael@0 | 215 | </div> |
michael@0 | 216 | <div class="section" id="mocksignature-with-classes"> |
michael@0 | 217 | <h3>mocksignature with classes<a class="headerlink" href="#mocksignature-with-classes" title="Permalink to this headline">¶</a></h3> |
michael@0 | 218 | <p>When used with a class <cite>mocksignature</cite> copies the signature of the <cite>__init__</cite> |
michael@0 | 219 | method.</p> |
michael@0 | 220 | <div class="highlight-python"><div class="highlight"><pre><span class="gp">>>> </span><span class="k">class</span> <span class="nc">Something</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span> |
michael@0 | 221 | <span class="gp">... </span> <span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">foo</span><span class="p">,</span> <span class="n">bar</span><span class="p">):</span> |
michael@0 | 222 | <span class="gp">... </span> <span class="k">pass</span> |
michael@0 | 223 | <span class="gp">...</span> |
michael@0 | 224 | <span class="gp">>>> </span><span class="n">MockSomething</span> <span class="o">=</span> <span class="n">mocksignature</span><span class="p">(</span><span class="n">Something</span><span class="p">)</span> |
michael@0 | 225 | <span class="gp">>>> </span><span class="n">instance</span> <span class="o">=</span> <span class="n">MockSomething</span><span class="p">(</span><span class="mi">10</span><span class="p">,</span> <span class="mi">9</span><span class="p">)</span> |
michael@0 | 226 | <span class="gp">>>> </span><span class="k">assert</span> <span class="n">instance</span> <span class="ow">is</span> <span class="n">MockSomething</span><span class="o">.</span><span class="n">return_value</span> |
michael@0 | 227 | <span class="gp">>>> </span><span class="n">MockSomething</span><span class="o">.</span><span class="n">assert_called_with</span><span class="p">(</span><span class="mi">10</span><span class="p">,</span> <span class="mi">9</span><span class="p">)</span> |
michael@0 | 228 | <span class="gp">>>> </span><span class="n">MockSomething</span><span class="p">()</span> |
michael@0 | 229 | <span class="gt">Traceback (most recent call last):</span> |
michael@0 | 230 | <span class="c">...</span> |
michael@0 | 231 | <span class="gr">TypeError</span>: <span class="n"><lambda>() takes at least 2 arguments (0 given)</span> |
michael@0 | 232 | </pre></div> |
michael@0 | 233 | </div> |
michael@0 | 234 | <p>Because the object returned by <cite>mocksignature</cite> is a function rather than a |
michael@0 | 235 | <cite>Mock</cite> you lose the other capabilities of <cite>Mock</cite>, like dynamic attribute |
michael@0 | 236 | creation.</p> |
michael@0 | 237 | </div> |
michael@0 | 238 | <div class="section" id="mocksignature-with-callable-objects"> |
michael@0 | 239 | <h3>mocksignature with callable objects<a class="headerlink" href="#mocksignature-with-callable-objects" title="Permalink to this headline">¶</a></h3> |
michael@0 | 240 | <p>When used with a callable object <cite>mocksignature</cite> copies the signature of the |
michael@0 | 241 | <cite>__call__</cite> method.</p> |
michael@0 | 242 | <div class="highlight-python"><div class="highlight"><pre><span class="gp">>>> </span><span class="k">class</span> <span class="nc">Something</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span> |
michael@0 | 243 | <span class="gp">... </span> <span class="k">def</span> <span class="nf">__call__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">spam</span><span class="p">,</span> <span class="n">eggs</span><span class="p">):</span> |
michael@0 | 244 | <span class="gp">... </span> <span class="k">pass</span> |
michael@0 | 245 | <span class="gp">...</span> |
michael@0 | 246 | <span class="gp">>>> </span><span class="n">something</span> <span class="o">=</span> <span class="n">Something</span><span class="p">()</span> |
michael@0 | 247 | <span class="gp">>>> </span><span class="n">mock_something</span> <span class="o">=</span> <span class="n">mocksignature</span><span class="p">(</span><span class="n">something</span><span class="p">)</span> |
michael@0 | 248 | <span class="gp">>>> </span><span class="n">result</span> <span class="o">=</span> <span class="n">mock_something</span><span class="p">(</span><span class="mi">10</span><span class="p">,</span> <span class="mi">9</span><span class="p">)</span> |
michael@0 | 249 | <span class="gp">>>> </span><span class="n">mock_something</span><span class="o">.</span><span class="n">assert_called_with</span><span class="p">(</span><span class="mi">10</span><span class="p">,</span> <span class="mi">9</span><span class="p">)</span> |
michael@0 | 250 | <span class="gp">>>> </span><span class="n">mock_something</span><span class="p">()</span> |
michael@0 | 251 | <span class="gt">Traceback (most recent call last):</span> |
michael@0 | 252 | <span class="c">...</span> |
michael@0 | 253 | <span class="gr">TypeError</span>: <span class="n"><lambda>() takes at least 2 arguments (0 given)</span> |
michael@0 | 254 | </pre></div> |
michael@0 | 255 | </div> |
michael@0 | 256 | </div> |
michael@0 | 257 | </div> |
michael@0 | 258 | <div class="section" id="mocksignature-argument-to-patch"> |
michael@0 | 259 | <h2>mocksignature argument to patch<a class="headerlink" href="#mocksignature-argument-to-patch" title="Permalink to this headline">¶</a></h2> |
michael@0 | 260 | <p><cite>mocksignature</cite> is available as a keyword argument to <a class="reference internal" href="patch.html#mock.patch" title="mock.patch"><tt class="xref py py-func docutils literal"><span class="pre">patch()</span></tt></a> or |
michael@0 | 261 | <a class="reference internal" href="patch.html#mock.patch.object" title="mock.patch.object"><tt class="xref py py-func docutils literal"><span class="pre">patch.object()</span></tt></a>. It can be used with functions / methods / classes and |
michael@0 | 262 | callable objects.</p> |
michael@0 | 263 | <div class="highlight-python"><div class="highlight"><pre><span class="gp">>>> </span><span class="k">class</span> <span class="nc">SomeClass</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span> |
michael@0 | 264 | <span class="gp">... </span> <span class="k">def</span> <span class="nf">method</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">a</span><span class="p">,</span> <span class="n">b</span><span class="p">,</span> <span class="n">c</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span> |
michael@0 | 265 | <span class="gp">... </span> <span class="k">pass</span> |
michael@0 | 266 | <span class="gp">...</span> |
michael@0 | 267 | <span class="gp">>>> </span><span class="nd">@patch.object</span><span class="p">(</span><span class="n">SomeClass</span><span class="p">,</span> <span class="s">'method'</span><span class="p">,</span> <span class="n">mocksignature</span><span class="o">=</span><span class="bp">True</span><span class="p">)</span> |
michael@0 | 268 | <span class="gp">... </span><span class="k">def</span> <span class="nf">test</span><span class="p">(</span><span class="n">mock_method</span><span class="p">):</span> |
michael@0 | 269 | <span class="gp">... </span> <span class="n">instance</span> <span class="o">=</span> <span class="n">SomeClass</span><span class="p">()</span> |
michael@0 | 270 | <span class="gp">... </span> <span class="n">mock_method</span><span class="o">.</span><span class="n">return_value</span> <span class="o">=</span> <span class="bp">None</span> |
michael@0 | 271 | <span class="gp">... </span> <span class="n">instance</span><span class="o">.</span><span class="n">method</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">)</span> |
michael@0 | 272 | <span class="gp">... </span> <span class="n">mock_method</span><span class="o">.</span><span class="n">assert_called_with</span><span class="p">(</span><span class="n">instance</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="mi">2</span><span class="p">,</span> <span class="bp">None</span><span class="p">)</span> |
michael@0 | 273 | <span class="gp">...</span> |
michael@0 | 274 | <span class="gp">>>> </span><span class="n">test</span><span class="p">()</span> |
michael@0 | 275 | </pre></div> |
michael@0 | 276 | </div> |
michael@0 | 277 | </div> |
michael@0 | 278 | </div> |
michael@0 | 279 | |
michael@0 | 280 | |
michael@0 | 281 | </div> |
michael@0 | 282 | </div> |
michael@0 | 283 | </div> |
michael@0 | 284 | <div class="sphinxsidebar"> |
michael@0 | 285 | <div class="sphinxsidebarwrapper"> |
michael@0 | 286 | <h3><a href="index.html">Table Of Contents</a></h3> |
michael@0 | 287 | <ul> |
michael@0 | 288 | <li><a class="reference internal" href="#">mocksignature</a><ul> |
michael@0 | 289 | <li><a class="reference internal" href="#mocksignature-api">mocksignature api</a></li> |
michael@0 | 290 | <li><a class="reference internal" href="#example-use">Example use</a><ul> |
michael@0 | 291 | <li><a class="reference internal" href="#basic-use">Basic use</a></li> |
michael@0 | 292 | <li><a class="reference internal" href="#keyword-arguments">Keyword arguments</a></li> |
michael@0 | 293 | <li><a class="reference internal" href="#mocking-methods-and-self">Mocking methods and self</a></li> |
michael@0 | 294 | <li><a class="reference internal" href="#mocksignature-with-classes">mocksignature with classes</a></li> |
michael@0 | 295 | <li><a class="reference internal" href="#mocksignature-with-callable-objects">mocksignature with callable objects</a></li> |
michael@0 | 296 | </ul> |
michael@0 | 297 | </li> |
michael@0 | 298 | <li><a class="reference internal" href="#mocksignature-argument-to-patch">mocksignature argument to patch</a></li> |
michael@0 | 299 | </ul> |
michael@0 | 300 | </li> |
michael@0 | 301 | </ul> |
michael@0 | 302 | |
michael@0 | 303 | <h4>Previous topic</h4> |
michael@0 | 304 | <p class="topless"><a href="magicmock.html" |
michael@0 | 305 | title="previous chapter">Mocking Magic Methods</a></p> |
michael@0 | 306 | <h4>Next topic</h4> |
michael@0 | 307 | <p class="topless"><a href="getting-started.html" |
michael@0 | 308 | title="next chapter">Getting Started with Mock</a></p> |
michael@0 | 309 | <h3>This Page</h3> |
michael@0 | 310 | <ul class="this-page-menu"> |
michael@0 | 311 | <li><a href="_sources/mocksignature.txt" |
michael@0 | 312 | rel="nofollow">Show Source</a></li> |
michael@0 | 313 | </ul> |
michael@0 | 314 | <div id="searchbox" style="display: none"> |
michael@0 | 315 | <h3>Quick search</h3> |
michael@0 | 316 | <form class="search" action="search.html" method="get"> |
michael@0 | 317 | <input type="text" name="q" /> |
michael@0 | 318 | <input type="submit" value="Go" /> |
michael@0 | 319 | <input type="hidden" name="check_keywords" value="yes" /> |
michael@0 | 320 | <input type="hidden" name="area" value="default" /> |
michael@0 | 321 | </form> |
michael@0 | 322 | <p class="searchtip" style="font-size: 90%"> |
michael@0 | 323 | Enter search terms or a module, class or function name. |
michael@0 | 324 | </p> |
michael@0 | 325 | </div> |
michael@0 | 326 | <script type="text/javascript">$('#searchbox').show(0);</script> |
michael@0 | 327 | </div> |
michael@0 | 328 | </div> |
michael@0 | 329 | <div class="clearer"></div> |
michael@0 | 330 | </div> |
michael@0 | 331 | <div class="related"> |
michael@0 | 332 | <h3>Navigation</h3> |
michael@0 | 333 | <ul> |
michael@0 | 334 | <li class="right" style="margin-right: 10px"> |
michael@0 | 335 | <a href="genindex.html" title="General Index" |
michael@0 | 336 | >index</a></li> |
michael@0 | 337 | <li class="right" > |
michael@0 | 338 | <a href="getting-started.html" title="Getting Started with Mock" |
michael@0 | 339 | >next</a> |</li> |
michael@0 | 340 | <li class="right" > |
michael@0 | 341 | <a href="magicmock.html" title="Mocking Magic Methods" |
michael@0 | 342 | >previous</a> |</li> |
michael@0 | 343 | <li><a href="index.html">Mock 0.8.1alpha1 documentation</a> »</li> |
michael@0 | 344 | </ul> |
michael@0 | 345 | </div> |
michael@0 | 346 | <div class="footer"> |
michael@0 | 347 | © Copyright 2007-2012, Michael Foord & the mock team. |
michael@0 | 348 | Last updated on Feb 16, 2012. |
michael@0 | 349 | Created using <a href="http://sphinx.pocoo.org/">Sphinx</a> 1.1.2. |
michael@0 | 350 | </div> |
michael@0 | 351 | </body> |
michael@0 | 352 | </html> |