pestctrl/newfol
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

testdatabase: use with statement to open files.

Browse source 
Signed-off-by: brian m. carlson <sandals@crustytoothpaste.net>
This commit is contained in:
brian m. carlson 2014-07-18 23:07:21 +00:00
parent 72e4372532
commit 74ab2d9fc7
No known key found for this signature in database
GPG key ID: BF535D811F52F68B
1 changed files with 11 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
test/testdatabase.py
View file

@ -62,9 +62,8 @@ class TestDatabaseAccessors(unittest.TestCase):
class TestDatabaseIntegrity(unittest.TestCase):
def create_temp_db(self):
ddir = tempfile.TemporaryDirectory()
fp = open(ddir.name + "/schema", "w")
fp.write("fmt:0:newfol schema file:\ntxn:git\n")
fp.close()
with open(ddir.name + "/schema", "w") as fp:
fp.write("fmt:0:newfol schema file:\ntxn:git\n")
db = Database.load(ddir.name)
return (ddir, db)
@ -107,9 +106,8 @@ class TestDatabaseIntegrity(unittest.TestCase):
class TestDatabaseUpgrades(unittest.TestCase):
def create_temp_db(self):
ddir = tempfile.TemporaryDirectory()
fp = open(ddir.name + "/schema", "w")
fp.write("fmt:0:newfol schema file:\ntxn:git\n")
fp.close()
with open(ddir.name + "/schema", "w") as fp:
fp.write("fmt:0:newfol schema file:\ntxn:git\n")
db = Database.load(ddir.name)
return (ddir, db)
@ -123,10 +121,9 @@ class TestDatabaseUpgrades(unittest.TestCase):
db.upgrade(version=version)
self.assertEqual(Database.read_version(ddir.name),
version)
fp = open(ddir.name + "/dtb", "rb")
data = fp.read(len(pattern))
self.assertEqual(data, pattern)
fp.close()
with open(ddir.name + "/dtb", "rb") as fp:
data = fp.read(len(pattern))
self.assertEqual(data, pattern)
ddir.cleanup()
def test_upgrade_to_xz(self):
@ -145,9 +142,8 @@ class TestDatabaseUpgrades(unittest.TestCase):
class TestMultipleTransactions(unittest.TestCase):
def test_multiple_types(self):
ddir = tempfile.TemporaryDirectory()
fp = open(ddir.name + "/schema", "w")
fp.write("fmt:0:newfol schema file:\ntxn:git:hash\n")
fp.close()
with open(ddir.name + "/schema", "w") as fp:
fp.write("fmt:0:newfol schema file:\ntxn:git:hash\n")
db = Database.load(ddir.name)
db.records()[:] = [Record([1, 2, 3])]
db.store()
@ -159,9 +155,8 @@ class TestMultipleTransactions(unittest.TestCase):
class TestDatabaseFiltering(unittest.TestCase):
def create_temp_db(self):
ddir = tempfile.TemporaryDirectory()
fp = open(ddir.name + "/schema", "w")
fp.write("fmt:0:newfol schema file:\ntxn:git\n")
fp.close()
with open(ddir.name + "/schema", "w") as fp:
fp.write("fmt:0:newfol schema file:\ntxn:git\n")
db = Database.load(ddir.name)
return (ddir, db)