#
# Copyright 2023-present ScyllaDB
#
# SPDX-License-Identifier: LicenseRef-ScyllaDB-Source-Available-1.0
#

from test.nodetool.rest_api_mock import expected_request


def test_disablebinary(nodetool):
    nodetool("disablebinary", expected_requests=[
        expected_request("DELETE", "/storage_service/native_transport")])


def test_enablebinary(nodetool):
    nodetool("enablebinary", expected_requests=[
        expected_request("POST", "/storage_service/native_transport")])


def test_statusbinary(nodetool):
    res = nodetool("statusbinary", expected_requests=[
        expected_request("GET", "/storage_service/native_transport", response=False)])
    assert res.stdout == "not running\n"

    res = nodetool("statusbinary", expected_requests=[
        expected_request("GET", "/storage_service/native_transport", response=True)])
    assert res.stdout == "running\n"