class WeedFSTests(unittest.TestCase):
def setUp(self):
self.weed = WeedFS()
pass
def test_repr(self):
self.assertEqual(str(self.weed), "<WeedFS localhost:9333>")
def test_exception(self):
with HTTMock(response_content):
with self.assertRaises(ValueError):
self.weed.upload_file(stream=None, name="test.py")
with self.assertRaises(ValueError):
self.weed.upload_file(stream=open(__file__, "rb"))
class WeedFSTests(unittest.TestCase):
def setUp(self):
self.weed = WeedFS()
pass
def test_repr(self):
self.assertEqual(str(self.weed), "<WeedFS localhost:9333>")
def test_exception(self):
with HTTMock(response_content):
with self.assertRaises(ValueError):
self.weed.upload_file(stream=None, name="test.py")
with self.assertRaises(ValueError):
self.weed.upload_file(stream=open(__file__, "rb"))
class FunctionalTestsSession(unittest.TestCase):
def setUp(self):
self.weed = WeedFS(use_session=True)
def test_head_file(self):
_file = os.path.join(os.path.dirname(__file__), "../tox.ini")
print(_file)
fid = self.weed.upload_file(_file)
self.assertIsNotNone(fid)
res = self.weed.get_file_size(fid)
self.assertEqual(res, os.path.getsize(_file))
res = self.weed.delete_file(fid)
self.assertTrue(res)
res = self.weed.get_file_size("3,123456790")
self.assertIsNone(res)
def test_upload_delete(self):
fid = self.weed.upload_file(__file__)
self.assertIsNotNone(fid)
res = self.weed.delete_file(fid)
self.assertTrue(res)
def test_version(self):
ver = self.weed.version
self.assertIsNotNone(ver)
def test_exists(self):
fid = self.weed.upload_file(__file__)
self.assertTrue(self.weed.file_exists(fid))
res = self.weed.delete_file(fid)
self.assertTrue(res)
self.assertFalse(self.weed.file_exists(fid))
def test_upload_stream(self):
with open(__file__, "rb") as stream:
fid = self.weed.upload_file(stream=stream, name="test.py")
self.assertIsNotNone(fid)
res = self.weed.delete_file(fid)
self.assertTrue(res)
# Test vacuum generated problems with Weed-FS on windows.
# TODO: Investigate
# def test_vacuum(self):
# res = self.weed.vacuum()
# self.assertTrue(res)
def test_bad_fid(self):
self.assertRaises(BadFidFormat, self.weed.get_file_url, ("a"))
def test_get_file(self):
fid = self.weed.upload_file(__file__)
self.assertIsNotNone(fid)
file_content = self.weed.get_file(fid)
self.assertIsNotNone(file_content)
with open(__file__, "rb") as f:
content = f.read()
self.assertEqual(content, file_content)
res = self.weed.delete_file(fid)
self.assertTrue(res)
def test_get_wrong_file(self):
file_content = self.weed.get_file("3,123456790")
self.assertIsNone(file_content)
class FunctionalTestsSession(unittest.TestCase):
def setUp(self):
self.weed = WeedFS(use_session=True)
def test_head_file(self):
_file = os.path.join(os.path.dirname(__file__), "../tox.ini")
print(_file)
fid = self.weed.upload_file(_file)
self.assertIsNotNone(fid)
res = self.weed.get_file_size(fid)
self.assertEqual(res, os.path.getsize(_file))
res = self.weed.delete_file(fid)
self.assertTrue(res)
res = self.weed.get_file_size("3,123456790")
self.assertIsNone(res)
def test_upload_delete(self):
fid = self.weed.upload_file(__file__)
self.assertIsNotNone(fid)
res = self.weed.delete_file(fid)
self.assertTrue(res)
def test_version(self):
ver = self.weed.version
self.assertIsNotNone(ver)
def test_exists(self):
fid = self.weed.upload_file(__file__)
self.assertTrue(self.weed.file_exists(fid))
res = self.weed.delete_file(fid)
self.assertTrue(res)
self.assertFalse(self.weed.file_exists(fid))
def test_upload_stream(self):
with open(__file__, "rb") as stream:
fid = self.weed.upload_file(stream=stream, name="test.py")
self.assertIsNotNone(fid)
res = self.weed.delete_file(fid)
self.assertTrue(res)
# Test vacuum generated problems with Weed-FS on windows.
# TODO: Investigate
# def test_vacuum(self):
# res = self.weed.vacuum()
# self.assertTrue(res)
def test_bad_fid(self):
self.assertRaises(BadFidFormat, self.weed.get_file_url, ("a"))
def test_get_file(self):
fid = self.weed.upload_file(__file__)
self.assertIsNotNone(fid)
file_content = self.weed.get_file(fid)
self.assertIsNotNone(file_content)
with open(__file__, "rb") as f:
content = f.read()
self.assertEqual(content, file_content)
res = self.weed.delete_file(fid)
self.assertTrue(res)
def test_get_wrong_file(self):
file_content = self.weed.get_file("3,123456790")
self.assertIsNone(file_content)