← Back to branch summary

~jelmer/brz-debian/git-import-fixes

« back to all changes in this revision

Viewing changes to tests/__init__.py

  • Committer: Jelmer Vernooij
  • Date: 2018-10-15 07:45:11 UTC
  • Revision ID: jelmer@jelmer.uk-20181015074511-0kwd639wr7y4w5ot
Use context managers.

Show diffs side-by-side

added added

removed removed

Lines of Context:
tests/__init__.py
193
193
        return c
194
194
 
195
195
    def write_changelog(self, changelog, filename):
196
 
        f = open(filename, 'w')
197
 
        changelog.write_to_open_file(f)
198
 
        f.close()
 
196
        with open(filename, 'w') as f:
 
197
            changelog.write_to_open_file(f)
199
198
 
200
199
    def check_tarball_contents(self, tarball, expected, basedir=None,
201
200
                             skip_basedir=False, mode=None):
338
337
            if dirname is not None and dirname != "":
339
338
                if not os.path.exists(os.path.join(basedir, dirname)):
340
339
                    os.makedirs(os.path.join(basedir, dirname))
341
 
            f = open(os.path.join(basedir, path), 'w')
342
 
            try:
 
340
            with open(os.path.join(basedir, path), 'w') as f:
343
341
                if content is None:
344
342
                    content = ''
345
343
                f.write(content)
346
 
            finally:
347
 
                f.close()
348
344
 
349
345
    def _make_symlinks(self, files_list, basedir):
350
346
        for (path, target) in files_list.items():