Page MenuHomec4science
Files F97876836

ut_adapter.py
No OneTemporary
Actions

File Metadata

Created
Tue, Jan 7, 02:08

ut_adapter.py
View Options

import pybut
from Pyblio import Adapter, Store, Schema, Attribute, Registry
class A2BAdapter(Adapter.OneToOneAdapter):
""" This adapter renames field 'a' into field 'b' """
schema = Schema.Schema(open(pybut.src('ut_adapter/b.sip')))
def source2target(self, record):
new = Store.Record()
new.key = record.key
if 'a' in record:
new['b'] = record['a']
return new
def target2source(self, record):
new = Store.Record()
new.key = record.key
if 'b' in record:
new['a'] = record['b']
return new
class TestFromAtoB(pybut.TestCase):
def setUp(self):
sa = Schema.Schema(open(pybut.src('ut_adapter/a.sip')))
fmt = Store.get('memory')
self.dba = fmt.dbcreate(None, sa)
self.dbb = A2BAdapter(self.dba)
return
def testAddAndDel(self):
""" Add to A, and check in B """
r = Store.Record()
k = r.add('a', 'A sample !', Attribute.Text)
# Add a record in A...
self.dba.add(r)
# ...and it show up in B (with the proper value)
res = [self.dbb[k] for k in self.dbb.entries]
self.failUnlessEqual(len(res), 1)
new = res[0]
self.failUnlessEqual(new['b'], r['a'])
res = list(self.dbb.entries.itervalues())[0]
self.failUnlessEqual(res['b'], r['a'])
k, res = list(self.dbb.entries.iteritems())[0]
self.failUnlessEqual(res['b'], r['a'])
del self.dba[k]
self.failUnlessEqual(list(self.dbb.entries.itervalues()), [])
def testResultSet(self):
r = Store.Record()
k = r.add('a', 'A sample !', Attribute.Text)
# Add a record in A...
k = self.dba.add(r)
rs = self.dbb.rs.new()
rs.add(k)
self.failUnlessEqual(list(rs.iteritems()), [(k, self.dbb[k])])
def testXMLWrite(self):
r = Store.Record()
k = r.add('a', 'A sample !', Attribute.Text)
self.dba.add(r)
tmp = pybut.dbname()
fd = open(tmp, 'w')
self.dbb.xmlwrite(fd)
fd.close()
pybut.fileeq(tmp, pybut.src('ut_adapter/saved-b.bip'))
class TestResolution(pybut.TestCase):
def testResolveA2B(self):
sa = Schema.Schema(open(pybut.src('ut_adapter/a.sip')))
fmt = Store.get('memory')
self.dba = fmt.dbcreate(None, sa)
from Pyblio import Registry
Registry.reset()
Registry.load_settings(pybut.src('ut_adapter'))
dest = Adapter.adapt_schema(self.dba, 'b')
self.failUnlessEqual(dest.schema.id, 'b')
class TestAdapters(pybut.TestCase):
def setUp(self):
Registry.load_default_settings()
def tearDown(self):
Registry.reset()
def check(self, name, schema):
src = Store.get('file').dbopen(
pybut.src('ut_adapter/' + name + '-src.bip'))
dst = Adapter.adapt_schema(src, schema)
tmp = pybut.dbname()
fd = open(tmp, 'w')
dst.xmlwrite(fd)
fd.close()
pybut.fileeq(pybut.src('ut_adapter/' + name + '-dst.bip'), tmp)
def testWOK2BibTeX(self):
self.check('wok-to-bibtex', 'org.pybliographer/bibtex/0.1')
suite = pybut.suite(TestFromAtoB, TestResolution)
if __name__ == '__main__': pybut.run (suite)

Event Timeline

c4science · Help