def setUp(self):
self.tmpdir = mkdtemp()
self.csvfile = os.path.join(self.tmpdir,"%s.csv" % LoadCSVPathTestCase.fkey1)
f=open(self.csvfile,"w")
f.write(LoadCSVPathTestCase.csvrec)
f.close()
self.ds = CSVDataSource()
class LoadCSVPathTestCase(unittest.TestCase):
""" Test cases for CSVDataSource class"""
csvrec="ID,Name\n1,Marvin\n"
csvdict=dict(ID="1",Name="Marvin")
fkey1="listtest"
def setUp(self):
self.tmpdir = mkdtemp()
self.csvfile = os.path.join(self.tmpdir,"%s.csv" % LoadCSVPathTestCase.fkey1)
f=open(self.csvfile,"w")
f.write(LoadCSVPathTestCase.csvrec)
f.close()
self.ds = CSVDataSource()
def tearDown(self):
os.unlink(self.csvfile)
os.removedirs(self.tmpdir)
def testValidCSVFile(self):
"""test the structure and content of the resulting dictionary"""
self.ds.loadFromPath(self.csvfile)
self.assertTrue(LoadCSVPathTestCase.fkey1 in self.ds)
fkey1list=self.ds[LoadCSVPathTestCase.fkey1]
self.assertTrue(isinstance(fkey1list,list))
self.assertTrue(len(fkey1list)==1)
fkey1contents=fkey1list[0]
self.assertTrue(isinstance(fkey1contents,dict))
self.assertTrue(len(fkey1contents)==len(LoadCSVPathTestCase.csvdict))
for (key,value) in fkey1contents.iteritems():
self.assertTrue(LoadCSVPathTestCase.csvdict.has_key(key))
self.assertEqual(LoadCSVPathTestCase.csvdict[key],value)
