fix tests

pv_site_api/_db_helpers.py

@@ -244,6 +244,7 @@ def site_to_pydantic(site: SiteSQL) -> PVSiteMetadata:
         inverter_capacity_kw=site.inverter_capacity_kw,
         module_capacity_kw=site.module_capacity_kw,
         created_utc=site.created_utc,
+        capacity_kw=site.capacity_kw,
     )
     return pv_site
 

pv_site_api/fake.py

@@ -33,6 +33,7 @@ def make_fake_site() -> PVSites:
         longitude=0,
         inverter_capacity_kw=1,
         module_capacity_kw=1.3,
+        capacity_kw=1,
     )
     pv_site_list = PVSites(
         site_list=[pv_site],

pv_site_api/main.py

@@ -258,8 +258,8 @@ def post_site_info(
 
     if is_fake():
         print(f"Successfully added {site_info.dict()} for site {site_info.client_site_name}")
-        print("Not doing anything with it (yet!)")
-        return
+        site = make_fake_site().site_list[0]
+        return site
 
     user = get_user_by_email(session=session, email=auth["https://openclimatefix.org/email"])
 

tests/test_sites.py

@@ -4,7 +4,7 @@
 
 from pvsite_datamodel.sqlmodels import SiteSQL
 
-from pv_site_api.pydantic_models import PVSiteMetadata, PVSites
+from pv_site_api.pydantic_models import PVSiteInputMetadata, PVSites
 
 
 def test_get_site_list_fake(client, fake):
@@ -48,7 +48,7 @@ def test_get_site_list_min(client, sites):
 
 
 def test_put_site_fake(client, fake):
-    pv_site = PVSiteMetadata(
+    pv_site = PVSiteInputMetadata(
         client_name="client_name_1",
         client_site_id="the site id used by the user",
         client_site_name="the site name",
@@ -72,7 +72,7 @@ def test_put_site_fake(client, fake):
 
 def test_put_site(db_session, client):
     # make site object
-    pv_site = PVSiteMetadata(
+    pv_site = PVSiteInputMetadata(
         client_name="test_client",
         client_site_id="1",
         client_site_name="the site name",