Tests: remake fill SD card test. Get progress bar value when busy.

Signed-off-by: Szczepan Zalega <szczepan@nitrokey.com>

2 files changed, 15 insertions, 1 deletions

diff --git a/unittest/constants.py b/unittest/constants.py
index 258619e..78a219b 100644
--- a/unittest/constants.py
+++ b/unittest/constants.py
@@ -18,6 +18,7 @@ class DefaultPasswords(Enum):
 
 class DeviceErrorCode(Enum):
     STATUS_OK = 0
+    BUSY = 1 # busy or busy progressbar in place of wrong_CRC status
     NOT_PROGRAMMED = 3
     WRONG_PASSWORD = 4
     STATUS_NOT_AUTHORIZED = 5
diff --git a/unittest/test_storage.py b/unittest/test_storage.py
index b02b1eb..600faf2 100644
--- a/unittest/test_storage.py
+++ b/unittest/test_storage.py
@@ -83,7 +83,20 @@ def test_clear_new_sd_card_notification(C):
 
 @pytest.mark.skip
 def test_fill_SD_card(C):
-    assert C.NK_fill_SD_card_with_random_data(DefaultPasswords.ADMIN) == DeviceErrorCode.STATUS_OK
+    status = C.NK_fill_SD_card_with_random_data(DefaultPasswords.ADMIN)
+    assert status == DeviceErrorCode.STATUS_OK or status == DeviceErrorCode.BUSY
+    while 1:
+        value = C.NK_get_progress_bar_value()
+        if value == -1: break
+        assert 0 <= value <= 100
+        assert C.NK_get_last_command_status() == DeviceErrorCode.STATUS_OK
+        wait(5)
+
+
+def test_get_busy_progress_on_idle(C):
+    value = C.NK_get_progress_bar_value()
+    assert value == -1
+    assert C.NK_get_last_command_status() == DeviceErrorCode.STATUS_OK
 
 
 def test_change_update_password(C):