<!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_sqlbuilder — 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_sqlbuilder</h1><div class="highlight"><pre> <span></span><span class="kn">from</span> <span class="nn">sqlobject</span> <span class="k">import</span> <span class="n">IntCol</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.compat</span> <span class="k">import</span> <span class="n">PY2</span> <span class="kn">from</span> <span class="nn">sqlobject.sqlbuilder</span> <span class="k">import</span> <span class="n">AND</span><span class="p">,</span> <span class="n">CONCAT</span><span class="p">,</span> <span class="n">Delete</span><span class="p">,</span> <span class="n">Insert</span><span class="p">,</span> <span class="n">SQLOp</span><span class="p">,</span> <span class="n">Select</span><span class="p">,</span> \ <span class="n">Union</span><span class="p">,</span> <span class="n">Update</span><span class="p">,</span> <span class="n">const</span><span class="p">,</span> <span class="n">func</span><span class="p">,</span> <span class="n">sqlrepr</span> <span class="kn">from</span> <span class="nn">sqlobject.tests.dbtest</span> <span class="k">import</span> <span class="n">getConnection</span><span class="p">,</span> <span class="n">raises</span><span class="p">,</span> <span class="n">setupClass</span> <div class="viewcode-block" id="SOTestSQLBuilder"><a class="viewcode-back" href="../../../api/sqlobject.tests.test_sqlbuilder.html#sqlobject.tests.test_sqlbuilder.SOTestSQLBuilder">[docs]</a><span class="k">class</span> <span class="nc">SOTestSQLBuilder</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">so_value</span> <span class="o">=</span> <span class="n">IntCol</span><span class="p">()</span> </div> <div class="viewcode-block" id="test_Select"><a class="viewcode-back" href="../../../api/sqlobject.tests.test_sqlbuilder.html#sqlobject.tests.test_sqlbuilder.test_Select">[docs]</a><span class="k">def</span> <span class="nf">test_Select</span><span class="p">():</span> <span class="n">setupClass</span><span class="p">(</span><span class="n">SOTestSQLBuilder</span><span class="p">)</span> <span class="n">select1</span> <span class="o">=</span> <span class="n">Select</span><span class="p">([</span><span class="n">const</span><span class="o">.</span><span class="n">id</span><span class="p">,</span> <span class="n">func</span><span class="o">.</span><span class="n">MAX</span><span class="p">(</span><span class="n">const</span><span class="o">.</span><span class="n">salary</span><span class="p">)],</span> <span class="n">staticTables</span><span class="o">=</span><span class="p">[</span><span class="s1">'employees'</span><span class="p">])</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">select1</span><span class="p">)</span> <span class="o">==</span> <span class="s1">'SELECT id, MAX(salary) FROM employees'</span> <span class="n">select2</span> <span class="o">=</span> <span class="n">Select</span><span class="p">([</span><span class="n">SOTestSQLBuilder</span><span class="o">.</span><span class="n">q</span><span class="o">.</span><span class="n">name</span><span class="p">,</span> <span class="n">SOTestSQLBuilder</span><span class="o">.</span><span class="n">q</span><span class="o">.</span><span class="n">so_value</span><span class="p">])</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">select2</span><span class="p">)</span> <span class="o">==</span> \ <span class="s1">'SELECT so_test_sql_builder.name, so_test_sql_builder.so_value '</span> \ <span class="s1">'FROM so_test_sql_builder'</span> <span class="n">union</span> <span class="o">=</span> <span class="n">Union</span><span class="p">(</span><span class="n">select1</span><span class="p">,</span> <span class="n">select2</span><span class="p">)</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">union</span><span class="p">)</span> <span class="o">==</span> \ <span class="s1">'SELECT id, MAX(salary) FROM employees '</span> \ <span class="s1">'UNION SELECT so_test_sql_builder.name, '</span> \ <span class="s1">'so_test_sql_builder.so_value '</span> \ <span class="s1">'FROM so_test_sql_builder'</span> <span class="n">union</span> <span class="o">=</span> <span class="n">Union</span><span class="p">(</span><span class="n">SOTestSQLBuilder</span><span class="o">.</span><span class="n">select</span><span class="p">()</span><span class="o">.</span><span class="n">queryForSelect</span><span class="p">())</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">union</span><span class="p">)</span> <span class="o">==</span> \ <span class="s1">'SELECT so_test_sql_builder.id, so_test_sql_builder.name, '</span> \ <span class="s1">'so_test_sql_builder.so_value FROM so_test_sql_builder WHERE 1 = 1'</span> </div> <div class="viewcode-block" id="test_empty_AND"><a class="viewcode-back" href="../../../api/sqlobject.tests.test_sqlbuilder.html#sqlobject.tests.test_sqlbuilder.test_empty_AND">[docs]</a><span class="k">def</span> <span class="nf">test_empty_AND</span><span class="p">():</span> <span class="k">assert</span> <span class="n">AND</span><span class="p">()</span> <span class="ow">is</span> <span class="kc">None</span> <span class="k">assert</span> <span class="n">AND</span><span class="p">(</span><span class="kc">True</span><span class="p">)</span> <span class="ow">is</span> <span class="kc">True</span> <span class="c1"># sqlrepr() is needed because AND() returns an SQLExpression that overrides</span> <span class="c1"># comparison. The following</span> <span class="c1"># AND('x', 'y') == "foo bar"</span> <span class="c1"># is True! (-: Eeek!</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">AND</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="o">==</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">SQLOp</span><span class="p">(</span><span class="s2">"AND"</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="o">==</span> <span class="s2">"((1) AND (2))"</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">AND</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="s1">'3'</span><span class="p">),</span> <span class="s2">"sqlite"</span><span class="p">)</span> <span class="o">==</span> \ <span class="n">sqlrepr</span><span class="p">(</span><span class="n">SQLOp</span><span class="p">(</span><span class="s2">"AND"</span><span class="p">,</span> <span class="mi">1</span><span class="p">,</span> <span class="n">SQLOp</span><span class="p">(</span><span class="s2">"AND"</span><span class="p">,</span> <span class="mi">2</span><span class="p">,</span> <span class="s1">'3'</span><span class="p">)),</span> <span class="s2">"sqlite"</span><span class="p">)</span> <span class="o">==</span> \ <span class="s2">"((1) AND ((2) AND ('3')))"</span> </div> <div class="viewcode-block" id="test_modulo"><a class="viewcode-back" href="../../../api/sqlobject.tests.test_sqlbuilder.html#sqlobject.tests.test_sqlbuilder.test_modulo">[docs]</a><span class="k">def</span> <span class="nf">test_modulo</span><span class="p">():</span> <span class="n">setupClass</span><span class="p">(</span><span class="n">SOTestSQLBuilder</span><span class="p">)</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">SOTestSQLBuilder</span><span class="o">.</span><span class="n">q</span><span class="o">.</span><span class="n">so_value</span> <span class="o">%</span> <span class="mi">2</span> <span class="o">==</span> <span class="mi">0</span><span class="p">,</span> <span class="s1">'mysql'</span><span class="p">)</span> <span class="o">==</span> \ <span class="s2">"((MOD(so_test_sql_builder.so_value, 2)) = (0))"</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">SOTestSQLBuilder</span><span class="o">.</span><span class="n">q</span><span class="o">.</span><span class="n">so_value</span> <span class="o">%</span> <span class="mi">2</span> <span class="o">==</span> <span class="mi">0</span><span class="p">,</span> <span class="s1">'sqlite'</span><span class="p">)</span> <span class="o">==</span> \ <span class="s2">"(((so_test_sql_builder.so_value) % (2)) = (0))"</span> </div> <div class="viewcode-block" id="test_division"><a class="viewcode-back" href="../../../api/sqlobject.tests.test_sqlbuilder.html#sqlobject.tests.test_sqlbuilder.test_division">[docs]</a><span class="k">def</span> <span class="nf">test_division</span><span class="p">():</span> <span class="n">SOTestSQLBuilder</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="s1">'test'</span><span class="p">,</span> <span class="n">so_value</span><span class="o">=-</span><span class="mi">11</span><span class="p">)</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">SOTestSQLBuilder</span><span class="o">.</span><span class="n">q</span><span class="o">.</span><span class="n">so_value</span> <span class="o">/</span> <span class="mi">4</span> <span class="o">==</span> <span class="o">-</span><span class="mf">2.75</span><span class="p">,</span> <span class="s1">'mysql'</span><span class="p">)</span> <span class="o">==</span> \ <span class="s2">"(((so_test_sql_builder.so_value) / (4)) = (-2.75))"</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">SOTestSQLBuilder</span><span class="o">.</span><span class="n">q</span><span class="o">.</span><span class="n">so_value</span> <span class="o">//</span> <span class="mi">4</span> <span class="o">==</span> <span class="o">-</span><span class="mi">3</span><span class="p">,</span> <span class="s1">'mysql'</span><span class="p">)</span> <span class="o">==</span> \ <span class="s2">"((FLOOR(((so_test_sql_builder.so_value) / (4)))) = (-3))"</span> </div> <div class="viewcode-block" id="test_str_or_sqlrepr"><a class="viewcode-back" href="../../../api/sqlobject.tests.test_sqlbuilder.html#sqlobject.tests.test_sqlbuilder.test_str_or_sqlrepr">[docs]</a><span class="k">def</span> <span class="nf">test_str_or_sqlrepr</span><span class="p">():</span> <span class="n">select</span> <span class="o">=</span> <span class="n">Select</span><span class="p">([</span><span class="s1">'id'</span><span class="p">,</span> <span class="s1">'name'</span><span class="p">],</span> <span class="n">staticTables</span><span class="o">=</span><span class="p">[</span><span class="s1">'employees'</span><span class="p">],</span> <span class="n">where</span><span class="o">=</span><span class="s1">'value>0'</span><span class="p">,</span> <span class="n">orderBy</span><span class="o">=</span><span class="s1">'id'</span><span class="p">)</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">select</span><span class="p">,</span> <span class="s1">'sqlite'</span><span class="p">)</span> <span class="o">==</span> \ <span class="s1">'SELECT id, name FROM employees WHERE value>0 ORDER BY id'</span> <span class="n">select</span> <span class="o">=</span> <span class="n">Select</span><span class="p">([</span><span class="s1">'id'</span><span class="p">,</span> <span class="s1">'name'</span><span class="p">],</span> <span class="n">staticTables</span><span class="o">=</span><span class="p">[</span><span class="s1">'employees'</span><span class="p">],</span> <span class="n">where</span><span class="o">=</span><span class="s1">'value>0'</span><span class="p">,</span> <span class="n">orderBy</span><span class="o">=</span><span class="s1">'id'</span><span class="p">,</span> <span class="n">lazyColumns</span><span class="o">=</span><span class="kc">True</span><span class="p">)</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">select</span><span class="p">,</span> <span class="s1">'sqlite'</span><span class="p">)</span> <span class="o">==</span> \ <span class="s1">'SELECT id FROM employees WHERE value>0 ORDER BY id'</span> <span class="n">insert</span> <span class="o">=</span> <span class="n">Insert</span><span class="p">(</span><span class="s1">'employees'</span><span class="p">,</span> <span class="n">values</span><span class="o">=</span><span class="p">{</span><span class="s1">'id'</span><span class="p">:</span> <span class="mi">1</span><span class="p">,</span> <span class="s1">'name'</span><span class="p">:</span> <span class="s1">'test'</span><span class="p">})</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">insert</span><span class="p">,</span> <span class="s1">'sqlite'</span><span class="p">)</span> <span class="o">==</span> \ <span class="s2">"INSERT INTO employees (id, name) VALUES (1, 'test')"</span> <span class="n">update</span> <span class="o">=</span> <span class="n">Update</span><span class="p">(</span><span class="s1">'employees'</span><span class="p">,</span> <span class="p">{</span><span class="s1">'name'</span><span class="p">:</span> <span class="s1">'test'</span><span class="p">},</span> <span class="n">where</span><span class="o">=</span><span class="s1">'id=1'</span><span class="p">)</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">update</span><span class="p">,</span> <span class="s1">'sqlite'</span><span class="p">)</span> <span class="o">==</span> \ <span class="s2">"UPDATE employees SET name='test' WHERE id=1"</span> <span class="n">update</span> <span class="o">=</span> <span class="n">Update</span><span class="p">(</span><span class="s1">'employees'</span><span class="p">,</span> <span class="p">{</span><span class="s1">'name'</span><span class="p">:</span> <span class="s1">'test'</span><span class="p">,</span> <span class="s1">'age'</span><span class="p">:</span> <span class="mi">42</span><span class="p">},</span> <span class="n">where</span><span class="o">=</span><span class="s1">'id=1'</span><span class="p">)</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">update</span><span class="p">,</span> <span class="s1">'sqlite'</span><span class="p">)</span> <span class="o">==</span> \ <span class="s2">"UPDATE employees SET age=42, name='test' WHERE id=1"</span> <span class="n">delete</span> <span class="o">=</span> <span class="n">Delete</span><span class="p">(</span><span class="s1">'employees'</span><span class="p">,</span> <span class="n">where</span><span class="o">=</span><span class="s1">'id=1'</span><span class="p">)</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">delete</span><span class="p">,</span> <span class="s1">'sqlite'</span><span class="p">)</span> <span class="o">==</span> \ <span class="s2">"DELETE FROM employees WHERE id=1"</span> <span class="n">raises</span><span class="p">(</span><span class="ne">TypeError</span><span class="p">,</span> <span class="n">Delete</span><span class="p">,</span> <span class="s1">'employees'</span><span class="p">)</span> <span class="n">delete</span> <span class="o">=</span> <span class="n">Delete</span><span class="p">(</span><span class="s1">'employees'</span><span class="p">,</span> <span class="n">where</span><span class="o">=</span><span class="kc">None</span><span class="p">)</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">delete</span><span class="p">,</span> <span class="s1">'sqlite'</span><span class="p">)</span> <span class="o">==</span> \ <span class="s2">"DELETE FROM employees"</span> </div> <div class="viewcode-block" id="test_CONCAT"><a class="viewcode-back" href="../../../api/sqlobject.tests.test_sqlbuilder.html#sqlobject.tests.test_sqlbuilder.test_CONCAT">[docs]</a><span class="k">def</span> <span class="nf">test_CONCAT</span><span class="p">():</span> <span class="n">setupClass</span><span class="p">(</span><span class="n">SOTestSQLBuilder</span><span class="p">)</span> <span class="n">SOTestSQLBuilder</span><span class="p">(</span><span class="n">name</span><span class="o">=</span><span class="s1">'test'</span><span class="p">,</span> <span class="n">so_value</span><span class="o">=</span><span class="mi">42</span><span class="p">)</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">CONCAT</span><span class="p">(</span><span class="s1">'a'</span><span class="p">,</span> <span class="s1">'b'</span><span class="p">),</span> <span class="s1">'mysql'</span><span class="p">)</span> <span class="o">==</span> <span class="s2">"CONCAT('a', 'b')"</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">CONCAT</span><span class="p">(</span><span class="s1">'a'</span><span class="p">,</span> <span class="s1">'b'</span><span class="p">),</span> <span class="s1">'mssql'</span><span class="p">)</span> <span class="o">==</span> <span class="s2">"'a' + 'b'"</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">CONCAT</span><span class="p">(</span><span class="s1">'a'</span><span class="p">,</span> <span class="s1">'b'</span><span class="p">),</span> <span class="s1">'sqlite'</span><span class="p">)</span> <span class="o">==</span> <span class="s2">"'a' || 'b'"</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">CONCAT</span><span class="p">(</span><span class="s1">'prefix'</span><span class="p">,</span> <span class="n">SOTestSQLBuilder</span><span class="o">.</span><span class="n">q</span><span class="o">.</span><span class="n">name</span><span class="p">),</span> <span class="s1">'mysql'</span><span class="p">)</span> <span class="o">==</span> \ <span class="s2">"CONCAT('prefix', so_test_sql_builder.name)"</span> <span class="k">assert</span> <span class="n">sqlrepr</span><span class="p">(</span><span class="n">CONCAT</span><span class="p">(</span><span class="s1">'prefix'</span><span class="p">,</span> <span class="n">SOTestSQLBuilder</span><span class="o">.</span><span class="n">q</span><span class="o">.</span><span class="n">name</span><span class="p">),</span> <span class="s1">'sqlite'</span><span class="p">)</span> <span class="o">==</span> \ <span class="s2">"'prefix' || so_test_sql_builder.name"</span> <span class="n">select</span> <span class="o">=</span> <span class="n">Select</span><span class="p">([</span><span class="n">CONCAT</span><span class="p">(</span><span class="n">SOTestSQLBuilder</span><span class="o">.</span><span class="n">q</span><span class="o">.</span><span class="n">name</span><span class="p">,</span> <span class="s1">'-suffix'</span><span class="p">)],</span> <span class="n">staticTables</span><span class="o">=</span><span class="p">[</span><span class="s1">'so_test_sql_builder'</span><span class="p">])</span> <span class="n">connection</span> <span class="o">=</span> <span class="n">getConnection</span><span class="p">()</span> <span class="n">rows</span> <span class="o">=</span> <span class="n">connection</span><span class="o">.</span><span class="n">queryAll</span><span class="p">(</span><span class="n">connection</span><span class="o">.</span><span class="n">sqlrepr</span><span class="p">(</span><span class="n">select</span><span class="p">))</span> <span class="n">result</span> <span class="o">=</span> <span class="n">rows</span><span class="p">[</span><span class="mi">0</span><span class="p">][</span><span class="mi">0</span><span class="p">]</span> <span class="k">if</span> <span class="ow">not</span> <span class="n">PY2</span> <span class="ow">and</span> <span class="ow">not</span> <span class="nb">isinstance</span><span class="p">(</span><span class="n">result</span><span class="p">,</span> <span class="nb">str</span><span class="p">):</span> <span class="n">result</span> <span class="o">=</span> <span class="n">result</span><span class="o">.</span><span class="n">decode</span><span class="p">(</span><span class="s1">'ascii'</span><span class="p">)</span> <span class="k">assert</span> <span class="n">result</span> <span class="o">==</span> <span class="s2">"test-suffix"</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>