Only change mock_import_data if import should succeed

Author: dgreen-arm

tests/suites/test_suite_psa_crypto_se_driver_hal_mocks.function

@@ -125,12 +125,16 @@ static psa_status_t mock_import( psa_key_slot_number_t slot_number,
     (void) p_data;
 
     mock_import_data.called++;
-    mock_import_data.slot_number = slot_number;
-    mock_import_data.lifetime = lifetime;
-    mock_import_data.type = type;
-    mock_import_data.algorithm = algorithm;
-    mock_import_data.usage = usage;
-    mock_import_data.data_length = data_length;
+
+    if( mock_import_data.return_value == PSA_SUCCESS )
+    {
+        mock_import_data.slot_number = slot_number;
+        mock_import_data.lifetime = lifetime;
+        mock_import_data.type = type;
+        mock_import_data.algorithm = algorithm;
+        mock_import_data.usage = usage;
+        mock_import_data.data_length = data_length;
+    }
 
     return( mock_import_data.return_value );
 }
@@ -270,8 +274,6 @@ void mock_import( int mock_alloc_return_value,
                                  &handle ) == expected_result );
 
     TEST_ASSERT( mock_allocate_data.called == 1 );
-    TEST_ASSERT( mock_import_data.called ==
-                  (expected_result == PSA_SUCCESS? 1 : 0) );
     TEST_ASSERT( mock_import_data.type ==
                   (expected_result == PSA_SUCCESS? PSA_KEY_TYPE_RAW_DATA : 0));