Add apiserver tests

freqtrade/tests/rpc/test_rpc_apiserver.py

@@ -0,0 +1,52 @@
+"""
+Unit test file for rpc/api_server.py
+"""
+
+from unittest.mock import MagicMock
+
+from freqtrade.rpc.api_server import ApiServer
+from freqtrade.state import State
+from freqtrade.tests.conftest import get_patched_freqtradebot, patch_apiserver
+
+
+def test__init__(default_conf, mocker):
+    """
+    Test __init__() method
+    """
+    mocker.patch('freqtrade.rpc.telegram.Updater', MagicMock())
+    mocker.patch('freqtrade.rpc.api_server.ApiServer.run', MagicMock())
+
+    apiserver = ApiServer(get_patched_freqtradebot(mocker, default_conf))
+    assert apiserver._config == default_conf
+
+
+def test_start_endpoint(default_conf, mocker):
+    """Test /start endpoint"""
+    patch_apiserver(mocker)
+    bot = get_patched_freqtradebot(mocker, default_conf)
+    apiserver = ApiServer(bot)
+
+    bot.state = State.STOPPED
+    assert bot.state == State.STOPPED
+    result = apiserver.start()
+    assert result == '{"status": "starting trader ..."}'
+    assert bot.state == State.RUNNING
+
+    result = apiserver.start()
+    assert result == '{"status": "already running"}'
+
+
+def test_stop_endpoint(default_conf, mocker):
+    """Test /stop endpoint"""
+    patch_apiserver(mocker)
+    bot = get_patched_freqtradebot(mocker, default_conf)
+    apiserver = ApiServer(bot)
+
+    bot.state = State.RUNNING
+    assert bot.state == State.RUNNING
+    result = apiserver.stop()
+    assert result == '{"status": "stopping trader ..."}'
+    assert bot.state == State.STOPPED
+
+    result = apiserver.stop()
+    assert result == '{"status": "already stopped"}'