示例#1
0
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"))
示例#2
0
文件: tests.py 项目: utek/pyseaweed
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"))
示例#3
0
 def setUp(self):
     self.weed = WeedFS(use_session=True)
示例#4
0
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)
示例#5
0
 def setUp(self):
     self.weed = WeedFS()
示例#6
0
 def setUp(self):
     self.weed = WeedFS()
     pass
示例#7
0
 def setUp(self):
     self.weed = WeedFS(use_session=True)
示例#8
0
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)