def test_connector_init(self):
c1 = Connector2()
c2 = Connector2(path='temp_dir')
actual = [c1.base_address, c2.base_address]
expected = ['', 'temp_dir']
self.assertEqual(actual, expected,
"should take path or use an empty string")
def test_put_result(self):
tfb = tempfile.NamedTemporaryFile(mode='rb')
tf = tempfile.NamedTemporaryFile(mode='r')
cb = Connector2(binary=True)
c = Connector2(binary=False)
actual = [
cb.put(io.BytesIO(b'blah'), name=tfb.name),
cb.put(b'blah', name=tfb.name),
cb.put(42, name=tfb.name),
c.put(io.StringIO('blah'), name=tf.name),
c.put('blah', name=tf.name),
c.put(42, name=tf.name)
]
expected = [True, True, False, True, True, False]
tfb.close()
tf.close()
self.assertListEqual(actual, expected,
'should do two good writes and one failed')
def test_get_binary(self):
tf = tempfile.NamedTemporaryFile(mode='rb')
with open(tf.name, 'wb') as f:
f.write(b'blah')
c = Connector2(binary=True)
actual = c.get(name=tf.name).read()
expected = b'blah'
tf.close()
self.assertEqual(actual, expected, 'should read a temp file')
def test_get_do_read(self):
tf = tempfile.NamedTemporaryFile(mode='r')
with open(tf.name, 'w') as f:
f.write('blah')
c = Connector2(binary=False, do_read=True)
actual = c.get(name=tf.name)
expected = 'blah'
tf.close()
self.assertEqual(actual, expected,
'should reaturn a string instead of a buffer')
def test_put_check_written(self):
tfb = tempfile.NamedTemporaryFile(mode='rb')
tf = tempfile.NamedTemporaryFile(mode='r')
cb = Connector2(binary=True)
c = Connector2(binary=False)
_ = [
cb.put(io.BytesIO(b'blah'), name=tfb.name),
cb.put(b'blah', name=tfb.name),
cb.put(42, name=tfb.name),
c.put(io.StringIO('blah'), name=tf.name),
c.put('blah', name=tf.name),
c.put(42, name=tf.name)
]
with open(tfb.name, 'rb') as f:
resb = f.read()
with open(tf.name, 'r') as f:
res = f.read()
actual = [resb, res]
expected = [b'blah', 'blah']
tfb.close()
tf.close()
self.assertListEqual(actual, expected, 'should over-write blah')
def test_get_bad_name_binary(self):
c = Connector2(binary=True)
actual = c.get(name='non-existing-file').read()
expected = b''
self.assertEqual(actual, expected, 'should return an empty string')