1
0
Fork 0

Support writing bytes

master
Philip O'Toole 1 year ago
parent e7171b914e
commit 2bd75e38ee

@ -96,7 +96,7 @@ class TestAutoBackupS3(unittest.TestCase):
os.environ['AWS_SECRET_ACCESS_KEY'] = os.environ['RQLITE_S3_SECRET_ACCESS_KEY']
backupData = download_s3_object('rqlite-testing-circleci', path)
backupFile = write_random_file(backupData)
backupFile = write_random_file(backupData, mode='wb')
conn = sqlite3.connect(backupFile)
c = conn.cursor()
c.execute('SELECT * FROM foo')

@ -29,8 +29,8 @@ def random_string(n):
letters = string.ascii_lowercase
return ''.join(random.choice(letters) for i in range(n))
def write_random_file(data):
f = tempfile.NamedTemporaryFile('w', delete=False)
def write_random_file(data, mode='w'):
f = tempfile.NamedTemporaryFile(mode, delete=False)
f.write(data)
f.close()
return f.name

Loading…
Cancel
Save