rsoesemann / apex-domainbuilder Goto Github PK

@IsTest
public class Parent_t extends DomainBuilder {

    public Parent_t() {
        super(ParentObject__c.SObjectType);
    }
}

@IsTest
public class Child_t extends DomainBuilder{
	
    public Child_t(Parent_t p) {
        super(ChildObject__c.SObjectType);
        setParent(ChildObject__c.Parent__c, p);
    }
    
	public Child_t add(Parent_t p) {
        return (Child_t) p.setParent(ParentObject__c.Child__c, this);
    } 
}

@IsTest
private class TestDomainBuilder {
    
    @TestSetup
    private static void setup() {
    	Parent_t p = new Parent_t();
	Child_t c = new Child_t(p);
        //p.persist(); // this does not help as workaround
        c.add(p);
        p.persist();
    }
    
    @IsTest
    private static void testSetup() {
        System.assertEquals(1, [SELECT Id FROM ParentObject__c].size());
        System.assertEquals(1, [SELECT Id FROM ChildObject__c].size());      
	System.assertEquals(1, [SELECT Id FROM ParentObject__c WHERE Child__c != NULL].size());      
    }
}

System.DmlException: Insert failed. First exception on row 0; first error: REQUIRED_FIELD_MISSING, Required fields are missing: [Parent__c]: [Parent__c]

public class DemandService_Test {
  @isTest
  public static void testWithDomainBuilder() {
    // Given
    Account_t acc = new Account_t();
    Opportunity_t oppo = new Opportunity_t(acc).add(new Contact_t());
    List<Product_t> products = new List<Product_t>();
    List<Schema.PicklistEntry> picklistEntries = Product2.Family.getDescribe().getPicklistValues();
    for(Integer i=0; i<50; i++) {
      products.add(
        new Product_t().name('Test Product ' + i).code('TESTP' + i).family(picklistEntries[Math.floor(Math.random() * ((19 - 0) + 1) + 0).intValue()].value)
      );
    }
    for(Product_t p : products) {
      new Demand(acc, p);
    }
    acc.persist();

    // When
    Test.startTest();
    for(Product_t p : products) {
      new OpportunityLineItem_t(oppo, p).quantity(new Random().integer(1,100));
    }
    Test.stopTest();

    // Then
    List<OpportunityLineItem> oppoItems = [SELECT Id, Product2Id, KS_AchievableDemand__c, KS_AchievableDemand__r.KS_Product__c FROM OpportunityLineItem];
    for(OpportunityLineItem oppoItem : oppoItems) {
      Assert.areEqual(oppoItem.Product2Id, oppoItem.KS_AchievableDemand__r.KS_Product__c);
    }

    
  }

  @isTest
  public static void testWithoutDomainBuilder() {
    // Given
    Account acc = new Account(Name = 'Test');
    insert acc;
    Opportunity oppo = new Opportunity(AccountId = acc.Id, Name = 'test', StageName = 'Qualification', closeDate = Date.today(), Pricebook2Id = Test.getStandardPricebookId());
    insert oppo;
    List<Product2> products = new List<Product2>();
    List<Schema.PicklistEntry> picklistEntries = Product2.Family.getDescribe().getPicklistValues();
    for(Integer i=0; i<50; i++) {
      products.add(
        new Product2(Name = 'Test Product ' + i, ProductCode = 'TESTP' + i, Family = picklistEntries[Math.floor(Math.random() * ((19 - 0) + 1) + 0).intValue()].value)
      );
    }
    insert products;
    AchievableDemand__c[] demands = new List<AchievableDemand__c>();
    for(Product2 p : products) {
      new AchievableDemand__c(Account__c = acc.Id, Product__c = p.Id);
    }
    insert demands;
    
    PricebookEntry[] entries = new List<PricebookEntry>();

    for(Product2 p : products) {
      entries.add(
        new PricebookEntry(
          Product2Id = p.Id,
          Pricebook2Id = Test.getStandardPricebookId(),
          UnitPrice = 10,
          IsActive = true
        )
      );
    }
    insert entries;

    // When
    Test.startTest();
    OpportunityLineItem[] oppoItems = new List<OpportunityLineItem>();
    for(Product2 p : products) {
      oppoItems.add(
        new OpportunityLineItem(
          OpportunityId = oppo.Id,
          Product2Id = p.Id,
          Quantity = 10,
          TotalPrice = 10
        )
      );
    }
    insert oppoItems;

    Test.stopTest();

    // Then
    oppoItems = [SELECT Id, Product2Id, AchievableDemand__c, AchievableDemand__r.Product__c FROM OpportunityLineItem];
    for(OpportunityLineItem oppoItem : oppoItems) {
      Assert.areEqual(oppoItem.Product2Id, oppoItem.AchievableDemand__r.Product__c);
    }

  }
}