fix tests

jamsocket · Jan 24, 2024 · 6876076 · 6876076
1 parent 16488ea
commit 6876076
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/...5b4ea60cfbaa5f6ae7952e9cd389fce24d10.json → ...6dc365864aed25a34f4dc5e07d4e53a27957.json b/...5b4ea60cfbaa5f6ae7952e9cd389fce24d10.json → ...6dc365864aed25a34f4dc5e07d4e53a27957.json
diff --git a/plane/src/database/connect.rs b/plane/src/database/connect.rs
@@ -13,8 +13,8 @@ use crate::{
     names::{BackendName, Name},
     protocol::{AcquiredKey, BackendAction, KeyDeadlines},
     types::{
-        BackendStatus, BearerToken, ClusterName, ConnectRequest, ConnectResponse, KeyConfig,
-        SecretToken, SpawnConfig,
+        BackendState, BackendStatus, BearerToken, ClusterName, ConnectRequest, ConnectResponse,
+        KeyConfig, SecretToken, SpawnConfig,
     },
     util::random_token,
 };
@@ -95,9 +95,10 @@ async fn create_backend_with_key(
                 drone_id,
                 expiration_time,
                 allowed_idle_seconds,
-                last_keepalive
+                last_keepalive,
+                state
             )
-            values ($1, $2, $3, now(), $4, now() + $5, $6, now())
+            values ($1, $2, $3, now(), $4, now() + $5, $6, now(), $11)
             returning id
         )
         insert into backend_key (id, key_name, namespace, tag, expires_at, fencing_token)
@@ -119,6 +120,7 @@ async fn create_backend_with_key(
         key.namespace,
         key.tag,
         PgInterval::try_from(KEY_LEASE_EXPIRATION).expect("valid constant interval"),
+        serde_json::to_value(&BackendState::Scheduled).expect("valid json"),
     )
     .fetch_optional(&mut *txn)
     .await?;