lib: Add TST_ASSERT_INT() and TST_ASSERT_STR()

These functions are meant for asserting that a sysfs or a procfs file
holds the correct value, which is much faster than reading the file,
comapring the value and issuing PASS/FAIL message.

Signed-off-by: Cyril Hrubis <[email protected]>
Signed-off-by: Yang Xu <[email protected]>

Author: metan-ucw

include/tst_assert.h

@@ -0,0 +1,32 @@
+// SPDX-License-Identifier: GPL-2.0-or-later
+/*
+ * Copyright (c) 2020 FUJITSU LIMITED. All rights reserved.
+ * Author: Yang Xu <[email protected]>
+ * Copyright (c) 2020 Cyril Hrubis <[email protected]>
+ */
+#ifndef TST_ASSERT_H__
+#define TST_ASSERT_H__
+
+#define TST_ASSERT_INT(path, val) \
+	tst_assert_int(__FILE__, __LINE__, path, val)
+
+/*
+ * Asserts that integer value stored in file pointed by path equals to the
+ * value passed to this function. This is mostly useful for asserting correct
+ * values in sysfs, procfs, etc.
+ */
+void tst_assert_int(const char *file, const int lineno,
+                    const char *path, int val);
+
+#define TST_ASSERT_STR(path, val) \
+	tst_assert_str(__FILE__, __LINE__, path, val)
+
+/*
+ * Asserts that a string value stored in file pointed by path equals to the
+ * value passed to this function. This is mostly useful for asserting correct
+ * values in sysfs, procfs, etc.
+ */
+void tst_assert_str(const char *file, const int lineno,
+                    const char *path, const char *val);
+
+#endif /* TST_ASSERT_H__ */

include/tst_test.h

@@ -38,6 +38,7 @@
 #include "tst_buffers.h"
 #include "tst_capability.h"
 #include "tst_hugepage.h"
+#include "tst_assert.h"
 
 /*
  * Reports testcase result.

lib/tst_assert.c

@@ -0,0 +1,36 @@
+// SPDX-License-Identifier: GPL-2.0-or-later
+/*
+ * Copyright (c) 2020 FUJITSU LIMITED. All rights reserved.
+ * Author: Yang Xu <[email protected]>
+ * Copyright (c) 2020 Cyril Hrubis <[email protected]>
+ */
+#define TST_NO_DEFAULT_MAIN
+#include "tst_assert.h"
+#include "tst_test.h"
+
+void tst_assert_int(const char *file, const int lineno, const char *path, int val)
+{
+	int sys_val;
+
+	SAFE_FILE_SCANF(path, "%d", &sys_val);
+
+	if (val == sys_val) {
+		tst_res_(file, lineno, TPASS, "%s = %d", path, val);
+		return;
+	}
+
+	tst_res_(file, lineno, TFAIL, "%s != %d got %d", path, val, sys_val);
+}
+
+void tst_assert_str(const char *file, const int lineno, const char *path, const char *val)
+{
+	char sys_val[1024];
+
+	SAFE_FILE_SCANF(path, "%1024s", sys_val);
+	if (!strcmp(val, sys_val)) {
+		tst_res_(file, lineno, TPASS, "%s = '%s'", path, val);
+		return;
+	}
+
+	tst_res_(file, lineno, TFAIL, "%s != '%s' got '%s'", path, val, sys_val);
+}