test/test_verbose_output.py

   1 #!/usr/bin/env python3
   2 import os
   3 import subprocess
   4 import sys
   5 import unittest
   6
   7 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
   8
   9 rootDir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
  10
  11
  12 class TestVerboseOutput(unittest.TestCase):
  13     def test_private_info_arg(self):
  14         outp = subprocess.Popen(
  15             [
  16                 sys.executable, 'yt_dlp/__main__.py', '-v',
  17                 '--username', 'johnsmith@gmail.com',
  18                 '--password', 'my_secret_password',
  19             ], cwd=rootDir, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
  20         sout, serr = outp.communicate()
  21         self.assertTrue(b'--username' in serr)
  22         self.assertTrue(b'johnsmith' not in serr)
  23         self.assertTrue(b'--password' in serr)
  24         self.assertTrue(b'my_secret_password' not in serr)
  25
  26     def test_private_info_shortarg(self):
  27         outp = subprocess.Popen(
  28             [
  29                 sys.executable, 'yt_dlp/__main__.py', '-v',
  30                 '-u', 'johnsmith@gmail.com',
  31                 '-p', 'my_secret_password',
  32             ], cwd=rootDir, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
  33         sout, serr = outp.communicate()
  34         self.assertTrue(b'-u' in serr)
  35         self.assertTrue(b'johnsmith' not in serr)
  36         self.assertTrue(b'-p' in serr)
  37         self.assertTrue(b'my_secret_password' not in serr)
  38
  39     def test_private_info_eq(self):
  40         outp = subprocess.Popen(
  41             [
  42                 sys.executable, 'yt_dlp/__main__.py', '-v',
  43                 '--username=johnsmith@gmail.com',
  44                 '--password=my_secret_password',
  45             ], cwd=rootDir, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
  46         sout, serr = outp.communicate()
  47         self.assertTrue(b'--username' in serr)
  48         self.assertTrue(b'johnsmith' not in serr)
  49         self.assertTrue(b'--password' in serr)
  50         self.assertTrue(b'my_secret_password' not in serr)
  51
  52     def test_private_info_shortarg_eq(self):
  53         outp = subprocess.Popen(
  54             [
  55                 sys.executable, 'yt_dlp/__main__.py', '-v',
  56                 '-u=johnsmith@gmail.com',
  57                 '-p=my_secret_password',
  58             ], cwd=rootDir, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
  59         sout, serr = outp.communicate()
  60         self.assertTrue(b'-u' in serr)
  61         self.assertTrue(b'johnsmith' not in serr)
  62         self.assertTrue(b'-p' in serr)
  63         self.assertTrue(b'my_secret_password' not in serr)
  64
  65
  66 if __name__ == '__main__':
  67     unittest.main()