Merge pull request #170 from rstudio/get-meta

Author: isabelizimm

vetiver/server.py

@@ -100,6 +100,10 @@ def pin_url():
         async def ping():
             return {"ping": "pong"}
 
+        @app.get("/metadata")
+        async def get_metadata():
+            return self.model.metadata.to_dict()
+
         self.vetiver_post(
             self.model.handler_predict, "predict", check_prototype=self.check_prototype
         )

vetiver/tests/test_ping_server.py

@@ -0,0 +1,48 @@
+from vetiver import mock, VetiverModel, VetiverAPI
+from fastapi.testclient import TestClient
+import pytest
+import sys
+
+
+@pytest.fixture
+def vetiver_model():
+    X, y = mock.get_mock_data()
+    model = mock.get_mock_model().fit(X, y)
+    v = VetiverModel(
+        model=model,
+        prototype_data=X,
+        model_name="my_model",
+        versioned=None,
+        description="A regression model for testing purposes",
+    )
+    return v
+
+
+@pytest.fixture
+def client(vetiver_model):
+    app = VetiverAPI(vetiver_model)
+
+    return TestClient(app.app)
+
+
+def test_get_ping(client):
+    response = client.get("/ping")
+    assert response.status_code == 200, response.text
+    assert response.json() == {"ping": "pong"}
+
+
+def test_get_docs(client):
+    response = client.get("/__docs__")
+    assert response.status_code == 200, response.text
+
+
+def test_get_metadata(client):
+    response = client.get("/metadata")
+    assert response.status_code == 200, response.text
+    assert response.json() == {
+        "user": {},
+        "version": None,
+        "url": None,
+        "required_pkgs": ["scikit-learn"],
+        "python_version": list(sys.version_info),  # JSON will return a list
+    }