feature_shutdown.py raw

   1  #!/usr/bin/env python3
   2  # Copyright (c) 2018-2020 The Limenka developers
   3  # Distributed under the MIT software license, see the accompanying
   4  # file COPYING or http://www.opensource.org/licenses/mit-license.php.
   5  """Test limenkad shutdown."""
   6  
   7  from test_framework.test_framework import LimenkaTestFramework
   8  from test_framework.util import assert_equal, get_rpc_proxy
   9  from threading import Thread
  10  
  11  def test_long_call(node):
  12      block = node.waitfornewblock()
  13      assert_equal(block['height'], 0)
  14  
  15  class ShutdownTest(LimenkaTestFramework):
  16  
  17      def set_test_params(self):
  18          self.setup_clean_chain = True
  19          self.num_nodes = 1
  20          self.supports_cli = False
  21  
  22      def run_test(self):
  23          node = get_rpc_proxy(self.nodes[0].url, 1, timeout=600, coveragedir=self.nodes[0].coverage_dir)
  24          # Force connection establishment by executing a dummy command.
  25          node.getblockcount()
  26          Thread(target=test_long_call, args=(node,)).start()
  27          # Wait until the server is executing the above `waitfornewblock`.
  28          self.wait_until(lambda: len(self.nodes[0].getrpcinfo()['active_commands']) == 2)
  29          # Wait 1 second after requesting shutdown but not before the `stop` call
  30          # finishes. This is to ensure event loop waits for current connections
  31          # to close.
  32          self.stop_node(0, wait=1000)
  33  
  34  if __name__ == '__main__':
  35      ShutdownTest(__file__).main()
  36