dataimport/test/test_pgstore.py
changeset 10943 1079d68130e1
parent 10810 0768bf2333a7
equal deleted inserted replaced
10942:b089858c49a2 10943:1079d68130e1
    22 
    22 
    23 from six import PY2
    23 from six import PY2
    24 from logilab.common.testlib import TestCase, unittest_main
    24 from logilab.common.testlib import TestCase, unittest_main
    25 
    25 
    26 from cubicweb.dataimport import pgstore
    26 from cubicweb.dataimport import pgstore
       
    27 from cubicweb.devtools import testlib
    27 
    28 
    28 
    29 
    29 class CreateCopyFromBufferTC(TestCase):
    30 class CreateCopyFromBufferTC(TestCase):
    30 
    31 
    31     # test converters
    32     # test converters
    88                      date=DT.datetime(2014, 1, 1, 0, 0, 0)))
    89                      date=DT.datetime(2014, 1, 1, 0, 0, 0)))
    89         results = pgstore._create_copyfrom_buffer(data, ('integer', 'text'))
    90         results = pgstore._create_copyfrom_buffer(data, ('integer', 'text'))
    90         expected = u'''42\téléphant\n6\tbabar'''
    91         expected = u'''42\téléphant\n6\tbabar'''
    91         self.assertEqual(expected, results.getvalue())
    92         self.assertEqual(expected, results.getvalue())
    92 
    93 
       
    94 
       
    95 class SQLGenObjectStoreTC(testlib.CubicWebTC):
       
    96 
       
    97     def test_prepare_insert_entity(self):
       
    98         with self.admin_access.repo_cnx() as cnx:
       
    99             store = pgstore.SQLGenObjectStore(cnx)
       
   100             eid = store.prepare_insert_entity('CWUser', login=u'toto',
       
   101                                               upassword=u'pwd')
       
   102             self.assertIsNotNone(eid)
       
   103 
       
   104 
    93 if __name__ == '__main__':
   105 if __name__ == '__main__':
    94     unittest_main()
   106     unittest_main()