src/fs/test_gnunet_fs_rec.py.in

   1 #!@PYTHON@
   2 #    This file is part of GNUnet.
   3 #    (C) 2010 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 (recursive publishing & download)
  21 import sys
  22 import os
  23 import subprocess
  24 import re
  25 import shutil
  26 import tarfile
  27 import filecmp
  28
  29 srcdir = "../.."
  30 gnunet_pyexpect_dir = os.path.join(srcdir, "contrib/scripts")
  31 if gnunet_pyexpect_dir not in sys.path:
  32     sys.path.append(gnunet_pyexpect_dir)
  33
  34 from gnunet_pyexpect import pexpect
  35 from pydiffer import dcdiff
  36
  37 if os.name == 'posix':
  38     download = './gnunet-download'
  39     gnunetarm = 'gnunet-arm'
  40     publish = './gnunet-publish'
  41     unindex = './gnunet-unindex'
  42     search = './gnunet-search'
  43     directory = './gnunet-directory'
  44 elif os.name == 'nt':
  45     download = './gnunet-download.exe'
  46     gnunetarm = 'gnunet-arm.exe'
  47     publish = './gnunet-publish.exe'
  48     unindex = './gnunet-unindex.exe'
  49     search = './gnunet-search.exe'
  50     directory = './gnunet-directory.exe'
  51
  52 if os.name == "nt":
  53     shutil.rmtree(
  54         os.path.join(os.getenv("TEMP"), "gnunet-test-fs-py-rec"), True
  55     )
  56 else:
  57     shutil.rmtree("/tmp/gnunet-test-fs-py-rec", True)
  58
  59 arm = subprocess.Popen([gnunetarm, '-sq', '-c', 'test_gnunet_fs_rec_data.conf'])
  60 arm.communicate()
  61
  62 # pray that `tar' is in PATH.
  63 # FIXME: Actually we should check for that and output
  64 #        a message if it isn't found.
  65 tar = tarfile.open('test_gnunet_fs_rec_data.tgz')
  66 tar.extractall()
  67 # first, basic publish-search-download run
  68 try:
  69     pub = pexpect()
  70     pub.spawn(
  71         None, [
  72             publish, '-c', 'test_gnunet_fs_rec_data.conf', '-k', 'testdir',
  73             'dir/'
  74         ],
  75         stdout=subprocess.PIPE,
  76         stderr=subprocess.STDOUT
  77     )
  78     # Can't say much for publishing, except that the last one is the toplevel directory
  79     pub.expect("stdout", re.compile(r"Publishing `.+' done\.\r?\n"))
  80     pub.expect(
  81         "stdout",
  82         re.compile(
  83             r"URI is `gnunet://fs/chk/[A-Z0-9]{103}\.[A-Z0-9]{103}\.\d+'\.\r?\n"
  84         )
  85     )
  86     pub.expect("stdout", re.compile(r"Publishing `.+' done\.\r?\n"))
  87     pub.expect(
  88         "stdout",
  89         re.compile(
  90             r"URI is `gnunet://fs/chk/[A-Z0-9]{103}\.[A-Z0-9]{103}\.\d+'\.\r?\n"
  91         )
  92     )
  93     pub.expect("stdout", re.compile(r"Publishing `.+' done\.\r?\n"))
  94     pub.expect(
  95         "stdout",
  96         re.compile(
  97             r"URI is `gnunet://fs/chk/[A-Z0-9]{103}\.[A-Z0-9]{103}\.\d+'\.\r?\n"
  98         )
  99     )
 100     pub.expect("stdout", re.compile(r"Publishing `.+' done\.\r?\n"))
 101     pub.expect(
 102         "stdout",
 103         re.compile(
 104             r"URI is `gnunet://fs/chk/[A-Z0-9]{103}\.[A-Z0-9]{103}\.\d+'\.\r?\n"
 105         )
 106     )
 107     pub.expect("stdout", re.compile(r"Publishing `.+' done\.\r?\n"))
 108     pub.expect(
 109         "stdout",
 110         re.compile(
 111             r"URI is `gnunet://fs/chk/[A-Z0-9]{103}\.[A-Z0-9]{103}\.\d+'\.\r?\n"
 112         )
 113     )
 114     pub.expect("stdout", re.compile(r"Publishing `.+' done\.\r?\n"))
 115     pub.expect(
 116         "stdout",
 117         re.compile(
 118             r"URI is `gnunet://fs/chk/[A-Z0-9]{103}\.[A-Z0-9]{103}\.\d+'\.\r?\n"
 119         )
 120     )
 121     pub.expect(
 122         "stdout", re.compile(r"Publishing `.+[\\/]dir[\\/]' done\.\r?\n")
 123     )
 124     m = pub.expect("stdout", re.compile(r".+\r?\n"))
 125     if not m:
 126         sys.exit(3)
 127     output = m.string
 128     url = output[output.find("`") + 1:output.find("'")]
 129
 130     down = pexpect()
 131     down.spawn(
 132         None, [
 133             download, '-c', 'test_gnunet_fs_rec_data.conf', '-R', '-o',
 134             'rdir.gnd', url
 135         ],
 136         stdout=subprocess.PIPE,
 137         stderr=subprocess.STDOUT
 138     )
 139     down.expect("stdout", re.compile(r"Downloading `rdir.gnd' done (.*).\r?\n"))
 140
 141     d = pexpect()
 142     d.spawn(
 143         None, [directory, '-c', 'test_gnunet_fs_rec_data.conf', 'rdir/a.gnd'],
 144         stdout=subprocess.PIPE,
 145         stderr=subprocess.STDOUT
 146     )
 147     d.expect("stdout", re.compile(r"Directory `a/' meta data:\r?\n"))
 148     d.expect("stdout", re.compile(r"Directory `a/' contents:\r?\n"))
 149     d.expect("stdout", re.compile(r"COPYING (.*)\r?\n"))
 150     d.expect("stdout", re.compile(r"INSTALL (.*)\r?\n"))
 151
 152     os.remove("rdir/b.gnd")
 153     os.remove("rdir/a.gnd")
 154     diff = dcdiff('dir', 'rdir')
 155     if len(diff) != 0:
 156         raise Exception(
 157             "Unexpected difference between source directory and downloaded result:\n{}"
 158             .format(diff)
 159         )
 160
 161 finally:
 162     arm = subprocess.Popen([
 163         gnunetarm, '-eq', '-c', 'test_gnunet_fs_rec_data.conf'
 164     ])
 165     arm.communicate()
 166     if os.name == "nt":
 167         shutil.rmtree(
 168             os.path.join(os.getenv("TEMP"), "gnunet-test-fs-py-rec"), True
 169         )
 170     else:
 171         shutil.rmtree("/tmp/gnunet-test-fs-py-rec", True)
 172     shutil.rmtree("dir", True)
 173     shutil.rmtree("rdir", True)
 174     shutil.rmtree("rdir.gnd", True)