From 2f18b61d840d26921fc563caf471d36360f00f51 Mon Sep 17 00:00:00 2001
From: Piero Toffanin <pt@masseranolabs.com>
Date: Thu, 31 May 2018 22:23:37 -0400
Subject: [PATCH] Set status to none when reassigning queued task pnode

---
 app/models/task.py         | 1 +
 app/tests/test_api_task.py | 1 +
 2 files changed, 2 insertions(+)

diff --git a/app/models/task.py b/app/models/task.py
index a520d8d3..379bc3ae 100644
--- a/app/models/task.py
+++ b/app/models/task.py
@@ -312,6 +312,7 @@ class Task(models.Model):
                         logger.info("Processing node {} went offline, reassigning {}...".format(self.processing_node, self))
                         self.uuid = ''
                         self.processing_node = None
+                        self.status = None
                         self.save()
                     else:
                         # Task was running and processing node went offline
diff --git a/app/tests/test_api_task.py b/app/tests/test_api_task.py
index 6708ab87..f28ef8d5 100644
--- a/app/tests/test_api_task.py
+++ b/app/tests/test_api_task.py
@@ -585,6 +585,7 @@ class TestApiTask(BootTransactionTestCase):
 
         task.refresh_from_db()
         self.assertTrue(task.processing_node.id == another_pnode.id)
+        self.assertTrue(task.status == None)
 
         # Set task to queued, bring node offline
         task.status = status_codes.RUNNING