Actions - Add tests

freqtrade/tests/rpc/test_rpc_apiserver.py

@@ -12,9 +12,10 @@ from freqtrade.tests.conftest import get_patched_freqtradebot, patch_apiserver
 
 
 @pytest.fixture
-def client(default_conf, mocker):
+def botclient(default_conf, mocker):
-    apiserver = ApiServer(get_patched_freqtradebot(mocker, default_conf))
+    ftbot = get_patched_freqtradebot(mocker, default_conf)
-    yield apiserver.app.test_client()
+    apiserver = ApiServer(ftbot)
+    yield ftbot, apiserver.app.test_client()
     # Cleanup ... ?
 
 
@@ -23,19 +24,32 @@ def response_success_assert(response):
     assert response.content_type == "application/json"
 
 
-def test_start(client):
+def test_api_stop_workflow(botclient):
+    ftbot, client = botclient
+    assert ftbot.state == State.RUNNING
+    rc = client.post("/stop")
+    response_success_assert(rc)
+    assert rc.json == {'status': 'stopping trader ...'}
+    assert ftbot.state == State.STOPPED
+
+    # Stop bot again
+    rc = client.post("/stop")
+    response_success_assert(rc)
+    assert rc.json == {'status': 'already stopped'}
+
+    # Start bot
+    rc = client.post("/start")
+    response_success_assert(rc)
+    assert rc.json == {'status': 'starting trader ...'}
+    assert ftbot.state == State.RUNNING
+
+    # Call start again
     rc = client.post("/start")
     response_success_assert(rc)
     assert rc.json == {'status': 'already running'}
 
 
-def test_stop(client):
+def test_api__init__(default_conf, mocker):
-    rc = client.post("/stop")
-    response_success_assert(rc)
-    assert rc.json == {'status': 'stopping trader ...'}
-
-
-def test__init__(default_conf, mocker):
     """
     Test __init__() method
     """
@@ -46,33 +60,20 @@ def test__init__(default_conf, mocker):
     assert apiserver._config == default_conf
 
 
-def test_start_endpoint(default_conf, mocker):
+def test_api_reloadconf(botclient):
-    """Test /start endpoint"""
+    ftbot, client = botclient
-    patch_apiserver(mocker)
-    bot = get_patched_freqtradebot(mocker, default_conf)
-    apiserver = ApiServer(bot)
 
-    bot.state = State.STOPPED
+    rc = client.post("/reload_conf")
-    assert bot.state == State.STOPPED
+    response_success_assert(rc)
-    result = apiserver.start()
+    assert rc.json == {'status': 'reloading config ...'}
-    assert result == '{"status": "starting trader ..."}'
+    assert ftbot.state == State.RELOAD_CONF
-    assert bot.state == State.RUNNING
-
-    result = apiserver.start()
-    assert result == '{"status": "already running"}'
 
 
-def test_stop_endpoint(default_conf, mocker):
+def test_api_stopbuy(botclient):
-    """Test /stop endpoint"""
+    ftbot, client = botclient
-    patch_apiserver(mocker)
+    assert ftbot.config['max_open_trades'] != 0
-    bot = get_patched_freqtradebot(mocker, default_conf)
-    apiserver = ApiServer(bot)
 
-    bot.state = State.RUNNING
+    rc = client.post("/stopbuy")
-    assert bot.state == State.RUNNING
+    response_success_assert(rc)
-    result = apiserver.stop()
+    assert rc.json == {'status': 'No more buy will occur from now. Run /reload_conf to reset.'}
-    assert result == '{"status": "stopping trader ..."}'
+    assert ftbot.config['max_open_trades'] == 0
-    assert bot.state == State.STOPPED
-
-    result = apiserver.stop()
-    assert result == '{"status": "already stopped"}'