From 95f04094c7b5e88bf5fc575084b97b380c2cb2b4 Mon Sep 17 00:00:00 2001
From: Cayo Puigdefabregas <cayo@ribaguifi.com>
Date: Thu, 4 Feb 2021 12:51:51 +0100
Subject: [PATCH] fixing problem in live test2

---
 tests/test_action.py | 22 +++++++++++++++++++++-
 1 file changed, 21 insertions(+), 1 deletion(-)

diff --git a/tests/test_action.py b/tests/test_action.py
index d1990a0c..421a5c80 100644
--- a/tests/test_action.py
+++ b/tests/test_action.py
@@ -279,6 +279,8 @@ def test_live(user: UserClient, client: Client, app: Devicehub):
     assert action_live[0].serial_number == 'wd-wx11a80w7430'
     assert action_live[0].licence_version == '1.0'
     assert str(action_live[0].snapshot_uuid) == acer['uuid']
+    tmp_snapshots = app.config['TMP_LIVES']
+    shutil.rmtree(tmp_snapshots)
 
 
 @pytest.mark.mvp
@@ -303,6 +305,8 @@ def test_live_example(user: UserClient, client: Client, app: Devicehub):
     action_live = [a for a in db_device.actions if a.type == 'Live']
     assert len(action_live) == 1
     assert str(action_live[0].snapshot_uuid) == acer['uuid']
+    tmp_snapshots = app.config['TMP_LIVES']
+    shutil.rmtree(tmp_snapshots)
 
 
 @pytest.mark.mvp
@@ -330,6 +334,8 @@ def test_live_two_users(user: UserClient, user2: UserClient, client: Client, app
     action_live = [a for a in db_device.actions if a.type == 'Live']
     assert len(action_live) == 1
     assert str(action_live[0].snapshot_uuid) == acer['uuid']
+    tmp_snapshots = app.config['TMP_LIVES']
+    shutil.rmtree(tmp_snapshots)
 
 
 @pytest.mark.mvp
@@ -363,6 +369,8 @@ def test_live_two_allocated(user: UserClient, user2: UserClient, client: Client,
     live, _ = client.post(acer, res=models.Live, status=422)
     message = 'Expected only one Device but multiple where found'
     assert live['message'] == message
+    tmp_snapshots = app.config['TMP_LIVES']
+    shutil.rmtree(tmp_snapshots)
 
 
 @pytest.mark.mvp
@@ -394,6 +402,8 @@ def test_live_without_TestDataStorage(user: UserClient, client: Client, app: Dev
     assert live['description'] == description
     db_live = models.Live.query.filter_by(id=live['id']).one()
     assert db_live.usage_time_hdd is None
+    tmp_snapshots = app.config['TMP_LIVES']
+    shutil.rmtree(tmp_snapshots)
 
 
 @pytest.mark.mvp
@@ -421,6 +431,8 @@ def test_live_without_hdd_1(user: UserClient, client: Client, app: Devicehub):
     acer['licence_version'] = '1.0.0'
     response, _ = client.post(acer, res=models.Live, status=404)
     assert "The There aren't any disk in this device" in response['message']
+    tmp_snapshots = app.config['TMP_LIVES']
+    shutil.rmtree(tmp_snapshots)
 
 
 @pytest.mark.mvp
@@ -448,6 +460,8 @@ def test_live_without_hdd_2(user: UserClient, client: Client, app: Devicehub):
     acer['licence_version'] = '1.0.0'
     response, _ = client.post(acer, res=models.Live, status=404)
     assert "The There aren't any disk in this device" in response['message']
+    tmp_snapshots = app.config['TMP_LIVES']
+    shutil.rmtree(tmp_snapshots)
 
 
 @pytest.mark.mvp
@@ -482,6 +496,8 @@ def test_live_without_hdd_3(user: UserClient, client: Client, app: Devicehub):
     db_live = models.Live.query.filter_by(id=live['id']).one()
     assert str(db_live.usage_time_hdd) == '195 days, 12:00:00'
     assert str(db_live.usage_time_allocate) == '0:00:00'
+    tmp_snapshots = app.config['TMP_LIVES']
+    shutil.rmtree(tmp_snapshots)
 
 
 @pytest.mark.mvp
@@ -516,6 +532,8 @@ def test_live_with_hdd_with_old_time(user: UserClient, client: Client, app: Devi
     db_live = models.Live.query.filter_by(id=live['id']).one()
     assert str(db_live.usage_time_hdd) == '191 days, 8:00:00'
     assert str(db_live.usage_time_allocate) == '0:00:00'
+    tmp_snapshots = app.config['TMP_LIVES']
+    shutil.rmtree(tmp_snapshots)
 
 
 @pytest.mark.mvp
@@ -546,6 +564,8 @@ def test_live_search_last_allocate(user: UserClient, client: Client, app: Device
     acer['components'][7]['actions'] = actions
     live, _ = client.post(acer, res=models.Live)
     assert live['usageTimeAllocate'] == 1000
+    tmp_snapshots = app.config['TMP_LIVES']
+    shutil.rmtree(tmp_snapshots)
 
 
 @pytest.mark.mvp
@@ -584,7 +604,7 @@ def test_save_live_json(app: Devicehub, user: UserClient, client: Client):
         with open(path_snapshot) as file_snapshot:
             snapshot = json.loads(file_snapshot.read())
 
-        shutil.rmtree(tmp_snapshots)
+    shutil.rmtree(tmp_snapshots)
 
     assert snapshot['debug'] == debug