src/fs/test_gnunet_fs_psd.py.in

   1 #!@PYTHON@
   2 #    This file is part of GNUnet.
   3 #    (C) 2010, 2018 Christian Grothoff (and other contributing authors)
   4 #
   5 #    GNUnet is free software: you can redistribute it and/or modify it
   6 #    under the terms of the GNU Affero General Public License as published
   7 #    by the Free Software Foundation, either version 3 of the License,
   8 #    or (at your option) any later version.
   9 #
  10 #    GNUnet is distributed in the hope that it will be useful, but
  11 #    WITHOUT ANY WARRANTY; without even the implied warranty of
  12 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13 #    Affero General Public License for more details.
  14 #
  15 #    You should have received a copy of the GNU Affero General Public License
  16 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 #
  18 #    SPDX-License-Identifier: AGPL3.0-or-later
  19 #
  20 # Testcase for file-sharing command-line tools (publish, search, download)
  21
  22 import sys
  23 import os
  24 import subprocess
  25 import re
  26 import shutil
  27 try:
  28     # Python 2.7
  29     reload
  30 except NameError:
  31     try:
  32         # Python 3.4+:
  33         from importlib import reload
  34     except ImportError:
  35         # Python 3.0 - 3.3
  36         from imp import reload
  37
  38 reload(sys)
  39
  40 # Force encoding to utf-8, as this test otherwise fails
  41 # on some systems (see #5094). In Python 3+ there is no attribute
  42 # sys.setdefaultencoding anymore.
  43 if (3 < sys.version_info[0]):
  44     sys.setdefaultencoding('utf8')
  45
  46 srcdir = "../.."
  47 gnunet_pyexpect_dir = os.path.join(srcdir, "contrib/scripts")
  48 if gnunet_pyexpect_dir not in sys.path:
  49     sys.path.append(gnunet_pyexpect_dir)
  50
  51 from gnunet_pyexpect import pexpect
  52
  53 if os.name == 'posix':
  54     download = './gnunet-download'
  55     gnunetarm = 'gnunet-arm'
  56     publish = './gnunet-publish'
  57     unindex = './gnunet-unindex'
  58     search = './gnunet-search'
  59 elif os.name == 'nt':
  60     download = './gnunet-download.exe'
  61     gnunetarm = 'gnunet-arm.exe'
  62     publish = './gnunet-publish.exe'
  63     unindex = './gnunet-unindex.exe'
  64     search = './gnunet-search.exe'
  65
  66 if "GNUNET_PREFIX" in os.environ:
  67     pass
  68 else:
  69     print("You need to export GNUNET_PREFIX")
  70     sys.exit(1)
  71
  72 if os.name == "nt":
  73     shutil.rmtree(
  74         os.path.join(os.getenv("TEMP"), "gnunet-test-fs-py-psd"), True
  75     )
  76 else:
  77     shutil.rmtree("/tmp/gnunet-test-fs-py-psd", True)
  78
  79 arm = subprocess.Popen([gnunetarm, '-sq', '-c', 'test_gnunet_fs_psd_data.conf'])
  80 arm.communicate()
  81
  82 # first, basic publish-search-download run
  83 try:
  84     pub = pexpect()
  85     pub.spawn(
  86         None, [
  87             publish, '-c', 'test_gnunet_fs_psd_data.conf', '-n', '-m',
  88             "description:Test archive", '-k', 'tst',
  89             'test_gnunet_fs_rec_data.tgz'
  90         ],
  91         stdout=subprocess.PIPE,
  92         stderr=subprocess.STDOUT
  93     )
  94     pub.expect(
  95         "stdout",
  96         re.compile(r"Publishing `.+test_gnunet_fs_rec_data.tgz' done\.\r?\n")
  97     )
  98     pub.expect(
  99         "stdout",
 100         re.compile(
 101             r"URI is `gnunet://fs/chk/2ZMHKPV74CB6GB1GFKQRR95BXJQA2SER25FN48GAW7WSBPA0GDEM5Y74V1ZJHM0NA6919TVW376BHTFDRE3RYS0KRY92M1QJVKPHFCR\.49BT3V5C10KA1695JF71FCT8ZZG4JMJSH04BD9CT22R6KEM915A7CEST17RD0QYTHXV5M4HHEGJMEZSFRDB7JAYC0EMJAN2V781E9DG\.17822'\.\r?\n"
 102         )
 103     )
 104
 105     s = pexpect()
 106     s.spawn(
 107         None, [
 108             search, '-V', '-t', '1000 ms', '-N', '1', '-c',
 109             'test_gnunet_fs_psd_data.conf', 'tst'
 110         ],
 111         stdout=subprocess.PIPE,
 112         stderr=subprocess.STDOUT
 113     )
 114     s.expect(
 115         "stdout",
 116         re.compile(
 117             r'gnunet-download -o "test_gnunet_fs_rec_data.tgz" gnunet://fs/chk/2ZMHKPV74CB6GB1GFKQRR95BXJQA2SER25FN48GAW7WSBPA0GDEM5Y74V1ZJHM0NA6919TVW376BHTFDRE3RYS0KRY92M1QJVKPHFCR\.49BT3V5C10KA1695JF71FCT8ZZG4JMJSH04BD9CT22R6KEM915A7CEST17RD0QYTHXV5M4HHEGJMEZSFRDB7JAYC0EMJAN2V781E9DG\.17822\r?\n'
 118         )
 119     )
 120
 121     down = pexpect()
 122     down.spawn(
 123         None, [
 124             download, '-c', 'test_gnunet_fs_psd_data.conf', '-o',
 125             'test_gnunet_fs_rec_data.tar.gz',
 126             'gnunet://fs/chk/2ZMHKPV74CB6GB1GFKQRR95BXJQA2SER25FN48GAW7WSBPA0GDEM5Y74V1ZJHM0NA6919TVW376BHTFDRE3RYS0KRY92M1QJVKPHFCR.49BT3V5C10KA1695JF71FCT8ZZG4JMJSH04BD9CT22R6KEM915A7CEST17RD0QYTHXV5M4HHEGJMEZSFRDB7JAYC0EMJAN2V781E9DG.17822'
 127         ],
 128         stdout=subprocess.PIPE,
 129         stderr=subprocess.STDOUT
 130     )
 131     down.expect(
 132         "stdout",
 133         re.compile(
 134             r"Downloading `test_gnunet_fs_rec_data.tar.gz' done (.*).\r?\n"
 135         )
 136     )
 137     os.remove("test_gnunet_fs_rec_data.tar.gz")
 138
 139 finally:
 140     arm = subprocess.Popen([
 141         gnunetarm, '-eq', '-c', 'test_gnunet_fs_psd_data.conf'
 142     ])
 143     arm.communicate()
 144     if os.name == "nt":
 145         shutil.rmtree(
 146             os.path.join(os.getenv("TEMP"), "gnunet-test-fs-py-psd"), True
 147         )
 148     else:
 149         shutil.rmtree("/tmp/gnunet-test-fs-py-psd", True)