From 0a0a47a5e464147706767b88646cb51bc54f9a3d Mon Sep 17 00:00:00 2001
From: eeintech <eeintech@eeinte.ch>
Date: Tue, 12 Jan 2021 17:11:46 -0500
Subject: [PATCH] Added location owner change test case to verify stock item
 owner in that location is also updated

---
 InvenTree/stock/test_views.py | 17 ++++++++++++++++-
 InvenTree/stock/views.py      |  1 -
 2 files changed, 16 insertions(+), 2 deletions(-)

diff --git a/InvenTree/stock/test_views.py b/InvenTree/stock/test_views.py
index 366f15239f..8789e9f76e 100644
--- a/InvenTree/stock/test_views.py
+++ b/InvenTree/stock/test_views.py
@@ -384,7 +384,6 @@ class StockOwnershipTest(StockViewTestCase):
         # Try to create new item with no owner
         response = self.client.post(reverse('stock-item-create'),
                                     new_item, HTTP_X_REQUESTED_WITH='XMLHttpRequest')
-        # print(response.content)
         self.assertContains(response, '"form_valid": false', status_code=200)
 
         # Try to create new item with invalid owner
@@ -398,3 +397,19 @@ class StockOwnershipTest(StockViewTestCase):
         response = self.client.post(reverse('stock-item-create'),
                                     new_item, HTTP_X_REQUESTED_WITH='XMLHttpRequest')
         self.assertContains(response, '"form_valid": true', status_code=200)
+
+        # Logout
+        self.client.logout()
+
+        # Login with admin
+        self.client.login(username='username', password='password')
+
+        # Switch owner of location
+        response = self.client.post(reverse('stock-location-edit', args=(location_created.pk,)),
+                                    {'name': new_location['name'], 'owner': user_group_owner.pk},
+                                    HTTP_X_REQUESTED_WITH='XMLHttpRequest')
+        self.assertContains(response, '"form_valid": true', status_code=200)
+
+        # Check that owner was updated for item in this location
+        stock_item = StockItem.objects.all().last()
+        self.assertEqual(stock_item.owner, user_group_owner)
diff --git a/InvenTree/stock/views.py b/InvenTree/stock/views.py
index 9b67923514..78c1d5fb13 100644
--- a/InvenTree/stock/views.py
+++ b/InvenTree/stock/views.py
@@ -1800,7 +1800,6 @@ class StockItemCreate(AjaxCreateView):
         if not stock_ownership_control:
             form.fields['owner'].widget = HiddenInput()
         else:
-            print('> Stock ownership is enabled')
             try:
                 location_owner = location.owner
             except AttributeError: