make user triggered reimaging happen immediately (#566)

microsoft · Feb 18, 2021 · 929d9ce · 929d9ce
1 parent 21f08f6
commit 929d9ce
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/api-service/__app__/node/__init__.py b/src/api-service/__app__/node/__init__.py
@@ -85,7 +85,7 @@ def delete(req: func.HttpRequest) -> func.HttpResponse:
 def patch(req: func.HttpRequest) -> func.HttpResponse:
     request = parse_request(NodeGet, req)
     if isinstance(request, Error):
-        return not_ok(request, context="NodeRestart")
+        return not_ok(request, context="NodeReimage")
 
     node = Node.get_by_machine_id(request.machine_id)
     if not node:
@@ -94,7 +94,7 @@ def patch(req: func.HttpRequest) -> func.HttpResponse:
             context=request.machine_id,
         )
 
-    node.stop()
+    node.stop(done=True)
     if node.debug_keep_node:
         node.debug_keep_node = False
         node.save()

diff --git a/src/api-service/__app__/onefuzzlib/workers/nodes.py b/src/api-service/__app__/onefuzzlib/workers/nodes.py
@@ -281,8 +281,8 @@ def add_ssh_public_key(self, public_key: str) -> Result[None]:
         )
         return None
 
-    def stop(self) -> None:
-        self.to_reimage()
+    def stop(self, done: bool = False) -> None:
+        self.to_reimage(done=done)
         self.send_message(NodeCommand(stop=StopNodeCommand()))
 
     def set_shutdown(self) -> None: