PA-752: Missed some changes in the last commit. By Nina, Sam

Author: SamBoyd

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pythonanywhere-core"
-version = "0.2.1"
+version = "0.2.3"
 description = "API wrapper for programmatic management of PythonAnywhere services."
 authors = ["PythonAnywhere <[email protected]>"]
 license = "MIT"

pythonanywhere_core/website.py

@@ -58,13 +58,15 @@ def sanity_check(self, domain_name: str, nuke: bool = False) -> None:
             )
 
 
-    def create(self, domain_name: str, command: str) -> dict:
+    def create(self, domain_name: str, command: str, nuke: bool = False) -> dict:
         """Creates new website with ``domain_name`` and ``command``.
 
         :param domain_name: domain name for new website
         :param command: command for new website
         :returns: dictionary with created website info"""
 
+        self.sanity_check(domain_name, nuke)
+
         response = call_api(
             self.websites_base_url,
             "post",

tests/test_website.py

@@ -3,6 +3,7 @@
 
 import pytest
 import responses
+from unittest.mock import patch
 
 from pythonanywhere_core.base import get_api_endpoint
 from pythonanywhere_core.exceptions import PythonAnywhereApiException
@@ -188,3 +189,11 @@ def test_website_sanity_check_tests_api_token_present(api_responses, websites_ba
         Website().sanity_check(domain_name=domain_name, nuke=False)
 
     assert "Could not find your API token" in str(e.value)
+
+
+@patch('pythonanywhere_core.website.Website.sanity_check')
+def test_create_website_calls_sanity_check(mock_sanity_check, api_responses, websites_base_url, website_info, domain_name, command):
+    mock_sanity_check.side_effect = SanityException('Boom!')
+
+    with pytest.raises(SanityException):
+        Website().create(domain_name=domain_name, command=command, nuke=False)