123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205 |
- #!@PYTHON@
- # This file is part of GNUnet.
- # (C) 2010, 2017 Christian Grothoff (and other contributing authors)
- #
- # GNUnet is free software: you can redistribute it and/or modify it
- # under the terms of the GNU Affero General Public License as published
- # by the Free Software Foundation, either version 3 of the License,
- # or (at your option) any later version.
- #
- # GNUnet is distributed in the hope that it will be useful, but
- # WITHOUT ANY WARRANTY; without even the implied warranty of
- # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
- # Affero General Public License for more details.
- #
- # You should have received a copy of the GNU Affero General Public License
- # along with this program. If not, see <http://www.gnu.org/licenses/>.
- #
- # SPDX-License-Identifier: AGPL3.0-or-later
- #
- #
- import sys
- import signal
- import os
- import subprocess
- import re
- import shutil
- import time
- from gnunet_testing import Peer
- from gnunet_testing import Test
- from gnunet_testing import Check
- from gnunet_testing import Condition
- from gnunet_testing import *
- #
- # This test tests if a fresh peer bootstraps from a hostlist server and then
- # successfully connects to the server. When both peers are connected
- # in transport, core, topology, fs, the server is shutdown
- #
- # Conditions for successful exit:
- # Client peer has 0 connected peer in transport, core, topology, dht, fs
- #definitions
- testname = "test_integration_disconnect"
- verbose = True
- check_timeout = 180
- if os.name == "nt":
- tmp = os.getenv ("TEMP")
- signals = [signal.SIGTERM, signal.SIGINT]
- else:
- tmp = "/tmp"
- signals = [signal.SIGTERM, signal.SIGINT, signal.SIGHUP, signal.SIGQUIT]
- def cleanup_onerror (function, path, excinfo):
- import stat
- if not os.path.exists (path):
- pass
- elif not os.access(path, os.W_OK):
- # Is the error an access error ?
- os.chmod (path, stat.S_IWUSR)
- function (path)
- else:
- raise
- def cleanup ():
- shutil.rmtree (os.path.join (tmp, "c_bootstrap_server"), False, cleanup_onerror)
- shutil.rmtree (os.path.join (tmp, "c_no_nat_client"), False, cleanup_onerror)
- def success_disconnect_cont (check):
- print('Peers disconnected successfully')
- global success
- success = True;
- def fail_disconnect_cont (check):
- global success
- success = False;
- print('Peers failed to disconnect')
- check.evaluate(True)
- def check_disconnect ():
- test.p ('Shutting down bootstrap server')
- server.stop ()
- check = Check (test)
- check.add (StatisticsCondition (client, 'transport', '# peers connected',0))
- check.add (StatisticsCondition (client, 'core', '# peers connected',0))
- check.add (StatisticsCondition (client, 'topology', '# peers connected',0))
- check.add (StatisticsCondition (client, 'dht', '# peers connected',0))
- check.add (StatisticsCondition (client, 'fs', '# peers connected',0))
- check.run_blocking (check_timeout, success_disconnect_cont, fail_disconnect_cont)
- def success_connect_cont (check):
- print('Peers connected successfully')
- check_disconnect ()
- def fail_connect_cont (check):
- global success
- success= False
- print('Peers failed to connected!')
- check.evaluate(True)
- def check_connect ():
- check = Check (test)
- check.add (StatisticsCondition (server, 'transport', '# peers connected',1))
- check.add (StatisticsCondition (server, 'core', '# peers connected',1))
- check.add (StatisticsCondition (server, 'topology', '# peers connected',1))
- check.add (StatisticsCondition (server, 'dht', '# peers connected',1))
- check.add (StatisticsCondition (server, 'fs', '# peers connected',1))
- check.add (StatisticsCondition (client, 'transport', '# peers connected',1))
- check.add (StatisticsCondition (client, 'core', '# peers connected',1))
- check.add (StatisticsCondition (client, 'topology', '# peers connected',1))
- check.add (StatisticsCondition (client, 'dht', '# peers connected',1))
- check.add (StatisticsCondition (client, 'fs', '# peers connected',1))
- check.run_blocking (check_timeout, success_connect_cont, fail_connect_cont)
- #
- # Test execution
- #
- def SigHandler(signum = None, frame = None):
- global success
- global server
- global client
- print('Test was aborted!')
- if (None != server):
- server.stop ()
- if (None != client):
- client.stop ()
- cleanup ()
- sys.exit(success)
- def run ():
- global success
- global test
- global server
- global client
- server = None
- client = None
- success = False
- for sig in signals:
- signal.signal(sig, SigHandler)
- test = Test ('test_integration_bootstrap_and_connect.py', verbose)
- cleanup ()
- server = Peer(test, './confs/c_bootstrap_server.conf');
- client = Peer(test, './confs/c_no_nat_client.conf');
- if (True != server.start()):
- print('Failed to start server')
- if (None != server):
- server.stop ()
- cleanup ()
- sys.exit(success)
- # Give the server time to start
- time.sleep(5)
- if (True != client.start()):
- print('Failed to start client')
- if (None != server):
- server.stop ()
- if (None != client):
- client.stop ()
- cleanup ()
- sys.exit(success)
- if ((client.started == True) and (server.started == True)):
- test.p ('Peers started, running check')
- time.sleep(5)
- check_connect ()
- server.stop ()
- client.stop ()
- cleanup ()
- if (success == False):
- print ('Test failed')
- return False
- else:
- return True
- try:
- run ()
- except (KeyboardInterrupt, SystemExit):
- print('Test interrupted')
- server.stop ()
- client.stop ()
- cleanup ()
- if (success == False):
- sys.exit(1)
- else:
- sys.exit(0)
|