trigger UKCheck on Domain__c (before insert, before update) {
    
    Map<String,Country_Codes__c> getCodes = Country_Codes__c.getAll();
        
    for (Domain__c dom : Trigger.new) {
        	
        if(dom.Country__c != NULL) {
            	for(String s : getCodes.keySet()) {
                	if(s.containsIgnoreCase(dom.Country__c)) {
                    
                    	dom.UK_Team_Checkbox__c = TRUE;
                    
                	}
               } 
     }
        
        System.debug('dom.Owner is ' + dom.Owner);
        System.debug(dom.Company__r.Owner.Id);
        System.debug('dom.Company__c is ' + dom.Company__c);
    }

    Map<Id, Account> a = new Map<Id, Account>();
    
    for(Domain__c domain : Trigger.new) {
        a.put(domain.Company__c, null);
    }
    System.debug('Before putAll ' + a);
    
    a.putAll([SELECT OwnerId FROM Account WHERE Id IN :a.keySet()]);
    System.debug('After putAll ' + a);
    
    for(Domain__c domain : Trigger.new) {
        if(domain.Company__c != NULL) {
            domain.OwnerId = a.get(domain.Company__c).OwnerId;
        }
    } 
}

@isTest
public class UKCheck_Test {
    
    @TestSetup 
    static void createUser() {
        
        UserRole r = new UserRole(DeveloperName = 'SovrnCustomRole', Name = 'Sovrn Role');
        insert r;
        
        User u1 = new User();
        u1.LastName          = 'Test1';
        u1.Alias             = 'alias1';
        u1.Email             = 'test1@fake.com';
        u1.Username          = 'testsovrn2018@fake.com';
        u1.CompanyName       = 'Test1';
        u1.TimeZoneSidKey    ='America/Los_Angeles';
        u1.EmailEncodingKey  ='UTF-8';
        u1.LocaleSidKey      = 'en_US';
        u1.UserRoleId        = r.Id;
        u1.LanguageLocaleKey = 'en_US';
        u1.ProfileId         = [SELECT Id FROM Profile WHERE Name = 'Publisher Advocate'].Id;
        insert u1;
        
    }

    static testmethod void validateUKCheck () {
        
        List<Account> accts = new List<Account>();
        List<Domain__c> domList = new List<Domain__c>();
        
        //Create Companies(Account)        
        Account acct   = new Account();
        acct.Name      = 'TestCompany1';
        acct.OwnerId   = UserInfo.getUserId();
        accts.add(acct);
        System.debug('Acct equals ' + acct.Name);
        System.debug('Acct Owner is ' + acct.OwnerId);
        
        Account acct1 = new Account();
        acct1.Name    = 'TestCompany2';
        acct1.OwnerId = [SELECT Id FROM User WHERE UserName = 'testsovrn2018@fake.com'].Id;
        accts.add(acct1);
        System.debug('Acct1 equals ' + acct1.Name);
        System.debug('Acct1 Owner is ' + acct1.OwnerId);
        
        Test.startTest();
        insert accts;
        Test.stopTest();
        
        //Create Custom Setting
        Country_Codes__c cS = new Country_Codes__c();
        cS.Name             = 'GB';
        cS.Country__c       = 'GB';
        insert cS;
        
        //Create Domain records
        Domain__c UKDom  = new Domain__c();
        UKDom.Name       = 'testdomain.com';
        UKDom.Website__c = 'www.testdomain.com';
        UKDom.Country__c = 'GB';
        UKDom.Uk_Team__c = TRUE;
        UKDom.Company__c = acct.Id;
        UKDom.OwnerId    = UserInfo.getUserId();
        System.debug('UKDom domain Company__c is ' + acct.Name);
        System.debug('UKDom domain ownerID is ' + UKDom.OwnerId);        
        
        domList.add(UKDom);
        
        Domain__c noUK  = new Domain__c();
        noUK.Name       = 'testdomainnoUK.com';
        noUK.Website__c = 'www.testdomainnoUK.com';
        noUK.Country__c = 'US';
        noUK.OwnerId    = acct1.OwnerId;
        noUk.Company__c = acct1.Id;
        noUK.UK_Team__c = FALSE;
        System.debug('noUK domain Company__c is ' + acct1.Name);
        System.debug('noUK domain Company__c is ' + acct1.Id);
        System.debug('noUK domain ownerID is ' + noUK.OwnerId);
        
        domList.add(noUK);
        
		insert domList;
        System.debug('noUK domain ID ' + noUk.Id);
        
        Domain__c domainToUpdate = [SELECT Id, Name, Company__c, Company__r.OwnerId, OwnerId FROM Domain__c WHERE Id = :noUK.Id];
        domainToUpdate.Company__c = acct.Id;
        domainToUpdate.OwnerId = acct.OwnerId;
        update domainToUpdate;
        
        System.debug(domainToUpdate);
        System.debug('NOW noUK domain Company__c is ' + domainToUpdate.Company__c);
        System.debug('NOW noUK domain ownerID is ' + domainToUpdate.OwnerId);


        System.assertEquals(UKDom.UK_Team__c, True);
        System.assertEquals(noUK.UK_Team__c, FALSE);
        System.assertEquals(domainToUpdate.OwnerId, acct.OwnerId);

    	}
    
}