dataimport/test/test_pgstore.py
changeset 10984 a42f6e7cef35
parent 10513 7bec01a59f92
child 10987 a945fe89b33f
equal deleted inserted replaced
10983:da8168612e61 10984:a42f6e7cef35
    21 import datetime as DT
    21 import datetime as DT
    22 
    22 
    23 from logilab.common.testlib import TestCase, unittest_main
    23 from logilab.common.testlib import TestCase, unittest_main
    24 
    24 
    25 from cubicweb.dataimport import pgstore
    25 from cubicweb.dataimport import pgstore
       
    26 from cubicweb.devtools import testlib
    26 
    27 
    27 
    28 
    28 class CreateCopyFromBufferTC(TestCase):
    29 class CreateCopyFromBufferTC(TestCase):
    29 
    30 
    30     # test converters
    31     # test converters
    86                      date=DT.datetime(2014, 1, 1, 0, 0, 0)))
    87                      date=DT.datetime(2014, 1, 1, 0, 0, 0)))
    87         results = pgstore._create_copyfrom_buffer(data, ('integer', 'text'))
    88         results = pgstore._create_copyfrom_buffer(data, ('integer', 'text'))
    88         expected = '''42\téléphant\n6\tbabar'''
    89         expected = '''42\téléphant\n6\tbabar'''
    89         self.assertMultiLineEqual(expected, results.getvalue())
    90         self.assertMultiLineEqual(expected, results.getvalue())
    90 
    91 
       
    92 
       
    93 class SQLGenObjectStoreTC(testlib.CubicWebTC):
       
    94 
       
    95     def test_prepare_insert_entity(self):
       
    96         with self.admin_access.repo_cnx() as cnx:
       
    97             store = pgstore.SQLGenObjectStore(cnx)
       
    98             eid = store.prepare_insert_entity('CWUser', login=u'toto',
       
    99                                               upassword=u'pwd')
       
   100             self.assertIsNotNone(eid)
       
   101 
       
   102 
    91 if __name__ == '__main__':
   103 if __name__ == '__main__':
    92     unittest_main()
   104     unittest_main()