From 8b0f31994c38b9f349fdda88a74e95a961582902 Mon Sep 17 00:00:00 2001 From: Jelmer Vernooij Date: Thu, 21 Jan 2010 16:56:23 +1300 Subject: pyxattr: Simplify tests. --- source4/scripting/python/pyxattr_tdb.c | 7 +- source4/scripting/python/samba/tests/xattr.py | 112 ++++++++++++++------------ 2 files changed, 62 insertions(+), 57 deletions(-) (limited to 'source4/scripting/python') diff --git a/source4/scripting/python/pyxattr_tdb.c b/source4/scripting/python/pyxattr_tdb.c index f90cfd5a26..e49cd887d0 100644 --- a/source4/scripting/python/pyxattr_tdb.c +++ b/source4/scripting/python/pyxattr_tdb.c @@ -33,11 +33,7 @@ static PyObject *py_is_xattr_supported(PyObject *self) { -#if !defined(HAVE_XATTR_SUPPORT) - return Py_False; -#else return Py_True; -#endif } static PyObject *py_wrap_setxattr(PyObject *self, PyObject *args) @@ -63,7 +59,8 @@ static PyObject *py_wrap_setxattr(PyObject *self, PyObject *args) talloc_free(mem_ctx); return NULL; } - status = push_xattr_blob_tdb_raw(eadb,mem_ctx,attribute,filename,-1,&blob); + status = push_xattr_blob_tdb_raw(eadb, mem_ctx, attribute, filename, -1, + &blob); if (!NT_STATUS_IS_OK(status)) { PyErr_FromNTSTATUS(status); talloc_free(mem_ctx); diff --git a/source4/scripting/python/samba/tests/xattr.py b/source4/scripting/python/samba/tests/xattr.py index 7133a2b5b3..7854f84c8a 100644 --- a/source4/scripting/python/samba/tests/xattr.py +++ b/source4/scripting/python/samba/tests/xattr.py @@ -19,77 +19,85 @@ import samba.xattr_native, samba.xattr_tdb from samba.dcerpc import xattr -from samba.ndr import ndr_pack, ndr_unpack -from unittest import TestCase +from samba.ndr import ndr_pack +from testtools import TestCase, TestSkipped import random import os -import tdb -import sys class XattrTests(TestCase): def test_set_xattr_native(self): - if samba.xattr_native.is_xattr_supported(): - random.seed() - path=os.environ['SELFTEST_PREFIX'] - tempf=os.path.join(path,"pytests"+str(int(100000*random.random()))) - ntacl=xattr.NTACL() - ntacl.version = 1 - open(tempf, 'w').write("empty") - try: - samba.xattr_native.wrap_setxattr(tempf,"user.unittests",ndr_pack(ntacl)) - except IOError: - print >>sys.stderr, "WARNING: the filesystem where the tests are runned do not support XATTR, tests SKIPED" - os.unlink(tempf) - else: - print >>sys.stderr, "WARNING: the filesystem where the tests are runned do not support XATTR, tests SKIPED" + if not samba.xattr_native.is_xattr_supported(): + raise TestSkipped() + random.seed() + path = os.environ['SELFTEST_PREFIX'] + tempf = os.path.join(path,"pytests"+str(int(100000*random.random()))) + ntacl = xattr.NTACL() + ntacl.version = 1 + open(tempf, 'w').write("empty") + try: + samba.xattr_native.wrap_setxattr(tempf, "user.unittests", + ndr_pack(ntacl)) + except IOError: + raise TestSkipped("the filesystem where the tests are runned do not support XATTR") + os.unlink(tempf) def test_set_and_get_native(self): - if samba.xattr_native.is_xattr_supported(): - random.seed() - path = os.environ['SELFTEST_PREFIX'] - tempf=os.path.join(path,"pytests"+str(int(100000*random.random()))) - reftxt="this is a test" - open(tempf, 'w').write("empty") - try: - samba.xattr_native.wrap_setxattr(tempf,"user.unittests",reftxt) - text = samba.xattr_native.wrap_getxattr(tempf,"user.unittests") - self.assertEquals(text,reftxt) - except IOError: - print >>sys.stderr,"WARNING: the filesystem where the tests are runned do not support XATTR, tests SKIPED" - os.unlink(tempf) - else: - print >>sys.stderr,"WARNING: the filesystem where the tests are runned do not support XATTR, tests SKIPED" + if not samba.xattr_native.is_xattr_supported(): + raise TestSkipped() + random.seed() + path = os.environ['SELFTEST_PREFIX'] + tempf = os.path.join(path,"pytests"+str(int(100000*random.random()))) + reftxt = "this is a test" + open(tempf, 'w').write("empty") + try: + samba.xattr_native.wrap_setxattr(tempf, "user.unittests", reftxt) + text = samba.xattr_native.wrap_getxattr(tempf, "user.unittests") + self.assertEquals(text, reftxt) + except IOError: + raise TestSkipped("the filesystem where the tests are runned do not support XATTR") + os.unlink(tempf) def test_set_xattr_tdb(self): - path=os.environ['SELFTEST_PREFIX'] + path = os.environ['SELFTEST_PREFIX'] random.seed() - tempf=os.path.join(path,"pytests"+str(int(100000*random.random()))) - ntacl=xattr.NTACL() + tempf = os.path.join(path, "pytests"+str(int(100000*random.random()))) + ntacl = xattr.NTACL() ntacl.version = 1 open(tempf, 'w').write("empty") - samba.xattr_tdb.wrap_setxattr(os.path.join(path,"eadb.tdb"),tempf,"user.unittests",ndr_pack(ntacl)) - os.unlink(tempf) - os.unlink(os.path.join(path,"eadb.tdb")) + try: + samba.xattr_tdb.wrap_setxattr(os.path.join(path, "eadb.tdb"), + tempf, "user.unittests", ndr_pack(ntacl)) + finally: + os.unlink(tempf) + os.unlink(os.path.join(path, "eadb.tdb")) def test_set_tdb_not_open(self): - path=os.environ['SELFTEST_PREFIX'] + path = os.environ['SELFTEST_PREFIX'] random.seed() - tempf=os.path.join(path,"pytests"+str(int(100000*random.random()))) - ntacl=xattr.NTACL() + tempf = os.path.join(path, "pytests"+str(int(100000*random.random()))) + ntacl = xattr.NTACL() ntacl.version = 1 open(tempf, 'w').write("empty") - self.assertRaises(IOError,samba.xattr_tdb.wrap_setxattr,os.path.join(path,os.path.join("nonexistent","eadb.tdb")),tempf,"user.unittests",ndr_pack(ntacl)) - os.unlink(tempf) + try: + self.assertRaises(IOError, samba.xattr_tdb.wrap_setxattr, + os.path.join(path, "nonexistent","eadb.tdb"), tempf, + "user.unittests", ndr_pack(ntacl)) + finally: + os.unlink(tempf) def test_set_and_get_tdb(self): - path=os.environ['SELFTEST_PREFIX'] + path = os.environ['SELFTEST_PREFIX'] random.seed() - tempf=os.path.join(path,"pytests"+str(int(100000*random.random()))) - reftxt="this is a test" + tempf = os.path.join(path, "pytests"+str(int(100000*random.random()))) + reftxt = "this is a test" open(tempf, 'w').write("empty") - samba.xattr_tdb.wrap_setxattr(os.path.join(path,"eadb.tdb"),tempf,"user.unittests",reftxt) - text = samba.xattr_tdb.wrap_getxattr(os.path.join(path,"eadb.tdb"),tempf,"user.unittests") - self.assertEquals(text,reftxt) - os.unlink(tempf) - os.unlink(os.path.join(path,"eadb.tdb")) + try: + samba.xattr_tdb.wrap_setxattr(os.path.join(path, "eadb.tdb"), + tempf, "user.unittests", reftxt) + text = samba.xattr_tdb.wrap_getxattr( + os.path.join(path, "eadb.tdb"), tempf, "user.unittests") + self.assertEquals(text, reftxt) + finally: + os.unlink(tempf) + os.unlink(os.path.join(path, "eadb.tdb")) -- cgit