From 5088caa69e0a5f6822bbf6184c404eaf18783ed5 Mon Sep 17 00:00:00 2001
From: Kamen Mazdrashki <kamen.mazdrashki@postpath.com>
Date: Thu, 10 Sep 2009 23:44:00 +0300
Subject: s4/tort: assert macro for drsuapi dcerpc call

The macro actually wraps common code pattern used in
almost every test for DRSUAPI interface
---
 source4/torture/rpc/drsuapi.h | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

(limited to 'source4/torture/rpc/drsuapi.h')

diff --git a/source4/torture/rpc/drsuapi.h b/source4/torture/rpc/drsuapi.h
index e9a7dd5ad1..d32a35ee2c 100644
--- a/source4/torture/rpc/drsuapi.h
+++ b/source4/torture/rpc/drsuapi.h
@@ -23,6 +23,9 @@
 
 #include "librpc/gen_ndr/drsuapi.h"
 
+/**
+ * Data structure common for most of DRSUAPI tests
+ */
 struct DsPrivate {
 	struct dcerpc_pipe *pipe;
 	struct policy_handle bind_handle;
@@ -35,3 +38,26 @@ struct DsPrivate {
 	struct test_join *join;
 };
 
+
+/**
+ * Custom torture macro to check dcerpc_drsuapi_ call
+ * return values printing more friendly messages
+ * \param _tctx torture context
+ * \param _p DCERPC pipe handle
+ * \param _ntstatus NTSTATUS for dcerpc_drsuapi_ call
+ * \param _pr in/out DCEPRC request structure
+ * \param _msg error message prefix
+ */
+#define torture_drsuapi_assert_call(_tctx, _p, _ntstat, _pr, _msg) \
+	do { \
+		NTSTATUS __nt = _ntstat; \
+		if (!NT_STATUS_IS_OK(__nt)) { \
+			const char *errstr = nt_errstr(__nt); \
+			if (NT_STATUS_EQUAL(__nt, NT_STATUS_NET_WRITE_FAULT)) { \
+				errstr = dcerpc_errstr(_tctx, _p->last_fault_code); \
+			} \
+			torture_fail(tctx, talloc_asprintf(_tctx, "%s failed - %s", _msg, errstr)); \
+		} \
+		torture_assert_werr_ok(_tctx, (_pr)->out.result, _msg); \
+	} while(0)
+
-- 
cgit