Your IP : 216.73.216.213
s_�We8 � @ s� d d l Z d d l Z d d l Z d d l Z d d l m Z d d l m Z d d l Z d d l m
Z
d d l m Z d d d g Z
Gd d � d e
j � Z Gd
d � d e
j � Z Gd d � d e
j � Z d S) � N)�SkipTest)�util)�base)�ProgrammingError�TestConversion�
TestCursor�TestBulkInsertsc @ sp e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d S)r c C sh | j d } | j � } | j d � z-d d" d d d d d j | j � t j d
d d � t j d d
d d d d � t j d
d � t j d d � t j � f } | j d | � | j d � | j
� } | j t j
d � | d � | j | d d � | d d � � | j t j d d | d j d | d j � | d � | j t j | d# d d � � | d$ � | j d � | j d d g d � | j d � | j
� } | j t d g d � | � | j d � xu t t t t f D]a } | j d � | d d g � } | j d | f � | j � } | j d' | � | j d � q�WWd | j d! � Xd S)(z test every data type r z�create table test_datatypes (b bit, i int, l bigint, f real, s varchar(32), u varchar(32), bb blob, d date, dt datetime, ts timestamp, td time, t time, st datetime)T� l 2}r g������@z
hello'" worldu
Españolz binary datai� � i� � � � �- �9 � � � zeinsert into test_datatypes (b,i,l,f,s,u,bb,d,dt,td,t,st) values (%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s,%s)z6select b,i,l,f,s,u,bb,d,dt,td,t,st from test_datatypes� �
�< Nzdelete from test_datatypes� z>insert into test_datatypes (i, l) values (2,4), (6,8), (10,12)z5select l from test_datatypes where i in %s order by i� � zdrop table test_datatypes������r �r �r )r r )� connections�cursor�execute�encode�charset�datetime�date� timedelta�time� localtime�fetchone� assertEqualr �int2byte�hour�minute�tuple�list�set� frozenset�fetchall)�self�conn�c�v�rZseq_type�seq� r7 �I/opt/alt/python35/lib/python3.5/site-packages/pymysql/tests/test_basic.py�test_datatypes s4
{
$:+
z TestConversion.test_datatypesc C s� | j d } | j � } | j d � zI | j d d d d d d d i � | j d
� | j d
| j � � Wd | j d � Xd S)z test dict escaping r z8create table test_dict (a integer, b integer, c integer)z:insert into test_dict (a,b,c) values (%(a)s, %(b)s, %(c)s)�ar �br
r3 r zselect a,b,c from test_dictNzdrop table test_dict)r r
r )r r r r( r' )r1 r2 r3 r7 r7 r8 � test_dict: s
"
zTestConversion.test_dictc C s{ | j d } | j � } | j d � d } z: | j d | � | j d � | j | f | j � � Wd | j d � Xd S)Nr z create table test_dict (a text)zI am a test stringz%insert into test_dict (a) values (%s)zselect a from test_dictzdrop table test_dict)r r r r( r' )r1 r2 r3 �
test_valuer7 r7 r8 � test_stringF s
zTestConversion.test_stringc C s{ | j d } | j � } | j d � d } z: | j d | � | j d � | j | f | j � � Wd | j d � Xd S)Nr z"create table test_dict (a integer)i90 z%insert into test_dict (a) values (%s)zselect a from test_dictzdrop table test_dict)r r r r( r' )r1 r2 r3 r= r7 r7 r8 � test_integerR s
zTestConversion.test_integerc C s� t t t d � � d � } | j d } | j | d d � | j � �A } | j d | f � | j d � | j | | j � d � Wd QRXd S) ztest binary data� r r � test_blobz create table test_blob (b blob)z%insert into test_blob (b) values (%s)zselect b from test_blobN) �bytes� bytearray�ranger �safe_create_tabler r r( r' )r1 �datar2 r3 r7 r7 r8 rA ^ s
zTestConversion.test_blobc C sc | j d } | j � } | j d � | j d | j � � | j d � | j d | j � � d S)z' test conversion of null, empty string r zselect null,''N� zselect '',null)NrG )rG N)r r r r( r' )r1 r2 r3 r7 r7 r8 � test_untypedj s
zTestConversion.test_untypedc
C s� | j d } | j � } | j d � | j t j d d � t j d d � t j d d d � t j d d � t j d d � t j d d d � t j d d � f | j � � d S)z test timedelta conversion r z�select time('12:30'), time('23:12:59'), time('23:12:59.05100'), time('-12:30'), time('-23:12:59'), time('-23:12:59.05100'), time('-00:30')iȯ i{F i8� i N)r r r r( r" r$ r' )r1 r2 r3 r7 r7 r8 �test_timedeltas s
z TestConversion.test_timedeltac C s� | j d } | j | d � s+ t d � � | j � } t j d d d d d d d
� } | j d � z@ | j d d
| f � | j d � | j | f | j � � Wd | j d � Xd S)z( test datetime conversion w microsecondsr r r r z,target backend does not support microsecondsi� � r � i:� z3create table test_datetime (id int, ts datetime(6))z)insert into test_datetime values (%s, %s)r z select ts from test_datetimeNzdrop table test_datetime)r r r )r �mysql_server_isr r r" r r( r' )r1 r2 r3 �dtr7 r7 r8 �test_datetime_microseconds� s
!
z)TestConversion.test_datetime_microsecondsN) �__name__�
__module__� __qualname__r9 r< r> r? rA rH rI rN r7 r7 r7 r8 r s ' c @ s@ e Z d Z d d � Z d d � Z d d � Z d d � Z d S)
r c C sn | j d } | j � } | j d � z3 d } | j d | f � | j d | j � � Wd | j d � Xd S)z test a fetchone() with no rows r z$create table test_nr (b varchar(32))�pymysqlz#insert into test_nr (b) values (%s)Nzdrop table test_nr)r r r r( r' )r1 r2 r3 rF r7 r7 r8 �test_fetch_no_result� s
z TestCursor.test_fetch_no_resultc C s� | j d } | j � } zy | j d � x* t d d � D] } | j d | f � q9 W| j d � | j � \ } | j t t d d � � | � Wd | j d � Xd S)z test aggregate functions r z(create table test_aggregates (i integer)r z+insert into test_aggregates (i) values (%s)z"select sum(i) from test_aggregatesNzdrop table test_aggregates)r r r rD r' r( �sum)r1 r2 r3 �ir5 r7 r7 r8 �test_aggregates� s
#zTestCursor.test_aggregatesc C s� | j d } | j � } | j | d d � | j d � | j d � | j d d
� | j d g t | j � � � | j � d S) z test a single tuple r Zmystuffz-create table mystuff (id integer primary key)z#insert into mystuff (id) values (1)z#insert into mystuff (id) values (2)z%select id from mystuff where id in %sr N�r )rW )r )r r rE r r( r- r0 �close)r1 r2 r3 r7 r7 r8 �test_single_tuple� s
z TestCursor.test_single_tuplec C s
| j d j � } d | d <t j | � } | j | d
� sJ t d � � | j | d d � | j � } d } | j d
| f � | j d � | j � d } | j
t j | � t j | � � | j d | f � | j � d } | j
t j | � t j | � � d S)Nr �utf8mb4r! r r
z*JSON type is not supported on MySQL <= 5.6� test_jsonz]create table test_json (
id int not null,
json JSON not null,
primary key (id)
);u {"hello": "こんにちは"}z2INSERT INTO test_json (id, `json`) values (42, %s)z,SELECT `json` from `test_json` WHERE `id`=42z SELECT CAST(%s AS JSON) AS x)r r
r )
Z databases�copyrR �connectrL r rE r r r' r( �json�loads)r1 �argsr2 �curZjson_str�resr7 r7 r8 r[ � s
"zTestCursor.test_jsonN)rO rP rQ rS rV rY r[ r7 r7 r7 r8 r � s 6
c sv e Z d Z e j j Z � f d d � Z d d � Z d d � Z d d � Z
d d
� Z d d � Z d
d � Z
� S)r c sP t t | � j � | j d | _ } | j | j � } | j | d d � d S)Nr Z
bulkinsertz]CREATE TABLE bulkinsert
(
id int(11),
name char(20),
age int,
height int,
PRIMARY KEY (id)
)
)�superr �setUpr r2 r � cursor_typerE )r1 r2 r3 )� __class__r7 r8 rd s
zTestBulkInserts.setUpc C sR | j d } | j � } | j d � | j � } | j t | � t | � � d S)Nr z,SELECT id, name, age, height from bulkinsert)r r r r0 r( �sorted)r1 rF r2 r �resultr7 r7 r8 �_verify_records! s
z TestBulkInserts._verify_recordsc C so | j d } | j � } d d d g } | j d
| � | j | j t d � � | j d � | j | � d S)Nr �bob� �{ r �jim�8 r r
�fred�d � zCinsert into bulkinsert (id, name, age, height) values (%s,%s,%s,%s)si insert into bulkinsert (id, name, age, height) values (0,'bob',21,123),(1,'jim',56,45),(2,'fred',100,180)�commit)r rj rk rl )r rm rn r )r
ro rp rq )r r � executemanyr( �_last_executedrC r ri )r1 r2 r rF r7 r7 r8 �test_bulk_insert( s
z TestBulkInserts.test_bulk_insertc C su | j d } | j � } d d d g } | j d
| � | j | j j � t d � � | j d � | j | � d S)Nr rj rk rl r rm rn r r
ro rp rq zJinsert
into bulkinsert (id, name,
age, height)
values (%s,
%s , %s,
%s )
sx insert
into bulkinsert (id, name,
age, height)
values (0,
'bob' , 21,
123 ),(1,
'jim' , 56,
45 ),(2,
'fred' , 100,
180 )rr )r rj rk rl )r rm rn r )r
ro rp rq ) r r rs r( rt �striprC r ri )r1 r2 r rF r7 r7 r8 �$test_bulk_insert_multiline_statement6 s
z4TestBulkInserts.test_bulk_insert_multiline_statementc C sP | j d } | j � } d g } | j d | � | j d � | j | � d S)Nr rj rk rl zCinsert into bulkinsert (id, name, age, height) values (%s,%s,%s,%s)rr )r rj rk rl )r r rs r ri )r1 r2 r rF r7 r7 r8 � test_bulk_insert_single_recordN s
z.TestBulkInserts.test_bulk_insert_single_recordc C su | j d } | j � } d d d g } | j d
| � | j | j j � t d � � | j d � | j | � d S)z4executemany should work with "insert ... on update" r rj rk rl r rm rn r r
ro rp rq ztinsert
into bulkinsert (id, name,
age, height)
values (%s,
%s , %s,
%s ) on duplicate key update
age = values(age)
s� insert
into bulkinsert (id, name,
age, height)
values (0,
'bob' , 21,
123 ),(1,
'jim' , 56,
45 ),(2,
'fred' , 100,
180 ) on duplicate key update
age = values(age)rr N)r rj rk rl )r rm rn r )r
ro rp rq ) r r rs r( rt rv rC r ri )r1 r2 r rF r7 r7 r8 �test_issue_288W s
z TestBulkInserts.test_issue_288c C s� | j d } | j � } t j d d � �! } t j d � | j d � Wd QRX| j t | � d � | j | d j t j
� d t | d j � k r� | j
d t | d j � f � d S) Nr �recordT�alwaysz$drop table if exists no_exists_tabler Zno_exists_tablez'no_exists_table' not in %s)r r �warnings�catch_warnings� simplefilterr r( �len�categoryrR �Warning�str�messageZfail)r1 Zconra �wsr7 r7 r8 �
test_warningsr s
z TestBulkInserts.test_warnings)rO rP rQ rR �cursors�
DictCursorre rd ri ru rw rx ry r� r7 r7 )rf r8 r s )r" r^ r% r| Z unittest2r rR r Zpymysql.cursorsZ
pymysql.testsr Z pymysql.errr �__all__ZPyMySQLTestCaser r r r7 r7 r7 r8 �<module> s �w