3 (h O @sddlZddlZddlZddlZddlZddlmZddlmZddlm Z ddl Z yddl Z e j Z Wne k rzYnXdddgZGddde jZGd dde jZGd dde jZdS) N)cursors) text_type)base TestOldIssues TestNewIssuesTestGitHubIssuesc@sfeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ e j dddZdS)rcCs|jd}|j}tjtjd|jdWdQRX|jdz|jdd |jd|jd|jd|jd|jd|jd|jd |jd|jd|jd |jt |jdt j Wd|jd XdS) z2 undefined methods datetime_or_None, date_or_None rignorezdrop table if exists issue3Nz?create table issue3 (d date, t time, dt datetime, ts timestamp)z6insert into issue3 (d, t, dt, ts) values (%s,%s,%s,%s)zselect d from issue3zselect t from issue3zselect dt from issue3zselect ts from issue3zdrop table issue3)NNNN) connectionscursorwarningscatch_warningsfilterwarningsexecute assertEqualfetchone assertTrue isinstancedatetime)selfconncr>/tmp/pip-install-q3hcpn_q/PyMySQL/pymysql/tests/test_issues.py test_issue_3s"         zTestOldIssues.test_issue_3cCs|jd}|j}tjtjd|jdWdQRX|jdz2|jd|jd|jt|jdt j Wd|jdXdS) z! can't retrieve TIMESTAMP fields rrzdrop table if exists issue4Nz"create table issue4 (ts timestamp)z&insert into issue4 (ts) values (now())zselect ts from issue4zdrop table issue4) r r r r r rrrrr)rrrrrr test_issue_4+s      zTestOldIssues.test_issue_4cCs |jd}|j}|jddS)z* query on information_schema.tables fails rz'select * from information_schema.tablesN)r r r)rconcurrrr test_issue_5:s zTestOldIssues.test_issue_5cCs@|jdj}d|d<tjf|}|j}|jd|jdS)zP exception: TypeError: ord() expected a character, but string of length 0 found rZmysqldbzselect * from userN) databasescopypymysqlconnectr rclose)rkwargsrrrrr test_issue_6@s   zTestOldIssues.test_issue_6cCs|jd}|j}tjtjd|jdWdQRX|jdz2|jd|jd|jd|jd|jdWd|jdXdS) z1 Primary Key and Index error when selecting data rrzdrop table if exists testNaCREATE TABLE `test` (`station` int(10) NOT NULL DEFAULT '0', `dh` datetime NOT NULL DEFAULT '2015-01-01 00:00:00', `echeance` int(1) NOT NULL DEFAULT '0', `me` double DEFAULT NULL, `mo` double DEFAULT NULL, PRIMARY KEY (`station`,`dh`,`echeance`)) ENGINE=MyISAM DEFAULT CHARSET=latin1;zSELECT * FROM testz6ALTER TABLE `test` ADD INDEX `idx_station` (`station`)zdrop table test)r r r r r rr)rrrrrr test_issue_8Js    zTestOldIssues.test_issue_8c Cs.y ttWntk r(|jYnXdS)z' sets DeprecationWarning in Python 2.6 N)reloadr!DeprecationWarningfail)rrrr test_issue_9\s zTestOldIssues.test_issue_9cCs|jd}|j}tjtjd|jdWdQRXzL|jdd }|jdd |f|jd |jd}|jd ||kWd|jd XdS) z" can't handle large result fields rrzdrop table if exists issue13Nzcreate table issue13 (t text)iz#insert into issue13 (t) values (%s)xzselect t from issue13zdrop table issue13iH)r r r r r rrr)rrrsizerrrr test_issue_13cs      zTestOldIssues.test_issue_13cCs|jd}|j}tjtjd|jdWdQRX|jdz.|jdd |jd|jd|jdWd|jd XdS) z< query should be expanded before perform character encoding rrzdrop table if exists issue15Nz$create table issue15 (t varchar(32))z#insert into issue15 (t) values (%s)äöüzselect t from issue15zdrop table issue15)r0)r r r r r rrr)rrrrrr test_issue_15vs      zTestOldIssues.test_issue_15cCs|jd}|j}tjtjd|jdWdQRX|jdz.|jd|jdd |jd |jdWd|jd XdS) z% Patch for string and tuple escaping rrzdrop table if exists issue16NzFcreate table issue16 (name varchar(32) primary key, email varchar(32))z@insert into issue16 (name, email) values ('pete', 'floydophone')z'select email from issue16 where name=%speteZ floydophonezdrop table issue16)r2)r r r r r rrr)rrrrrr test_issue_16s      zTestOldIssues.test_issue_16zRtest_issue_17() requires a custom, legacy MySQL configuration and will not be run.cCs|jd}|jdd}|jdd}|j}ztjtjd|jdWdQRX|jd|jd|jd ||jtj |d d |d }|j}|jd |j d|j dWd|jdXdS)z#could not connect mysql use passwodrhostrrzdrop table if exists issue17Nz0create table issue17 (x varchar(32) primary key)z0insert into issue17 (x) values ('hello, world!')zMgrant all privileges on %s.issue17 to 'issue17user'@'%%' identified by '1234'Z issue17userZ1234)r4userpasswdrzselect x from issue17z hello, world!zdrop table issue17) r rr r r r rcommitr!r"rr)rrr4rrZconn2c2rrr test_issue_17s"      zTestOldIssues.test_issue_17N)__name__ __module__ __qualname__rrrr%r&r*r/r1r3 unittest2skipr9rrrrrs c@sNeZdZddZddZejdddZdd Zd d Z d d Z ddZ dS)rcCsrytjdddd|jWnPtjk rR}z|jd|jdWYdd}~Xntk rl|jYnXdS)N localhostiroot)r4portr5ir)r!r"r)OperationalErrorrargs Exception)rerrr test_issue_34s "zTestNewIssues.test_issue_34cCs\tjfddi|jd}|j|dd|j}|jd|jd|jd|jddS) Ncharsetutf8ruheißeu&create table heiße (name varchar(32))u,insert into heiße (name) values ('Pißata')uselect name from heißeuPißata)r!r"rsafe_create_tabler rrr)rrrrrr test_issue_33s  zTestNewIssues.test_issue_33z&This test requires manual interventioncCsl|jd}|j}tdy|jd|jWn6tjk rf}z|jd|jdWYdd}~XnXdS)Nrz1sudo killall -9 mysqld within the next 10 secondszselect sleep(10)i) r r printrr)r!rBrrC)rrrrErrr test_issue_35s   zTestNewIssues.test_issue_35cCs|jd}|j}|jdd}x.|jD]"}|d}|d}|dkr*|}Pq*W|j||j|jdj|y|jd|jWntk rYnX|j |j zFt j d|jdj}|jddd|jD}|j ||kWd|jd=XdS) Nzshow processlistrz show tablesg?cSsg|] }|dqS)rr).0rowrrr sz/TestNewIssues.test_issue_36..) r r rfetchallrZ thread_idkillr)rDr#timesleep assertFalse)rrrZkill_idrPidinfoidsrrr test_issue_36s4      zTestNewIssues.test_issue_36cCsT|jd}|j}|jd|jd|jd|j|jd|jd|jddS)NrrMz SELECT @foozSET @foo = 'bar'zset @foo = 'bar')N)r r rrr)rrrrrr test_issue_37s  zTestNewIssues.test_issue_37cCst|jd}|j}ddd}zDtjtjd|jdWdQRX|jd|jd|fWd|jd XdS) Nraiirzdrop table if exists issue38z2create table issue38 (id integer, data mediumblob)z"insert into issue38 values (1, %s)zdrop table issue38)r r r r r r)rrrZdatumrrr test_issue_38s     zTestNewIssues.test_issue_38cCs|jd}|j}tjtjd|jdWdQRXd}|djddtddD7}z6|jd |jd |j||jd |j dWd|jd XdS) Nrrzdrop table if exists issue54zselect * from issue54 where z and css|]}d||fVqdS)z%d=%dNr)rOirrr sz7TestNewIssues.disabled_test_issue_54..iz-create table issue54 (id integer primary key)z#insert into issue54 (id) values (7)rNzdrop table issue54) r r r r r rjoinrangerr)rrrZbig_sqlrrrdisabled_test_issue_54s      z$TestNewIssues.disabled_test_issue_54N) r:r;r<rFrJr=r>rLrZr[r]rbrrrrrs   $c@sTeZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ dS)rcCs|jd}|j}|jd|jzZtjtjd|jdWdQRX|jd|jd|jd|jd|jWd|jdXdS) z2 'Connection' object has no attribute 'insert_id' rrzdrop table if exists issue66NzGcreate table issue66 (id integer primary key auto_increment, x integer)z"insert into issue66 (x) values (1)zdrop table issue66)r r rZ insert_idr r r r)rrrrrr test_issue_66s      zTestGitHubIssues.test_issue_66cCs|jd}|jtjj}tj$tjd|jd|jdWdQRX|jd|jdd}d}zb|jd ||jd ||jd |j d}|j |dd|j |dd |j |dd Wd|jd|jdXdS)zI Duplicate field overwrites the previous one in the result of DictCursor rrzdrop table if exists azdrop table if exists bNz"CREATE TABLE a (id int, value int)z"CREATE TABLE b (id int, value int)rM zinsert into a values (%s, %s)zinsert into b values (%s, %s)z+SELECT * FROM a inner join b on a.id = b.idrWvaluezb.valuez drop table az drop table b)rMre)rMrf) r r r!rZ DictCursorr r r rrRr)rrrr\br.rrr test_issue_79's(           zTestGitHubIssues.test_issue_79cCs|jd}|j}tjtjd|jdWdQRX|jdz,|jd|jd|j|jddWdtjtjd|jdWdQRXXdS) z6 Leftover trailing OK packet for "CALL my_sp" queries rrzDROP PROCEDURE IF EXISTS `foo`NzICREATE PROCEDURE `foo` () BEGIN SELECT 1; ENDz CALL foo()zSELECT 1rM)r r r r r rrr)rrrrrr test_issue_95Bs       zTestGitHubIssues.test_issue_95cCstjfddi|jd}|jd|j}|jd|j|jd|j|j |jd|j|jd|jtjfddi|jd}|j}|jd|j|jd|j|j |jd|jd|j |jd|jdS)z6 autocommit is not set after reconnecting with ping() rGrHrFzselect @@autocommit;TN) r!r"rZ autocommitr rrVrr#Zpingr)rrrrrrtest_issue_114Vs(      zTestGitHubIssues.test_issue_114cCs|jd}|j}xd D]|}djddt|D}dj|}z*|j||jdt|j|ksftWd t j t j d |jd Wd QRXXqWd S) z> The number of fields returned by server is read in wrong way r,z, css|]}dj|VqdS)z c{0} integerN)format)rOr^rrrr_tsz2TestGitHubIssues.test_issue_175..z#create table test_field_count ({0})zselect * from test_field_countNrz%drop table if exists test_field_count)rlrm) r r r`rarnrlen descriptionAssertionErrorr r r )rrrlengthcolumnssqlrrrtest_issue_175os       zTestGitHubIssues.test_issue_175cCstjfddi|jd}|j|ddd}d}d}dd gd gd gd ggd d g}|j}|j|j||dd|j|j||dd|j|j||dd|j|j|dd gd|j|jd|j|jd|j|jddS)z" Test iterable as query argument. rGrHrZissue321z>create table issue321 (value_1 varchar(1), value_2 varchar(1))z7insert into issue321 (value_1, value_2) values (%s, %s)zIinsert into issue321 (value_1, value_2) values (%(value_1)s, %(value_2)s)z9select * from issue321 where value_1 in %s and value_2=%sr\аrhr)Zvalue_1Zvalue_2rMrcN)r\)r\rhr)r\rv)rhrv)rrv)r!r"rrIr rrr)rrZ sql_insertZsql_dict_insertZ sql_selectdatarrrrtest_issue_321s&zTestGitHubIssues.test_issue_321cCstjfddi|jd}|j|ddd}d}tjddg}|j}|j||d |jd |j|jt ||j||d |j ||||fd |jd x |j D]}|j|t |qW|j ||||fd d S) z5 Test mixed unicode/binary arguments in executemany. rGrHrZissue364z`create table issue364 (value_1 binary(3), value_2 varchar(3)) engine=InnoDB default charset=utf8z7insert into issue364 (value_1, value_2) values (%s, %s)suäöü)rCzselect * from issue364N) r!r"rrIBinaryr rrrtupleZ executemanyrR)rrrtZusqlvaluesrrPrrrtest_issue_364s"  zTestGitHubIssues.test_issue_364cCsDtjfddi|jd}|j|dd|j}d}|j|drf|jtjj}|j |Wd QRXn |j |d }|j|dr|jtjj}|j |Wd QRXn |j ||j }|j |dd }|j|dr|jtjj}|j |Wd QRXn |j ||j }|j |d|j d|j }|j t |dtd S)z Test binary / geometry types. rGrHrZissue363zCREATE TABLE issue363 ( id INTEGER PRIMARY KEY, geom LINESTRING NOT NULL, SPATIAL KEY geom (geom)) ENGINE=MyISAM default charset=utf8zYINSERT INTO issue363 (id, geom) VALUES(1998, GeomFromText('LINESTRING(1.1 1.1,2.2 2.2)'))rNNz!SELECT AsText(geom) FROM issue363LINESTRING(1.1 1.1,2.2 2.2)z#SELECT AsBinary(geom) FROM issue363)??@@zSELECT geom FROM issue363)r~rNr)r~rNr)r)r~rNr)r)r!r"rrIr Zmysql_server_isZ assertWarnserrWarningrrrrrbytes)rrrquerycmrPrrrtest_issue_363s8       zTestGitHubIssues.test_issue_363cCstjfddi|jd}tjtjdtjdtjdxtjtj fD]}|j |}zy|j d|j Wn^tjk r}z@|j t|jd|j |jdd |jt|jd tWYd d }~Xn X|jd Wd |jXqLWWd QRXd S) z Test warning propagation rGrHrrerror)categoryzSELECT CAST('124b' AS SIGNED)rci rMNzShould raise Warning)r!r"rr r simplefilterrrCursorZSSCursorr rrRrrorCrrrr)r#)rrZ cursor_classrrErrrtest_issue_491s     (zTestGitHubIssues.test_issue_491N) r:r;r<rdrirjrkruryr}rrrrrrrs4)rrTr sysr!rZpymysql._compatrZ pymysql.testsrr=impr'AttributeError__all__ZPyMySQLTestCaserrrrrrrs$     i