summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJelmer Vernooij <jelmer@samba.org>2010-01-21 16:56:23 +1300
committerJelmer Vernooij <jelmer@samba.org>2010-01-21 16:56:23 +1300
commit8b0f31994c38b9f349fdda88a74e95a961582902 (patch)
tree5b40e37557a4e5f18a5af35c751ea8cde42975be
parent05b4e29f7f35108a7a17f072dc1329fc14d6b9ea (diff)
downloadsamba-8b0f31994c38b9f349fdda88a74e95a961582902.tar.gz
samba-8b0f31994c38b9f349fdda88a74e95a961582902.tar.bz2
samba-8b0f31994c38b9f349fdda88a74e95a961582902.zip
pyxattr: Simplify tests.
-rw-r--r--source4/scripting/python/pyxattr_tdb.c7
-rw-r--r--source4/scripting/python/samba/tests/xattr.py112
2 files changed, 62 insertions, 57 deletions
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"))