Ejemplo n.º 1
0
def test_disable_registration_for_client_task_only_one_task_in_queue(
        redisdb, client_task_definition_path, client_task_definition_data):
    with open(client_task_definition_path) as request_file:
        request_data = request_file.read()

    node = CommitteeCandidate()
    node.conn = redisdb
    node.task_data = client_task_definition_data
    node.set_task_id(request_data)

    redisdb.lpush(CLIENT_TASK_CHANNEL, request_data)
    node.disable_registration_for_client_task()

    assert redisdb.llen(CLIENT_TASK_CHANNEL) == 0
Ejemplo n.º 2
0
def test_disable_registration_for_client_task():
    node = CommitteeCandidate()
    node.conn = MagicMock()

    node.disable_registration_for_client_task()
    node.conn.lrem.assert_called()