From 547d1a08563c46c75ef8b63fa61bf8caff9acd30 Mon Sep 17 00:00:00 2001
From: Lucy C <12953208+elvece@users.noreply.github.com>
Date: Tue, 4 Jan 2022 11:37:31 -0700
Subject: [PATCH] Bugfix/boot cleanup (#1023)

* do not expect the node to exist on cleanup

* fix reference

* fix types
---
 appmgr/src/context/rpc.rs | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/appmgr/src/context/rpc.rs b/appmgr/src/context/rpc.rs
index c2d117f6f..b4f8aacdc 100644
--- a/appmgr/src/context/rpc.rs
+++ b/appmgr/src/context/rpc.rs
@@ -7,6 +7,7 @@ use std::sync::Arc;
 use std::time::Duration;
 
 use bollard::Docker;
+use color_eyre::eyre::eyre;
 use patch_db::json_ptr::JsonPointer;
 use patch_db::{DbHandle, PatchDb, Revision};
 use reqwest::Url;
@@ -306,11 +307,14 @@ impl RpcContext {
                 let mut pde = crate::db::DatabaseModel::new()
                     .package_data()
                     .idx_model(&package_id)
-                    .expect(&mut db)
-                    .await?
                     .get_mut(&mut db)
                     .await?;
-                match &mut *pde {
+                match pde.as_mut().ok_or_else(|| {
+                    Error::new(
+                        eyre!("Node does not exist: /package-data/{}", package_id),
+                        crate::ErrorKind::Database,
+                    )
+                })? {
                     PackageDataEntry::Installing { .. }
                     | PackageDataEntry::Restoring { .. }
                     | PackageDataEntry::Updating { .. } => {