def test_loadKeys(self):
print "Testing loadKeys() with a test_file...",
keys.loadKeys(prependdir+'xml/keys')
t = [KEY,'foo']
self.assertEqual(t, keys._KEYS, 'FAILED (loadkeys)')
self.assert_(not keys._KEYS[-1].active, 'FAILED (bad key active)')
print "OK"
def test_loadKeys(self):
print "Testing loadKeys() with a test_file...",
keys.loadKeys(prependdir + "xml/keys")
t = [KEY, "foo"]
self.assertEqual(t, keys._KEYS, "FAILED (loadkeys)")
self.assert_(not keys._KEYS[-1].active, "FAILED (bad key active)")
print "OK"
def main():
options, args = handleopts()
if args: kfname = args[0]
else: kfname = None
keys.loadKeys(kfname)
if options.add:
if keys.addKey(options.add):
print "Successfully added key: %s" % (options.add)
keys.saveKeys()
sys.exit(0)
else:
print "Invalid key: %s, exiting" % (options.add)
sys.exit(1)
elif options.remove:
# Note on implementation this will call keys.remove(key) in a try
# block like above
print "Option currently disabled, key removal is not yet implemented"
sys.exit(1)
else:
i = 0
for k in keys._KEYS:
i += 1
print '%s. %s: %s' % (i,k.name, k)
sys.exit(0)
def test_loadKeysEmpty(self):
print "Testing loadKeys() with a non-existant file...",
import os, os.path
f = prependdir+'xml/keys2'
if os.path.exists(f):
os.remove(f)
keys.loadKeys(f)
self.assertEqual(keys._KEYS, [], "FAILED (new keyfile not empty)")
print "OK"
def test_loadKeysEmpty(self):
print "Testing loadKeys() with a non-existant file...",
import os, os.path
f = prependdir + "xml/keys2"
if os.path.exists(f):
os.remove(f)
keys.loadKeys(f)
self.assertEqual(keys._KEYS, [], "FAILED (new keyfile not empty)")
print "OK"
def test_saveKeys(self):
import os, os.path
print "Testing SaveKeys...",
keys._KEYFILE = ''
self.assertRaises(Exception, keys.saveKeys)
kf = prependdir+'xml/keys3'
keys.loadKeys(prependdir+'xml/keys')
os.remove(kf)
if os.path.exists(kf): raise Exception, 'didnt delete kf'
keys._KEYFILE = kf
x = keys._KEYS
keys.saveKeys()
keys.loadKeys(kf)
self.assert_(os.path.exists(kf), "FAILED (kf doesnt exits)")
self.assert_(x == keys._KEYS, "FAILED (didnt save right)")
print "OK"
def test_saveKeys(self):
import os, os.path
print "Testing SaveKeys...",
keys._KEYFILE = ""
self.assertRaises(Exception, keys.saveKeys)
kf = prependdir + "xml/keys3"
keys.loadKeys(prependdir + "xml/keys")
os.remove(kf)
if os.path.exists(kf):
raise Exception, "didnt delete kf"
keys._KEYFILE = kf
x = keys._KEYS
keys.saveKeys()
keys.loadKeys(kf)
self.assert_(os.path.exists(kf), "FAILED (kf doesnt exits)")
self.assert_(x == keys._KEYS, "FAILED (didnt save right)")
print "OK"
if sys.version_info[0] == 2 and sys.version_info[1] <= 4:
try:
import cElementTree as ElementTree
except:
from elementtree import ElementTree
elif sys.version_info[0] == 2 and sys.version_info[1] >= 5:
from xml.etree import ElementTree
import unittest
from isbndb.cursors import PageCursor, PyPageCursor, BookCursor, PyBookCursor
from isbndb.internet import params, fetch
from isbndb.keys import getKey, loadKeys, Key
from isbndb import IsbndbSearch
#set up keys...
loadKeys()
class ParamSetupMixin(object):
def setUp(self):
self.p = params()
self.key = getKey()
#set up params
self.p.searchIndex = 'full'
self.p.results = ['texts','details','subjects']
self.p.searchValue = 'sex'
self.p.key = self.key
class TestFetch(unittest.TestCase,ParamSetupMixin):
def setUp(self):
print "\nTestFetch:",
def test_loadKeysSetsFile(self):
print "Testing that loadKeys sets the KEYFILE global...",
keys.loadKeys(prependdir + "xml/keys")
self.assertEqual(keys._KEYFILE, prependdir + "xml/keys", "FAILED (set bad KEYFILE)\n%s" % keys._KEYFILE)
print "OK"
def test_loadKeysSetsFile(self):
print "Testing that loadKeys sets the KEYFILE global...",
keys.loadKeys(prependdir+'xml/keys')
self.assertEqual(keys._KEYFILE, prependdir+'xml/keys',
"FAILED (set bad KEYFILE)\n%s"%keys._KEYFILE)
print "OK"
