public void GetLatestCompanyProducers_ReturnsCompaniesOnly()
{
using (DatabaseWrapper database = new DatabaseWrapper())
{
ModelHelper helper = new ModelHelper(database.Model);
// Arrange
Scheme scheme1 = helper.CreateScheme();
MemberUpload memberUpload1 = helper.CreateMemberUpload(scheme1);
memberUpload1.ComplianceYear = 2015;
memberUpload1.IsSubmitted = true;
Weee.Tests.Core.Model.ProducerSubmission companyProducer1 = helper.CreateProducerAsCompany(memberUpload1, "AAAAAAA");
Weee.Tests.Core.Model.ProducerSubmission partnershipProducer = helper.CreateProducerAsPartnership(memberUpload1, "PPP1");
Weee.Tests.Core.Model.ProducerSubmission soleTraderProducer = helper.CreateProducerAsSoleTrader(memberUpload1, "SSS1");
Weee.Tests.Core.Model.ProducerSubmission companyProducer2 = helper.CreateProducerAsCompany(memberUpload1, "AAAAAAB");
database.Model.SaveChanges();
// Act
var result = ProducerQuerySet(database.WeeeContext).GetLatestCompanyProducers();
// Assert
Assert.Contains(result, p => p.Id == companyProducer1.Id);
Assert.Contains(result, p => p.Id == companyProducer2.Id);
Assert.DoesNotContain(result, p => p.Id == soleTraderProducer.Id);
Assert.DoesNotContain(result, p => p.Id == partnershipProducer.Id);
}
}
