Add enode link URL endpoint

pv_site_api/fake.py

@@ -120,3 +120,8 @@ def make_fake_status() -> PVSiteAPIStatus:
         message="The API is up and running",
     )
     return pv_api_status
+
+
+def make_fake_enode_link_url() -> str:
+    """Make fake Enode link URL"""
+    return "https://enode.com"

pv_site_api/main.py

@@ -9,7 +9,7 @@
 from fastapi import Depends, FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.openapi.utils import get_openapi
-from fastapi.responses import FileResponse, JSONResponse
+from fastapi.responses import FileResponse, JSONResponse, RedirectResponse
 from pvlib import irradiance, location, pvsystem
 from pvsite_datamodel.read.site import get_all_sites, get_site_by_uuid
 from pvsite_datamodel.read.status import get_latest_status
@@ -28,6 +28,7 @@
 )
 from .fake import (
     fake_site_uuid,
+    make_fake_enode_link_url,
     make_fake_forecast,
     make_fake_inverters,
     make_fake_pv_generation,
@@ -358,6 +359,26 @@ def get_pv_estimate_clearsky(site_uuid: str, session: Session = Depends(get_sess
     return res
 
 
+@app.get("/enode/link", response_class=RedirectResponse)
+def get_enode_link(redirect_uri: str, session: Session = Depends(get_session)):
+    """
+    ### Returns a URL from Enode that starts a user's Enode link flow.
+    """
+    if int(os.environ["FAKE"]):
+        return make_fake_enode_link_url()
+
+    client = session.query(ClientSQL).first()
+    assert client is not None
+
+    with httpx.Client() as httpx_client:
+        data = {"vendorType": "inverter", "redirectUri": redirect_uri}
+        res = httpx_client.post(
+            f"https://enode-api.production.enode.io/users/{client.client_uuid}/link", data=data
+        ).json()
+
+    return res["linkUrl"]
+
+
 @app.get("/inverters")
 async def get_inverters(
     session: Session = Depends(get_session),

tests/conftest.py

@@ -21,11 +21,6 @@
 from pv_site_api.session import get_session
 
 
-@pytest.fixture()
-def non_mocked_hosts():
-    return ["testserver"]
-
-
 @pytest.fixture(scope="session")
 def engine():
     """Make database engine"""
@@ -70,6 +65,12 @@ def clients(db_session):
     return clients
 
 
+@pytest.fixture
+def non_mocked_hosts() -> list:
+    """Prevent TestClient fixture from being mocked"""
+    return ["testserver"]
+
+
 @pytest.fixture()
 def sites(db_session, clients):
     """Create some fake sites"""

tests/test_enode.py

@@ -0,0 +1,25 @@
+def test_get_enode_link_fake(client, fake):
+    params = {"redirect_uri": "https://example.org"}
+    response = client.get("/enode/link", params=params, follow_redirects=False)
+
+    assert response.status_code == 307
+    assert len(response.headers["location"]) > 0
+
+
+def test_get_enode_link(client, clients, httpx_mock):
+    test_enode_link_uri = "https://example.com"
+
+    httpx_mock.add_response(
+        url=f"https://enode-api.production.enode.io/users/{clients[0].client_uuid}/link",
+        json={"linkUrl": test_enode_link_uri},
+    )
+
+    params = {"redirect_uri": "https://example.org"}
+    response = client.get(
+        "/enode/link",
+        params=params,
+        follow_redirects=False,
+    )
+
+    assert response.status_code == 307
+    assert response.headers["location"] == test_enode_link_uri