Move NEWIMPORT into config

Fix test_PokerStars test to catch exception when importing with the legacy importer
This commit is contained in:
Worros 2009-11-30 14:33:32 +08:00
parent 152933ed49
commit 36dda2174b
2 changed files with 26 additions and 13 deletions

View File

@ -99,7 +99,7 @@ class Importer:
for i in xrange(self.settings['threads']): for i in xrange(self.settings['threads']):
self.writerdbs.append( Database.Database(self.config, sql = self.sql) ) self.writerdbs.append( Database.Database(self.config, sql = self.sql) )
self.NEWIMPORT = False self.NEWIMPORT = Configuration.NEWIMPORT
#Set functions #Set functions
def setCallHud(self, value): def setCallHud(self, value):

View File

@ -97,7 +97,11 @@ def testStudImport():
(stored, dups, partial, errs, ttime) = importer.runImport() (stored, dups, partial, errs, ttime) = importer.runImport()
importer.clearFileList() importer.clearFileList()
# Should actually do some testing here
assert 1 == 1
def testDrawImport(): def testDrawImport():
try:
db.recreate_tables() db.recreate_tables()
importer = fpdb_import.Importer(False, settings, config) importer = fpdb_import.Importer(False, settings, config)
importer.setDropIndexes("don't drop") importer.setDropIndexes("don't drop")
@ -110,3 +114,12 @@ def testDrawImport():
importer.setCallHud(False) importer.setCallHud(False)
(stored, dups, partial, errs, ttime) = importer.runImport() (stored, dups, partial, errs, ttime) = importer.runImport()
importer.clearFileList() importer.clearFileList()
except FpdbError:
if Configuration.NEWIMPORT == False:
#Old import code doesn't support draw
pass
else:
assert 0 == 1
# Should actually do some testing here
assert 1 == 1