upstream/mercurial-mirror Files · contrib/automation/hgautomation/__init__.py

phabricator: add custom vcr matcher to match request bodies...

phabricator: add custom vcr matcher to match request bodies Currently when the phabricator extension's conduit output changes the tests don't notice since the default vcr matcher only matches on 'method' and 'uri', not the body. Add a custom matcher that checks the same params are in the body (ignoring ordering). vcr's in-built body matcher can't be used since it fails under py3 with a "UnicodeEncodeError" on the "€ in commit message" tests. The DREV ids have decreased since the recordings were generated against a different phabricator instance to avoid spamming mercurial-devel. Differential Revision: https://phab.mercurial-scm.org/D6347

Gregory Szorc - - Load All Authors

File last commit:

r42191:b05a3e28 default


                r42454:af13e208

default

Download file

             __init__.py
        
                    59 lines
            
             | 1.4 KiB
            
                | text/x-python
            
             |
                PythonLexer
            
             / contrib / automation / hgautomation / __init__.py
          
                    History
                
                 |
                  Annotation
                 | Raw
                 |Copy content
                 |Copy permalink

      # __init__.py - High-level automation interfaces

      #

      # Copyright 2019 Gregory Szorc <gregory.szorc@gmail.com>

      #

      # This software may be used and distributed according to the terms of the

      # GNU General Public License version 2 or any later version.

      # no-check-code because Python 3 native.

      import pathlib

      import secrets

      from .aws import (

          AWSConnection,

      )

      class HGAutomation:

          """High-level interface for Mercurial automation.

          Holds global state, provides access to other primitives, etc.

          """

          def __init__(self, state_path: pathlib.Path):

              self.state_path = state_path

              state_path.mkdir(exist_ok=True)

          def default_password(self):

              """Obtain the default password to use for remote machines.

              A new password will be generated if one is not stored.

              """

              p = self.state_path / 'default-password'

              try:

                  with p.open('r', encoding='ascii') as fh:

                      data = fh.read().strip()

                      if data:

                          return data

              except FileNotFoundError:

                  pass

              password = secrets.token_urlsafe(24)

              with p.open('w', encoding='ascii') as fh:

                  fh.write(password)

                  fh.write('\n')

              p.chmod(0o0600)

              return password

          def aws_connection(self, region: str):

              """Obtain an AWSConnection instance bound to a specific region."""

              return AWSConnection(self, region)

	Site-wide shortcuts
/	Use quick search box
g h	Goto home page
g g	Goto my private gists page
g G	Goto my public gists page
g 0-9	Goto bookmarked items from 0-9
n r	New repository page
n g	New gist page

	Repositories
g s	Goto summary page
g c	Goto changelog page
g f	Goto files page
g F	Goto files page with file search activated
g p	Goto pull requests page
g o	Goto repository settings
g O	Goto repository access permissions settings
t s	Toggle sidebar on some pages

				# __init__.py - High-level automation interfaces
				#
				# Copyright 2019 Gregory Szorc <gregory.szorc@gmail.com>
				#
				# This software may be used and distributed according to the terms of the
				# GNU General Public License version 2 or any later version.

				# no-check-code because Python 3 native.

				import pathlib
				import secrets

				from .aws import (
				AWSConnection,
				)


				class HGAutomation:
				"""High-level interface for Mercurial automation.

				Holds global state, provides access to other primitives, etc.
				"""

				def __init__(self, state_path: pathlib.Path):
				self.state_path = state_path

				state_path.mkdir(exist_ok=True)

				def default_password(self):
				"""Obtain the default password to use for remote machines.

				A new password will be generated if one is not stored.
				"""
				p = self.state_path / 'default-password'

				try:
				with p.open('r', encoding='ascii') as fh:
				data = fh.read().strip()

				if data:
				return data

				except FileNotFoundError:
				pass

				password = secrets.token_urlsafe(24)

				with p.open('w', encoding='ascii') as fh:
				fh.write(password)
				fh.write('\n')

				p.chmod(0o0600)

				return password

				def aws_connection(self, region: str):
				"""Obtain an AWSConnection instance bound to a specific region."""

				return AWSConnection(self, region)