<!doctype html> <html xmlns="http://www.w3.org/1999/xhtml" lang="en"> <head> <meta http-equiv="X-UA-Compatible" content="IE=Edge" /> <meta http-equiv="Content-Type" content="text/html; charset=utf-8" /> <title>sqlobject.tests.test_SQLMultipleJoin — SQLObject 3.7.0 documentation</title> <link rel="stylesheet" href="../../../_static/bizstyle.css" type="text/css" /> <link rel="stylesheet" href="../../../_static/pygments.css" type="text/css" /> <script type="text/javascript" src="../../../_static/documentation_options.js"></script> <script type="text/javascript" src="../../../_static/jquery.js"></script> <script type="text/javascript" src="../../../_static/underscore.js"></script> <script type="text/javascript" src="../../../_static/doctools.js"></script> <script type="text/javascript" src="../../../_static/bizstyle.js"></script> <link rel="index" title="Index" href="../../../genindex.html" /> <link rel="search" title="Search" href="../../../search.html" /> <meta name="viewport" content="width=device-width,initial-scale=1.0"> <!--[if lt IE 9]> <script type="text/javascript" src="_static/css3-mediaqueries.js"></script> <![endif]--> </head><body> <div class="related" role="navigation" aria-label="related navigation"> <h3>Navigation</h3> <ul> <li class="right" style="margin-right: 10px"> <a href="../../../genindex.html" title="General Index" accesskey="I">index</a></li> <li class="right" > <a href="../../../py-modindex.html" title="Python Module Index" >modules</a> |</li> <li class="nav-item nav-item-0"><a href="../../../index.html">SQLObject 3.7.0 documentation</a> »</li> <li class="nav-item nav-item-1"><a href="../../index.html" accesskey="U">Module code</a> »</li> </ul> </div> <div class="sphinxsidebar" role="navigation" aria-label="main navigation"> <div class="sphinxsidebarwrapper"> <div id="searchbox" style="display: none" role="search"> <h3>Quick search</h3> <div class="searchformwrapper"> <form class="search" action="../../../search.html" method="get"> <input type="text" name="q" /> <input type="submit" value="Go" /> <input type="hidden" name="check_keywords" value="yes" /> <input type="hidden" name="area" value="default" /> </form> </div> </div> <script type="text/javascript">$('#searchbox').show(0);</script> </div> </div> <div class="document"> <div class="documentwrapper"> <div class="bodywrapper"> <div class="body" role="main"> <h1>Source code for sqlobject.tests.test_SQLMultipleJoin</h1><div class="highlight"><pre> <span></span><span class="kn">import</span> <span class="nn">pytest</span> <span class="kn">from</span> <span class="nn">sqlobject</span> <span class="k">import</span> <span class="n">ForeignKey</span><span class="p">,</span> <span class="n">IntCol</span><span class="p">,</span> <span class="n">MultipleJoin</span><span class="p">,</span> <span class="n">SQLMultipleJoin</span><span class="p">,</span> \ <span class="n">SQLObject</span><span class="p">,</span> <span class="n">StringCol</span> <span class="kn">from</span> <span class="nn">sqlobject.tests.dbtest</span> <span class="k">import</span> <span class="n">setupClass</span><span class="p">,</span> <span class="n">supports</span> <div class="viewcode-block" id="Race"><a class="viewcode-back" href="../../../api/sqlobject.tests.test_SQLMultipleJoin.html#sqlobject.tests.test_SQLMultipleJoin.Race">[docs]</a><span class="k">class</span> <span class="nc">Race</span><span class="p">(</span><span class="n">SQLObject</span><span class="p">):</span> <span class="n">name</span> <span class="o">=</span> <span class="n">StringCol</span><span class="p">()</span> <span class="n">fightersAsList</span> <span class="o">=</span> <span class="n">MultipleJoin</span><span class="p">(</span><span class="s1">'RFighter'</span><span class="p">,</span> <span class="n">joinColumn</span><span class="o">=</span><span class="s2">"rf_id"</span><span class="p">)</span> <span class="n">fightersAsSResult</span> <span class="o">=</span> <span class="n">SQLMultipleJoin</span><span class="p">(</span><span class="s1">'RFighter'</span><span class="p">,</span> <span class="n">joinColumn</span><span class="o">=</span><span class="s2">"rf_id"</span><span class="p">)</span> </div> <div class="viewcode-block" id="RFighter"><a class="viewcode-back" href="../../../api/sqlobject.tests.test_SQLMultipleJoin.html#sqlobject.tests.test_SQLMultipleJoin.RFighter">[docs]</a><span class="k">class</span> <span class="nc">RFighter</span><span class="p">(</span><span class="n">SQLObject</span><span class="p">):</span> <span class="n">name</span> <span class="o">=</span> <span class="n">StringCol</span><span class="p">()</span> <span class="n">race</span> <span class="o">=</span> <span class="n">ForeignKey</span><span class="p">(</span><span class="s1">'Race'</span><span class="p">,</span> <span class="n">dbName</span><span class="o">=</span><span class="s2">"rf_id"</span><span class="p">)</span> <span class="n">power</span> <span class="o">=</span> <span class="n">IntCol</span><span class="p">()</span> </div> <div class="viewcode-block" id="createAllTables"><a class="viewcode-back" href="../../../api/sqlobject.tests.test_SQLMultipleJoin.html#sqlobject.tests.test_SQLMultipleJoin.createAllTables">[docs]</a><span class="k">def</span> <span class="nf">createAllTables</span><span class="p">():</span> <span class="n">setupClass</span><span class="p">([</span><span class="n">Race</span><span class="p">,</span> <span class="n">RFighter</span><span class="p">])</span> </div> <div class="viewcode-block" id="test_1"><a class="viewcode-back" href="../../../api/sqlobject.tests.test_SQLMultipleJoin.html#sqlobject.tests.test_SQLMultipleJoin.test_1">[docs]</a><span class="k">def</span> <span class="nf">test_1</span><span class="p">():</span> <span class="n">createAllTables</span><span class="p">()</span> <span class="c1"># create some races</span> <span class="n">human</span> <span class="o">=</span> <span class="n">Race</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="s1">'human'</span><span class="p">)</span> <span class="n">saiyajin</span> <span class="o">=</span> <span class="n">Race</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="s1">'saiyajin'</span><span class="p">)</span> <span class="n">hibrid</span> <span class="o">=</span> <span class="n">Race</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="s1">'hibrid (human with sayajin)'</span><span class="p">)</span> <span class="n">namek</span> <span class="o">=</span> <span class="n">Race</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="s1">'namekuseijin'</span><span class="p">)</span> <span class="c1"># create some fighters</span> <span class="n">RFighter</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="s1">'Gokou (Kakaruto)'</span><span class="p">,</span> <span class="n">race</span><span class="o">=</span><span class="n">saiyajin</span><span class="p">,</span> <span class="n">power</span><span class="o">=</span><span class="mi">10</span><span class="p">)</span> <span class="n">RFighter</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="s1">'Vegeta'</span><span class="p">,</span> <span class="n">race</span><span class="o">=</span><span class="n">saiyajin</span><span class="p">,</span> <span class="n">power</span><span class="o">=</span><span class="mi">9</span><span class="p">)</span> <span class="n">RFighter</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="s1">'Krilim'</span><span class="p">,</span> <span class="n">race</span><span class="o">=</span><span class="n">human</span><span class="p">,</span> <span class="n">power</span><span class="o">=</span><span class="mi">3</span><span class="p">)</span> <span class="n">RFighter</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="s1">'Yancha'</span><span class="p">,</span> <span class="n">race</span><span class="o">=</span><span class="n">human</span><span class="p">,</span> <span class="n">power</span><span class="o">=</span><span class="mi">2</span><span class="p">)</span> <span class="n">RFighter</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="s1">'Jackie Chan'</span><span class="p">,</span> <span class="n">race</span><span class="o">=</span><span class="n">human</span><span class="p">,</span> <span class="n">power</span><span class="o">=</span><span class="mi">2</span><span class="p">)</span> <span class="n">RFighter</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="s1">'Gohan'</span><span class="p">,</span> <span class="n">race</span><span class="o">=</span><span class="n">hibrid</span><span class="p">,</span> <span class="n">power</span><span class="o">=</span><span class="mi">8</span><span class="p">)</span> <span class="n">RFighter</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="s1">'Goten'</span><span class="p">,</span> <span class="n">race</span><span class="o">=</span><span class="n">hibrid</span><span class="p">,</span> <span class="n">power</span><span class="o">=</span><span class="mi">7</span><span class="p">)</span> <span class="n">trunks</span> <span class="o">=</span> <span class="n">RFighter</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="s1">'Trunks'</span><span class="p">,</span> <span class="n">race</span><span class="o">=</span><span class="n">hibrid</span><span class="p">,</span> <span class="n">power</span><span class="o">=</span><span class="mi">8</span><span class="p">)</span> <span class="n">picollo</span> <span class="o">=</span> <span class="n">RFighter</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="s1">'Picollo'</span><span class="p">,</span> <span class="n">race</span><span class="o">=</span><span class="n">namek</span><span class="p">,</span> <span class="n">power</span><span class="o">=</span><span class="mi">6</span><span class="p">)</span> <span class="n">RFighter</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="s1">'Neil'</span><span class="p">,</span> <span class="n">race</span><span class="o">=</span><span class="n">namek</span><span class="p">,</span> <span class="n">power</span><span class="o">=</span><span class="mi">5</span><span class="p">)</span> <span class="c1"># testing the SQLMultipleJoin stuff</span> <span class="k">for</span> <span class="n">i</span><span class="p">,</span> <span class="n">j</span> <span class="ow">in</span> <span class="nb">zip</span><span class="p">(</span><span class="n">human</span><span class="o">.</span><span class="n">fightersAsList</span><span class="p">,</span> <span class="n">human</span><span class="o">.</span><span class="n">fightersAsSResult</span><span class="p">):</span> <span class="k">assert</span> <span class="n">i</span> <span class="ow">is</span> <span class="n">j</span> <span class="c1"># the 2 ways should give the same result</span> <span class="k">assert</span> <span class="n">namek</span><span class="o">.</span><span class="n">fightersAsSResult</span><span class="o">.</span><span class="n">count</span><span class="p">()</span> <span class="o">==</span> <span class="nb">len</span><span class="p">(</span><span class="n">namek</span><span class="o">.</span><span class="n">fightersAsList</span><span class="p">)</span> <span class="k">assert</span> <span class="n">saiyajin</span><span class="o">.</span><span class="n">fightersAsSResult</span><span class="o">.</span><span class="n">max</span><span class="p">(</span><span class="s1">'power'</span><span class="p">)</span> <span class="o">==</span> <span class="mi">10</span> <span class="k">assert</span> <span class="n">trunks</span> <span class="ow">in</span> <span class="n">hibrid</span><span class="o">.</span><span class="n">fightersAsSResult</span> <span class="k">assert</span> <span class="n">picollo</span> <span class="ow">not</span> <span class="ow">in</span> <span class="n">hibrid</span><span class="o">.</span><span class="n">fightersAsSResult</span> <span class="k">assert</span> <span class="nb">str</span><span class="p">(</span><span class="n">hibrid</span><span class="o">.</span><span class="n">fightersAsSResult</span><span class="o">.</span><span class="n">sum</span><span class="p">(</span><span class="s1">'power'</span><span class="p">))</span> <span class="o">==</span> <span class="s1">'23'</span> </div> <div class="viewcode-block" id="test_multiple_join_transaction"><a class="viewcode-back" href="../../../api/sqlobject.tests.test_SQLMultipleJoin.html#sqlobject.tests.test_SQLMultipleJoin.test_multiple_join_transaction">[docs]</a><span class="k">def</span> <span class="nf">test_multiple_join_transaction</span><span class="p">():</span> <span class="k">if</span> <span class="ow">not</span> <span class="n">supports</span><span class="p">(</span><span class="s1">'transactions'</span><span class="p">):</span> <span class="n">pytest</span><span class="o">.</span><span class="n">skip</span><span class="p">(</span><span class="s2">"Transactions aren't supported"</span><span class="p">)</span> <span class="n">createAllTables</span><span class="p">()</span> <span class="n">trans</span> <span class="o">=</span> <span class="n">Race</span><span class="o">.</span><span class="n">_connection</span><span class="o">.</span><span class="n">transaction</span><span class="p">()</span> <span class="k">try</span><span class="p">:</span> <span class="n">namek</span> <span class="o">=</span> <span class="n">Race</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="s1">'namekuseijin'</span><span class="p">,</span> <span class="n">connection</span><span class="o">=</span><span class="n">trans</span><span class="p">)</span> <span class="n">RFighter</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="s1">'Gokou (Kakaruto)'</span><span class="p">,</span> <span class="n">race</span><span class="o">=</span><span class="n">namek</span><span class="p">,</span> <span class="n">power</span><span class="o">=</span><span class="mi">10</span><span class="p">,</span> <span class="n">connection</span><span class="o">=</span><span class="n">trans</span><span class="p">)</span> <span class="k">assert</span> <span class="n">namek</span><span class="o">.</span><span class="n">fightersAsSResult</span><span class="o">.</span><span class="n">count</span><span class="p">()</span> <span class="o">==</span> <span class="mi">1</span> <span class="k">assert</span> <span class="n">namek</span><span class="o">.</span><span class="n">fightersAsSResult</span><span class="p">[</span><span class="mi">0</span><span class="p">]</span><span class="o">.</span><span class="n">_connection</span> <span class="o">==</span> <span class="n">trans</span> <span class="k">finally</span><span class="p">:</span> <span class="n">trans</span><span class="o">.</span><span class="n">commit</span><span class="p">(</span><span class="kc">True</span><span class="p">)</span> <span class="n">Race</span><span class="o">.</span><span class="n">_connection</span><span class="o">.</span><span class="n">autoCommit</span> <span class="o">=</span> <span class="kc">True</span> </pre></div></div> </div> </div> </div> <div class="clearer"></div> </div> <div class="related" role="navigation" aria-label="related navigation"> <h3>Navigation</h3> <ul> <li class="right" style="margin-right: 10px"> <a href="../../../genindex.html" title="General Index" >index</a></li> <li class="right" > <a href="../../../py-modindex.html" title="Python Module Index" >modules</a> |</li> <li class="nav-item nav-item-0"><a href="../../../index.html">SQLObject 3.7.0 documentation</a> »</li> <li class="nav-item nav-item-1"><a href="../../index.html" >Module code</a> »</li> </ul> </div> <div class="footer" role="contentinfo"> © Copyright 2004-2018, Ian Bicking and contributors. Created using <a href="http://sphinx-doc.org/">Sphinx</a> 1.7.4. </div> </body> </html>