Apex Code Development You need to sign in to do that
Don't have an account?
BobP Test Class for a apex extension class
I have an apex extension controller that i need to write a test class, but I'm not sure how to write a test class for this type of controller. Any help would be greatly appreciated. My code is below.
Public Class AccountExtensionController{
private Account acc;
public List<Bids_Sent__c> bidsList {get;set;}
public Map<String,List<Site_Bid_Details__c>> bidsMap {get;set;}
public AccountExtensionController(ApexPages.StandardController sc){
acc = (Account)sc.getRecord();
bidsList = new List<Bids_Sent__c>();
bidsList = [SELECT Id,IsAddedToPDF__c,Customer__r.Service_Agreement_Verbiage__c,Site__c,Site__r.Contract_Start_Date__c,Site__r.Contract_End_Date__c,Site__r.Customer_Location_ID__c,Service_Year__c,Customer__r.Contract_Start_Date__c,Name,Customer__r.Contract_End_Date__c,Site__r.Name,Customer__r.Name,Primary_Contact__r.FirstName,Site__r.BillingCity,Site__r.BillingState,Site__r.BillingStreet,Site__r.BillingPostalCode FROM Bids_Sent__c WHERE Awarded__c =: acc.Id AND IsAddedToPDF__c=true];
Set<Id> bidId = new Set<Id>();
for(Bids_Sent__c bs : bidsList){
bidId.add(bs.Id);
}
bidsMap = new Map<String,List<Site_Bid_Details__c>> ();
for(Site_Bid_Details__c bd : [SELECT Id, Bid_Name__r.Name,Site__c,Contract_Start_Month__c,Site__r.Customer_Location_ID__c,Cost__c,Customer__r.Contract_Month__c,Increment__c,Total__c,Price__c,Scope__c,Bid_Name__r.Service_Type__c,Number_of_Months__c,Retainer_Fee__c,Monthly_Payment__c,UOM__c FROM Site_Bid_Details__c WHERE Bid_Name__c IN : bidId]){
if(bidsMap.containsKey(bd.Bid_Name__r.Name)){
System.debug('CONTAINS KEY: ' + bd.Bid_Name__r.Name);
bidsMap.get(bd.Bid_Name__r.Name).add(bd);
} else {
System.debug('CREATE: ' + bd.Bid_Name__r.Name);
bidsMap.put(bd.Bid_Name__r.Name,new List<Site_Bid_Details__c>{bd});
}
}
}
}
You need to instanctiate a new apexpages.Standardcontroller of specific standard object controller type and assign this as parameter to your extension. This is the main thing.
Test.setCurrentPage(pageRef);
PageReference pageRef = Page.<name of vfpage>;
ApexPages.StandardController sc = new ApexPages.StandardController(new Account(ID=<account id>));
AccountExtensionController testAccPlan = new AccountExtensionController(sc);
Linked article to check: https://developer.salesforce.com/forums/?id=906F00000008y7QIAQ
Choose this reply as best option if it solves your problem.
Cheers!!
if I have two visualforce pages using the controller can do the following?
PageReference pageRef = Page.<name of vfpage1>;
PageReference pageRef = Page.<name of vfpage2>;
I'm not having any luck writing this test class. Could you assist me with this entension controller test class?