From 25188ca49950267a74b35aab1359bd5d3b919fc7 Mon Sep 17 00:00:00 2001 From: Prashanth Pai Date: Thu, 3 Mar 2016 17:25:48 +0530 Subject: Fix dup fd leak A fd was not being closed after it was duplicated. This code path can be easily hit when doing a GET on a file that needs Etag (md5sum) to be recalculated. Change-Id: Ib2e10d990b9b2e1fa85d0079767892de8c8d4eec Signed-off-by: Prashanth Pai Reviewed-on: http://review.gluster.org/13593 Reviewed-by: Thiago da Silva Tested-by: Thiago da Silva --- test/unit/common/test_utils.py | 16 ++++++++++++++++ 1 file changed, 16 insertions(+) (limited to 'test/unit/common/test_utils.py') diff --git a/test/unit/common/test_utils.py b/test/unit/common/test_utils.py index 5ab6f36..f88229a 100644 --- a/test/unit/common/test_utils.py +++ b/test/unit/common/test_utils.py @@ -439,6 +439,22 @@ class TestUtils(unittest.TestCase): md5.update(chunk) assert hd == md5.hexdigest() + def test_get_etag_dup_fd_closed(self): + fd, path = tempfile.mkstemp() + data = "It's not who we are underneath, but what we do that defines us" + os.write(fd, data) + os.lseek(fd, 0, os.SEEK_SET) + + mock_do_close = Mock() + with patch("gluster.swift.common.utils.do_close", mock_do_close): + etag = utils._get_etag(fd) + self.assertEqual(etag, hashlib.md5(data).hexdigest()) + self.assertTrue(mock_do_close.called) + + # We mocked out close, so we have to close the fd for real + os.close(mock_do_close.call_args[0][0]) + os.close(fd) + def test_get_object_metadata_dne(self): md = utils.get_object_metadata("/tmp/doesNotEx1st") assert md == {} -- cgit