diff options
Diffstat (limited to 'test')
-rw-r--r-- | test/functional/libgfapi-python-tests.py | 34 | ||||
-rw-r--r-- | test/unit/gluster/test_gfapi.py | 12 |
2 files changed, 43 insertions, 3 deletions
diff --git a/test/functional/libgfapi-python-tests.py b/test/functional/libgfapi-python-tests.py index d0703df..08742f3 100644 --- a/test/functional/libgfapi-python-tests.py +++ b/test/functional/libgfapi-python-tests.py @@ -7,6 +7,36 @@ from nose import SkipTest from gluster import gfapi +class BinFileOpsTest(unittest.TestCase): + + vol = None + path = None + data = None + + @classmethod + def setUpClass(cls): + cls.vol = gfapi.Volume("gfshost", "test") + cls.vol.set_logging("/dev/null", 7) + cls.vol.mount() + + @classmethod + def tearDownClass(cls): + cls.vol = None + + def setUp(self): + self.data = bytearray([(k % 128) for k in range(0, 1024)]) + self.path = self._testMethodName + ".io" + with self.vol.creat(self.path, os.O_WRONLY | os.O_EXCL, 0644) as fd: + fd.write(self.data) + + def test_bin_open_and_read(self): + with self.vol.open(self.path, os.O_RDONLY) as fd: + self.assertTrue(isinstance(fd, gfapi.File)) + buf = fd.read(len(self.data)) + self.assertFalse(isinstance(buf, types.IntType)) + self.assertEqual(buf, self.data) + + class FileOpsTest(unittest.TestCase): vol = None @@ -27,7 +57,7 @@ class FileOpsTest(unittest.TestCase): self.data = loremipsum.get_sentence() self.path = self._testMethodName + ".io" with self.vol.creat(self.path, os.O_WRONLY | os.O_EXCL, 0644) as fd: - rc = fd.write(self.data) + fd.write(self.data) def tearDown(self): self.path = None @@ -38,7 +68,7 @@ class FileOpsTest(unittest.TestCase): self.assertTrue(isinstance(fd, gfapi.File)) buf = fd.read(len(self.data)) self.assertFalse(isinstance(buf, types.IntType)) - self.assertEqual(buf, self.data) + self.assertEqual(buf.value, self.data) def test_lstat(self): sb = self.vol.lstat(self.path) diff --git a/test/unit/gluster/test_gfapi.py b/test/unit/gluster/test_gfapi.py index 6270148..549357b 100644 --- a/test/unit/gluster/test_gfapi.py +++ b/test/unit/gluster/test_gfapi.py @@ -56,7 +56,7 @@ class TestFile(unittest.TestCase): with patch("gluster.gfapi.api.glfs_read", _mock_glfs_read): fd = gfapi.File(2) b = fd.read(5) - self.assertEqual(b, "hello") + self.assertEqual(b.value, "hello") def test_read_fail_exception(self): mock_glfs_read = Mock() @@ -84,6 +84,16 @@ class TestFile(unittest.TestCase): ret = fd.write("hello") self.assertEqual(ret, 5) + def test_write_binary_success(self): + mock_glfs_write = Mock() + mock_glfs_write.return_value = 3 + + with patch("gluster.gfapi.api.glfs_write", mock_glfs_write): + fd = gfapi.File(2) + b = bytearray(3) + ret = fd.write(b) + self.assertEqual(ret, 3) + def test_write_fail_exception(self): mock_glfs_write = Mock() mock_glfs_write.return_value = -1 |