Restore code

mostafascoober · Jul 14, 2024 · c8325e7 · c8325e7
1 parent 4411ea9
commit c8325e7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/security-demo/force-app/main/default/classes/AccountTriggerHandler.cls b/security-demo/force-app/main/default/classes/AccountTriggerHandler.cls
@@ -2,7 +2,6 @@ public with sharing class AccountTriggerHandler {
     public static handleBefore(List<SObject> records){
         List<Account> accounts = (List<Account>) records;
         for(Account account : accounts){
-            List<User> currentUser = [SELECT Id, Name FROM user WHERE Id = :UserInfo.getUserId()];
             if(account.NumberOfEmployees < 100){
                 account.Account_Class__c = 'Small Business';
             }
@@ -13,8 +12,8 @@ public with sharing class AccountTriggerHandler {
                 account.Account_Class__c = 'Large Business';
             }
             else{
-                account.Account_Class__c = '';              
-            }  
+                account.Account_Class__c = '';       
+            }         
         }
     }
 }
diff --git a/security-demo/force-app/main/default/classes/AccountTriggerHandler_Test.cls b/security-demo/force-app/main/default/classes/AccountTriggerHandler_Test.cls
@@ -1,4 +1,3 @@
-@isTest
 public with sharing class AccountTriggerHandler_Test {
     @isTest
     public static void testAccountClassAutomation(){