Format fdbcli_tests.py file

bindings/python/tests/fdbcli_tests.py

@@ -9,6 +9,7 @@ import json
 import time
 import random
 
+
 def enable_logging(level=logging.ERROR):
     """Enable logging in the function with the specified logging level
 
@@ -33,6 +34,7 @@ def enable_logging(level=logging.ERROR):
         return wrapper
     return func_decorator
 
+
 def run_fdbcli_command(*args):
     """run the fdbcli statement: fdbcli --exec '<arg1> <arg2> ... <argN>'.
 
@@ -42,6 +44,7 @@ def run_fdbcli_command(*args):
     commands = command_template + ["{}".format(' '.join(args))]
     return subprocess.run(commands, stdout=subprocess.PIPE).stdout.decode('utf-8').strip()
 
+
 def run_fdbcli_command_and_get_error(*args):
     """run the fdbcli statement: fdbcli --exec '<arg1> <arg2> ... <argN>'.
 
@@ -51,6 +54,7 @@ def run_fdbcli_command_and_get_error(*args):
     commands = command_template + ["{}".format(' '.join(args))]
     return subprocess.run(commands, stdout=subprocess.PIPE, stderr=subprocess.PIPE).stderr.decode('utf-8').strip()
 
+
 @enable_logging()
 def advanceversion(logger):
     # get current read version
@@ -73,6 +77,7 @@ def advanceversion(logger):
     logger.debug("Read version: {}".format(version4))
     assert version4 >= version3
 
+
 @enable_logging()
 def maintenance(logger):
     # expected fdbcli output when running 'maintenance' while there's no ongoing maintenance
@@ -95,6 +100,7 @@ def maintenance(logger):
     output3 = run_fdbcli_command('maintenance')
     assert output3 == no_maintenance_output
 
+
 @enable_logging()
 def setclass(logger):
     output1 = run_fdbcli_command('setclass')
@@ -135,6 +141,7 @@ def setclass(logger):
     logger.debug(class_type_line_3)
     assert class_type_line_3 == class_type_line_1
 
+
 @enable_logging()
 def lockAndUnlock(logger):
     # lock an unlocked database, should be successful
@@ -160,6 +167,7 @@ def lockAndUnlock(logger):
     assert output3.decode('utf-8').strip() == 'Database unlocked.'
     assert not get_value_from_status_json(True, 'cluster', 'database_lock_state', 'locked')
 
+
 @enable_logging()
 def kill(logger):
     output1 = run_fdbcli_command('kill')
@@ -182,6 +190,7 @@ def kill(logger):
     logger.debug("Old: {}, New: {}".format(old_generation, new_generation))
     assert new_generation > old_generation
 
+
 @enable_logging()
 def suspend(logger):
     output1 = run_fdbcli_command('suspend')
@@ -222,6 +231,7 @@ def suspend(logger):
     # at most after 60 seconds, the cluster should be available
     assert get_value_from_status_json(False, 'client', 'database_status', 'available')
 
+
 def get_value_from_status_json(retry, *args):
     while True:
         result = json.loads(run_fdbcli_command('status', 'json'))
@@ -233,6 +243,7 @@ def get_value_from_status_json(retry, *args):
 
     return result
 
+
 @enable_logging()
 def consistencycheck(logger):
     consistency_check_on_output = 'ConsistencyCheck is on'
@@ -246,6 +257,7 @@ def consistencycheck(logger):
     output3 = run_fdbcli_command('consistencycheck')
     assert output3 == consistency_check_on_output
 
+
 @enable_logging()
 def cache_range(logger):
     # this command is currently experimental
@@ -253,6 +265,7 @@ def cache_range(logger):
     run_fdbcli_command('cache_range', 'set', 'a', 'b')
     run_fdbcli_command('cache_range', 'clear', 'a', 'b')
 
+
 @enable_logging()
 def datadistribution(logger):
     output1 = run_fdbcli_command('datadistribution', 'off')
@@ -272,6 +285,7 @@ def datadistribution(logger):
     assert output6 == 'Data distribution is enabled for rebalance.'
     time.sleep(1)
 
+
 @enable_logging()
 def transaction(logger):
     """This test will cover the transaction related fdbcli commands.
@@ -333,6 +347,7 @@ def transaction(logger):
     output7 = run_fdbcli_command('get', 'key')
     assert output7 == "`key': not found"
 
+
 def get_fdb_process_addresses(logger):
     # get all processes' network addresses
     output = run_fdbcli_command('kill')
@@ -342,6 +357,7 @@ def get_fdb_process_addresses(logger):
     assert len(addresses) == process_number
     return addresses
 
+
 @enable_logging(logging.DEBUG)
 def coordinators(logger):
     # we should only have one coordinator for now
@@ -369,6 +385,7 @@ def coordinators(logger):
     assert len(get_value_from_status_json(True, 'client', 'coordinators', 'coordinators')) == 1
     wait_for_database_available(logger)
 
+
 @enable_logging(logging.DEBUG)
 def exclude(logger):
     # get all processes' network addresses
@@ -418,6 +435,8 @@ def exclude(logger):
     wait_for_database_available(logger)
 
 # read the system key 'k', need to enable the option first
+
+
 def read_system_key(k):
     output = run_fdbcli_command('option', 'on', 'READ_SYSTEM_KEYS;', 'get', k)
     if 'is' not in output:
@@ -426,6 +445,7 @@ def read_system_key(k):
     _, value = output.split(' is ')
     return value
 
+
 @enable_logging()
 def throttle(logger):
     # no throttled tags at the beginning
@@ -443,6 +463,7 @@ def throttle(logger):
     assert enable_flag == "`0'"
     # TODO : test manual throttling, not easy to do now
 
+
 def wait_for_database_available(logger):
     # sometimes the change takes some time to have effect and the database can be unavailable at that time
     # this is to wait until the database is available again
@@ -450,9 +471,11 @@ def wait_for_database_available(logger):
         logger.debug("Database unavailable for now, wait for one second")
         time.sleep(1)
 
+
 if __name__ == '__main__':
     # fdbcli_tests.py <path_to_fdbcli_binary> <path_to_fdb_cluster_file> <process_number> [external_client_library_path]
-    assert len(sys.argv) == 4 or len(sys.argv) == 5, "Please pass arguments: <path_to_fdbcli_binary> <path_to_fdb_cluster_file> <process_number> [external_client_library_path]"
+    assert len(sys.argv) == 4 or len(
+        sys.argv) == 5, "Please pass arguments: <path_to_fdbcli_binary> <path_to_fdb_cluster_file> <process_number> [external_client_library_path]"
     # set external client library
     if len(sys.argv) == 5:
         external_client_library_path = sys.argv[4]
@@ -481,5 +504,3 @@ if __name__ == '__main__':
         assert process_number > 1, "Process number should be positive"
         coordinators()
         exclude()
-
-