livebook/test/support/k8s_cluster_stub.ex

defmodule Livebook.K8sClusterStub do
  use Plug.Router

  require Logger

  plug :match

  plug Plug.Parsers,
    parsers: [:urlencoded, :json],
    json_decoder: JSON

  plug :dispatch

  post "/apis/authorization.k8s.io/v1/selfsubjectaccessreviews" do
    resource_attributes = conn.body_params["spec"]["resourceAttributes"]
    resource = resource_attributes["resource"]
    allowed = conn.host == "default" or resource == "selfsubjectaccessreviews"

    conn
    |> put_status(201)
    |> Req.Test.json(%{
      "status" => %{"allowed" => allowed},
      "spec" => %{"resourceAttributes" => resource_attributes}
    })
  end

  get "/api/v1/namespaces", host: "default" do
    Req.Test.json(conn, %{"items" => [%{"metadata" => %{"name" => "default"}}]})
  end

  get "/api/v1/namespaces", host: "no-permission" do
    send_resp(conn, 403, "")
  end

  get "apis/storage.k8s.io/v1/storageclasses", host: "default" do
    Req.Test.json(conn, %{
      "items" => [
        %{"metadata" => %{"name" => "first-storage-class"}},
        %{"metadata" => %{"name" => "second-storage-class"}}
      ]
    })
  end

  get "/api/v1/namespaces/default/persistentvolumeclaims", host: "default" do
    Req.Test.json(conn, %{
      "items" => [
        %{"metadata" => %{"name" => "foo-pvc"}},
        %{"metadata" => %{"name" => "new-pvc"}}
      ]
    })
  end

  delete "/api/v1/namespaces/default/persistentvolumeclaims/:name", host: "default" do
    send_resp(conn, 200, "")
  end

  post "/api/v1/namespaces/default/persistentvolumeclaims", host: "default" do
    conn
    |> put_status(201)
    |> Req.Test.json(%{"metadata" => %{"name" => "new-pvc"}})
  end

  match _ do
    Logger.error("Unimplemented #{conn.method} Stub Request to #{conn.request_path}")

    conn
    |> put_status(500)
    |> Req.Test.text("Endpoint not implemented")
  end
end