Setup data structures for tracking tests

lib/polar/machines.ex

@@ -1,2 +1,7 @@
 defmodule Polar.Machines do
+  alias __MODULE__.Manager
+
+  defdelegate create_cluster(params),
+    to: Manager,
+    as: :create
 end

lib/polar/machines/assessment.ex

@@ -0,0 +1,27 @@
+defmodule Polar.Machines.Assessment do
+  use Ecto.Schema
+  import Ecto.Changeset
+
+  alias Polar.Streams
+  alias Polar.Machines.Check
+  alias Polar.Machines.Cluster
+  s
+
+  schema "assessments" do
+    field :current_state, :string, default: "created"
+
+    belongs_to :check, Check
+    belongs_to :cluster, Cluster
+
+    belongs_to :version, Streams.Version
+
+    timestamps(type: :utc_datetime)
+  end
+
+  @doc false
+  def changeset(assessment, attrs) do
+    assessment
+    |> cast(attrs, [:current_state])
+    |> validate_required([:current_state])
+  end
+end

lib/polar/machines/check.ex

@@ -0,0 +1,18 @@
+defmodule Polar.Machines.Check do
+  use Ecto.Schema
+  import Ecto.Changeset
+
+  schema "checks" do
+    field :slug, :string
+    field :description, :string
+
+    timestamps(type: :utc_datetime_usec)
+  end
+
+  @doc false
+  def changeset(check, attrs) do
+    check
+    |> cast(attrs, [:slug, :description])
+    |> validate_required([:slug, :description])
+  end
+end

lib/polar/machines/cluster.ex

@@ -2,12 +2,19 @@ defmodule Polar.Machines.Cluster do
   use Ecto.Schema
   import Ecto.Changeset
 
+  @valid_attrs ~w(
+    name
+    type
+    arch
+    credential
+  )a
+
   schema "clusters" do
     field :name, :string
     field :current_state, :string, default: "created"
 
     field :type, :string
-    field :architecture, :string
+    field :arch, :string
 
     field :credential, Polar.Encrypted.Map
 
@@ -17,7 +24,7 @@ defmodule Polar.Machines.Cluster do
   @doc false
   def changeset(cluster, attrs) do
     cluster
-    |> cast(attrs, [:name, :type, :architecture, :current_state, :credential])
-    |> validate_required([:name, :type, :architecture, :current_state, :credential])
+    |> cast(attrs, @valid_attrs)
+    |> validate_required(@valid_attrs)
   end
 end

lib/polar/machines/cluster/manager.ex

@@ -0,0 +1,10 @@
+defp Polar.Machines.Cluster do
+  alias Polar.Repo
+  alias Polar.Machines.Cluster
+
+  def create(params) do
+    %Cluster{}
+    |> Cluster.changeset(params)
+    |> Repo.insert()
+  end
+end

priv/repo/migrations/20240619130455_create_checks.exs

@@ -0,0 +1,14 @@
+defmodule Polar.Repo.Migrations.CreateChecks do
+  use Ecto.Migration
+
+  def change do
+    create table(:checks) do
+      add :slug, :citext, null: false
+      add :description, :citext, null: false
+
+      timestamps(type: :utc_datetime_usec)
+    end
+
+    create index(:checks, [:slug], unique: true)
+  end
+end

priv/repo/migrations/20240619130844_create_assessments.exs

@@ -0,0 +1,21 @@
+defmodule Polar.Repo.Migrations.CreateAssessments do
+  use Ecto.Migration
+
+  def change do
+    create table(:assessments) do
+      add :current_state, :citext, default: "created", null: false
+
+      add :check_id, references(:checks, on_delete: :restrict), null: false
+      add :version_id, references(:versions, on_delete: :restrict), null: false
+      add :cluster_id, references(:clusters, on_delete: :restrict), null: false
+
+      timestamps(type: :utc_datetime_utc)
+    end
+
+    create index(:assessments, [:check_id])
+    create index(:assessments, [:version_id])
+    create index(:assessments, [:cluster_id])
+
+    create index(:assessments, [:check_id, :version_id], unique: true)
+  end
+end