git » chasquid » main » tree

[main] / test / util / mail_diff 

  1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
#!/usr/bin/env python3

import difflib
import email.parser
import email.policy
import itertools
import mailbox
import sys


def flexible_eq(expected, got):
    """Compare two strings, supporting wildcards.

    This functions compares two strings, but supports wildcards on the
    expected string. The following characters have special meaning:

     - ?  matches any character.
     - *  matches anything until the end of the line.

    Returns True if equal (considering wildcards), False otherwise.
    """
    posG = 0
    for c in expected:
        if posG >= len(got):
            return False

        if c == '?':
            posG += 1
            continue
        if c == '*':
            while posG < len(got) and got[posG] != '\t':
                posG += 1
                continue
            continue

        if c != got[posG]:
            return False

        posG += 1

    if posG != len(got):
        # We got more than we expected.
        return False

    return True


def msg_equals(expected, msg):
	"""Compare two messages recursively, using flexible_eq()."""
	diff = False
	for h, val in expected.items():
		if h not in msg:
			print("Header missing: %r" % h)
			diff = True
			continue

		if expected[h] == '*':
			continue

		if not flexible_eq(val, msg[h]):
			print("Header %r differs:" % h)
			print("Exp: %r" % val)
			print("Got: %r" % msg[h])
			diff = True

	if diff:
		return False

	if expected.is_multipart() != msg.is_multipart():
		print("Multipart differs, expected %s, got %s" % (
			expected.is_multipart(), msg.is_multipart()))
		return False

	if expected.is_multipart():
		for exp, got in itertools.zip_longest(expected.get_payload(), msg.get_payload()):
			if not msg_equals(exp, got):
				return False
	else:
		if not flexible_eq(expected.get_payload(), msg.get_payload()):
			exp = expected.get_payload().splitlines()
			got = msg.get_payload().splitlines()
			print("Payload differs:")
			for l in difflib.ndiff(exp, got):
				print(l)
			return False

	return True


if __name__ == "__main__":
	f1, f2 = sys.argv[1:3]

	# We use a custom strict policy to do more strict content validation.
	policy = email.policy.EmailPolicy(
			utf8=True,
			linesep="\r\n",
			refold_source='none',
			raise_on_defect=True)

	expected = email.parser.Parser(policy=policy).parse(open(f1))
	msg = email.parser.Parser(policy=policy).parse(open(f2))

	sys.exit(0 if msg_equals(expected, msg) else 1)