view test/cmp_helper.py @ 6571:cd408eb748dd

Add small utility script for dumping dbm based databases. Use it to dump dbm based files like db/sessions (records session keys) or db/otks (records one time key for password recovery, data for rate limiting logins and rest interface and other short duration data).
author John Rouillard <rouilj@ieee.org>
date Thu, 23 Dec 2021 02:20:53 -0500
parents 19bd4b413ed6
children
line wrap: on
line source

class StringFragmentCmpHelper:
    def compareStringFragments(self, s, fragments):
        """Compare a string agains a list of fragments where a tuple denotes a
        set of alternatives
        """
        pos = 0
        for frag in fragments:
            if type(frag) != tuple:
                self.assertEqual(s[pos:pos + len(frag)], frag)
                pos += len(frag)
            else:
                found = False
                for alt in frag:
                    if s[pos:pos + len(alt)] == alt:
                        pos += len(alt)
                        found = True
                        break

                if not found:
                    l = max(map(len, frag))
                    raise AssertionError('%s != %s' %
                                         (repr(s[pos:pos + l]), str(frag)))
        self.assertEqual(s[pos:], '')

Roundup Issue Tracker: http://roundup-tracker.org/