def test_call_connect_with_options_filename(self):
from afs import connect
local_config_file = os.path.join(
get_module_dir(),
'afs-test-config.ini',
)
with connect('localtest', local_config_file) as fs:
self.assertEqual(fs.base, '/tmp')
with connect('smbtest', local_config_file) as fs:
self.assertEqual(fs.username, 'jileon')
self.assertEqual(fs.host, 'nas1')
self.assertEqual(fs.domain, 'parcan.es')
self.assertEqual(fs.service, 'test$')
def test_save_text(self):
with afs.connect('localtmp') as fs:
stream = StringIO('hola, mundo reseña árbol épico '
'ínclito único óbice'.encode('utf-8'))
size = fs.save('testigo.txt', stream)
self.assertTrue(fs.is_file('testigo.txt'))
self.assertEqual(size, 56)
def test_cd(self):
with afs.connect('localtmp') as fs:
fs.cd('uno')
self.assertEqual(fs.cwd(), ['uno'])
self.assertEqual(fs.get_local_path(), '/tmp/uno')
fs.cd('dos')
self.assertEqual(fs.cwd(), ['uno', 'dos'])
self.assertEqual(fs.get_local_path(), '/tmp/uno/dos')
def test_rm(self):
test_filename = 'borrame.txt'
with afs.connect('localtmp') as fs:
self.assertFalse(fs.is_file(test_filename))
fs.save(test_filename, StringIO('Dumb file'.encode('utf-8')))
self.assertTrue(fs.is_file(test_filename))
fs.rm(test_filename)
self.assertFalse(fs.is_file(test_filename))
def test_connect(self):
with afs.connect('localtmp') as fs:
self.assertEqual(fs.name, 'localtmp')
self.assertEqual(fs.base, '/tmp')
self.assertTrue(fs.is_connected)
self.assertEqual(
fs.get_local_path('hola.txt'),
'/tmp/hola.txt',
)
def test_ls(self):
with afs.connect('localtmp') as fs:
found = False
for item in fs.ls():
if item.name == 'token.txt':
self.assertEqual(item.size, 5)
found = True
assert_archive_or_dir(item)
self.assertTrue(found, 'File token.txt not found')
def test_rmdir(self):
test_dir = 'removable_dir'
with afs.connect('localtmp') as fs:
self.assertFalse(fs.is_dir(test_dir),
'Directory exists (It must NOT)')
fs.mkdir(test_dir)
self.assertTrue(fs.is_dir(test_dir),
'Directory doesn\'t exists (and it must)')
fs.rmdir(test_dir)
self.assertFalse(fs.is_dir(test_dir),
'Directory exists (It must NOT)')
def test_try_to_rm_a_dir_fails(self):
'''Triying to remove a directory must fail.
'''
with afs.connect('localtmp') as fs:
fs.mkdir('uno')
self.assertRaises(
AgnosticFileStorageError,
fs.rm,
'uno',
)
fs.rmdir('uno')
def test_save_image_from_pil(self):
from PIL import Image, ImageDraw
size = (256, 256)
im = Image.new("RGB", size, "white")
draw = ImageDraw.Draw(im)
draw.line((0, 0) + im.size, fill=128)
draw.line((0, im.size[1], im.size[0], 0), fill=128)
with afs.connect('localtmp') as fs:
buff = StringIO()
im.save(buff, format="PNG")
buff.seek(0)
fs.save('imagen.png', buff)
def test_save_image(self):
with afs.connect('localtmp') as fs:
size = fs.save('escudo.png',
open('/home/jileon/wwwroot/art/escudo.png', 'rb'))
self.assertTrue(fs.is_file('escudo.png'))
self.assertEqual(size, 3514)
def test_mkdir(self):
with afs.connect('localtmp') as fs:
self.assertFalse(fs.is_dir('test_dir'))
fs.mkdir('test_dir')
self.assertTrue(fs.is_dir('test_dir'))
def test_rm_nonexistent_file(self):
'''Must return False and no error.
'''
with afs.connect('localtmp') as fs:
self.assertFalse(fs.rm('nonexistent_file.txt'))