pylibmount: add regression tests

Signed-off-by: Ondrej Oprala <ooprala@redhat.com>
Signed-off-by: Karel Zak <kzak@redhat.com>

libmount/python/test_mount_tab.py

@@ -0,0 +1,162 @@
+#!/bin/python2
+import os
+import sys
+import stat
+import errno
+import functools as ft
+import libmount as mnt
+
+def usage(tss):
+	print "\nUsage:\n\t{:s} <test> [testoptions]\nTests:\n".format(sys.argv[0])
+	for i in tss:
+		print "\t{15:-s}".format(i[0])
+		if i[2] != "":
+			print " {:s}\n".format(i[2])
+
+	print("\n")
+	return 1
+
+def mnt_run_test(tss, argv):
+	rc = -1
+	if ((len(argv) < 2) or (argv[1] == "--help") or (argv[1] == "-h")):
+		return usage(tss)
+
+	#mnt_init_debug(0)
+
+	i=()
+	for i in tss:
+		if i[0] == argv[1]:
+			rc = i[1](i, argv[1:])
+			if rc:
+				print "FAILED [rc={:d}]".format(rc)
+			break
+
+	if ((rc < 0) and (i == ())):
+		return usage(tss)
+	return not not rc #because !!rc is too mainstream for python
+
+def parser_errcb(tb, fname, line):
+	print "{:s}:{:d}: parse error".format(fname, line)
+	return 1
+
+def create_table(f, comments):
+	if not f:
+		return None
+	tb = mnt.Tab()
+	tb.enable_comments(comments)
+	tb.errcb = parser_errcb
+
+	try:
+		tb.parse_file(f)
+	except Exception:
+		print "{:s}: parsing failed".format(f)
+		return None
+	return tb
+
+def test_copy_fs(ts, argv):
+	rc = -1
+	tb = create_table(argv[1], False)
+	fs = tb.find_target("/", mnt.MNT_ITER_FORWARD)
+	if not fs:
+		return rc
+
+	print "ORIGINAL:"
+	fs.print_debug(sys.stdout)
+
+	fs = fs.copy_fs(None)
+	if not fs:
+		return rc
+	print "COPY:"
+	fs.print_debug(sys.stdout)
+	return 0
+
+def test_parse(ts, argv):
+	parse_comments = False
+
+	if len(argv) == 3 and argv[2] == "--comments":
+		parse_comments = True
+	tb = create_table(argv[1], parse_comments)
+
+	if tb.intro_comment:
+		print "Initial comment:\n\"{:s}\"".format(tb.intro_comment)
+	#while ((fs = tb.next_fs()) != None):
+	for fs in iter(ft.partial(tb.next_fs), None):
+		fs.print_debug(sys.stdout)
+	if tb.trailing_comment:
+		print "Trailing comment:\n\"{:s}\"".format(tb.trailing_comment)
+	return 0
+
+def test_find(ts, argv, dr):
+	if len(argv) != 4:
+		print "try --help"
+		return -errno.EINVAL
+
+	f, find, what = argv[1:]
+
+	tb = create_table(f, False)
+	if find.lower() == "source":
+		fs = tb.find_source(what, dr)
+	elif find.lower() == "target":
+		fs = tb.find_target(what, dr)
+
+	if not fs:
+		print "{:s}: not found {:s} '{:s}'".format(f, find, what)
+	else:
+		fs.print_debug(sys.stdout)
+	return 0
+
+def test_find_fw(ts, argv):
+	return test_find(ts, argv, mnt.MNT_ITER_FORWARD)
+
+def test_find_bw(ts, argv):
+	return test_find(ts, argv, mnt.MNT_ITER_BACKWARD)
+
+def test_find_pair(ts, argv):
+	rc = -1
+	tb = create_table(argv[1], False)
+	fs = tb.find_pair(argv[2], argv[3], mnt.MNT_ITER_FORWARD)
+	if not fs:
+		return rc
+	fs.print_debug(sys.stdout)
+	return 0
+
+def test_is_mounted(ts, argv):
+	rc = -1
+	tb = mnt.Tab(path="/proc/self/mountinfo")
+	if not tb:
+		print "failed to parse mountinto"
+		return rc
+
+	fstab = create_table(argv[1], False)
+	if not fstab:
+		return rc
+	fs = ()
+	for fs in ft.iter(tb.next_fs(), -1):
+		if tb.is_fs_mounted(fs):
+			print "{:s} already mounted on {:s}".format(fs.source, fs.target)
+		else:
+			print "{:s} not mounted on {:s}".format(fs.source, fs.target)
+	return 0
+
+def test_find_mountpoint(ts, argv):
+	rc = -1
+	tb = mnt.Tab("/proc/self/mountinfo")
+	if not tb:
+		return rc
+	fs = tb.find_mountpoint(argv[1], mnt.MNT_ITER_BACKWARD)
+	if not fs:
+		return rc
+	fs.print_debug(sys.stdout)
+	return 0
+
+
+tss = (
+	( "--parse",    test_parse,        "<file> [--comments] parse and print tab" ),
+	( "--find-forward",  test_find_fw, "<file> <source|target> <string>" ),
+	( "--find-backward", test_find_bw, "<file> <source|target> <string>" ),
+	( "--find-pair",     test_find_pair, "<file> <source> <target>" ),
+	( "--find-mountpoint", test_find_mountpoint, "<path>" ),
+	( "--copy-fs",       test_copy_fs, "<file>  copy root FS from the file" ),
+	( "--is-mounted",    test_is_mounted, "<fstab> check what from <file> are already mounted" ),
+)
+sys.exit(mnt_run_test(tss, sys.argv))