11import unittest2
2- import six
32
43
54class TestConnection (unittest2 .TestCase ):
@@ -440,7 +439,7 @@ def test_commit_wo_transaction(self):
440439 insert .key .CopyFrom (key_pb )
441440 prop = insert .property .add ()
442441 prop .name = 'foo'
443- prop .value .string_value = six . u ( 'Foo' )
442+ prop .value .string_value = u 'Foo'
444443 conn = self ._makeOne ()
445444 URI = '/' .join ([
446445 conn .API_BASE_URL ,
@@ -482,7 +481,7 @@ def id(self):
482481 insert .key .CopyFrom (key_pb )
483482 prop = insert .property .add ()
484483 prop .name = 'foo'
485- prop .value .string_value = six . u ( 'Foo' )
484+ prop .value .string_value = u 'Foo'
486485 conn = self ._makeOne ()
487486 conn .transaction (Xact ())
488487 URI = '/' .join ([
@@ -644,7 +643,7 @@ def test_save_entity_wo_transaction_w_upsert(self):
644643 'commit' ,
645644 ])
646645 http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
647- result = conn .save_entity (DATASET_ID , key_pb , {'foo' : six . u ( 'Foo' ) })
646+ result = conn .save_entity (DATASET_ID , key_pb , {'foo' : u 'Foo' })
648647 self .assertEqual (result , True )
649648 cw = http ._called_with
650649 self .assertEqual (cw ['uri' ], URI )
@@ -665,7 +664,7 @@ def test_save_entity_wo_transaction_w_upsert(self):
665664 props = list (upsert .property )
666665 self .assertEqual (len (props ), 1 )
667666 self .assertEqual (props [0 ].name , 'foo' )
668- self .assertEqual (props [0 ].value .string_value , six . u ( 'Foo' ) )
667+ self .assertEqual (props [0 ].value .string_value , u 'Foo' )
669668 self .assertEqual (props [0 ].value .indexed , True )
670669 self .assertEqual (len (mutation .delete ), 0 )
671670 self .assertEqual (request .mode , rq_class .NON_TRANSACTIONAL )
@@ -687,7 +686,7 @@ def test_save_entity_w_exclude_from_indexes(self):
687686 'commit' ,
688687 ])
689688 http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
690- result = conn .save_entity (DATASET_ID , key_pb , {'foo' : six . u ( 'Foo' ) },
689+ result = conn .save_entity (DATASET_ID , key_pb , {'foo' : u 'Foo' },
691690 exclude_from_indexes = ['foo' ])
692691 self .assertEqual (result , True )
693692 cw = http ._called_with
@@ -709,7 +708,7 @@ def test_save_entity_w_exclude_from_indexes(self):
709708 props = list (upsert .property )
710709 self .assertEqual (len (props ), 1 )
711710 self .assertEqual (props [0 ].name , 'foo' )
712- self .assertEqual (props [0 ].value .string_value , six . u ( 'Foo' ) )
711+ self .assertEqual (props [0 ].value .string_value , u 'Foo' )
713712 self .assertEqual (props [0 ].value .indexed , False )
714713 self .assertEqual (len (mutation .delete ), 0 )
715714 self .assertEqual (request .mode , rq_class .NON_TRANSACTIONAL )
@@ -736,7 +735,7 @@ def test_save_entity_wo_transaction_w_auto_id(self):
736735 'commit' ,
737736 ])
738737 http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
739- result = conn .save_entity (DATASET_ID , key_pb , {'foo' : six . u ( 'Foo' ) })
738+ result = conn .save_entity (DATASET_ID , key_pb , {'foo' : u 'Foo' })
740739 self .assertEqual (result , updated_key_pb )
741740 cw = http ._called_with
742741 self .assertEqual (cw ['uri' ], URI )
@@ -755,7 +754,7 @@ def test_save_entity_wo_transaction_w_auto_id(self):
755754 props = list (insert .property )
756755 self .assertEqual (len (props ), 1 )
757756 self .assertEqual (props [0 ].name , 'foo' )
758- self .assertEqual (props [0 ].value .string_value , six . u ( 'Foo' ) )
757+ self .assertEqual (props [0 ].value .string_value , u 'Foo' )
759758 self .assertEqual (len (inserts ), 1 )
760759 upserts = list (mutation .upsert )
761760 self .assertEqual (len (upserts ), 0 )
@@ -777,7 +776,7 @@ def mutation(self):
777776 conn = self ._makeOne ()
778777 conn .transaction (Xact ())
779778 http = conn ._http = Http ({'status' : '200' }, rsp_pb .SerializeToString ())
780- result = conn .save_entity (DATASET_ID , key_pb , {'foo' : six . u ( 'Foo' ) })
779+ result = conn .save_entity (DATASET_ID , key_pb , {'foo' : u 'Foo' })
781780 self .assertEqual (result , True )
782781 self .assertEqual (http ._called_with , None )
783782 mutation = conn .mutation ()
@@ -795,7 +794,7 @@ def mutation(self):
795794 return mutation
796795 DATASET_ID = 'DATASET'
797796 nested = Entity ()
798- nested ['bar' ] = six . u ( 'Bar' )
797+ nested ['bar' ] = u 'Bar'
799798 key_pb = Key (path = [{'kind' : 'Kind' , 'id' : 1234 }]).to_protobuf ()
800799 rsp_pb = datastore_pb .CommitResponse ()
801800 conn = self ._makeOne ()
0 commit comments