Skip to content

Commit 64ee50d

Browse files
committed
Fixing byte writing for unicode and python 3 happiness
1 parent 9b4db58 commit 64ee50d

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

test/platform_database.py

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -39,13 +39,13 @@ class EmptyPlatformDatabaseTests(unittest.TestCase):
3939

4040
def setUp(self):
4141
self.base_db_path = os.path.join(tempfile.mkdtemp(), 'base')
42-
self.base_db = open(self.base_db_path, 'w+')
42+
self.base_db = open(self.base_db_path, 'w+b')
4343
self.base_db.write(b'{}')
4444
self.base_db.seek(0)
4545
self.pdb = PlatformDatabase([self.base_db_path])
4646

4747
def tearDown(self):
48-
pass
48+
self.base_db.close()
4949

5050
def test_broken_database_io(self):
5151
"""Verify that the platform database still works without a
@@ -61,7 +61,7 @@ def test_broken_database_bad_json(self):
6161
"""Verify that the platform database still works without a
6262
working backing file
6363
"""
64-
self.base_db.write(b'{{}')
64+
self.base_db.write(b'{}')
6565
self.base_db.seek(0)
6666
self.pdb = PlatformDatabase([self.base_db_path])
6767
self.pdb.add("1234", "MYTARGET")
@@ -122,12 +122,12 @@ class OverriddenPlatformDatabaseTests(unittest.TestCase):
122122
def setUp(self):
123123
self.temp_dir = tempfile.mkdtemp()
124124
self.base_db_path = os.path.join(self.temp_dir, 'base')
125-
self.base_db = open(self.base_db_path, 'w+')
125+
self.base_db = open(self.base_db_path, 'w+b')
126126
self.base_db.write(json.dumps(dict([('0123', 'Base_Platform')])).
127127
encode('utf-8'))
128128
self.base_db.seek(0)
129129
self.overriding_db_path = os.path.join(self.temp_dir, 'overriding')
130-
self.overriding_db = open(self.overriding_db_path, 'w+')
130+
self.overriding_db = open(self.overriding_db_path, 'w+b')
131131
self.overriding_db.write(b'{}')
132132
self.overriding_db.seek(0)
133133
self.pdb = PlatformDatabase([self.overriding_db_path, self.base_db_path],
@@ -136,7 +136,8 @@ def setUp(self):
136136
self.overriding_db.seek(0)
137137

138138
def tearDown(self):
139-
pass
139+
self.base_db.close()
140+
self.overriding_db.close()
140141

141142
def assertBaseUnchanged(self):
142143
"""Assert that the base database has not changed
@@ -237,14 +238,14 @@ def setUp(self):
237238
self.release = self.mocked_lock.return_value.release
238239

239240
self.base_db_path = os.path.join(tempfile.mkdtemp(), 'base')
240-
self.base_db = open(self.base_db_path, 'w+')
241+
self.base_db = open(self.base_db_path, 'w+b')
241242
self.base_db.write(b'{}')
242243
self.base_db.seek(0)
243244
self.pdb = PlatformDatabase([self.base_db_path])
244245
self.addCleanup(patch.stopall)
245246

246247
def tearDown(self):
247-
pass
248+
self.base_db.close()
248249

249250
def test_no_update(self):
250251
"""Test that no locks are used when no modifications are specified

0 commit comments

Comments
 (0)