diff --git a/engines/dfc_provider/app/services/enterprise_builder.rb b/engines/dfc_provider/app/services/enterprise_builder.rb index a8327d97da..742d8d8f4f 100644 --- a/engines/dfc_provider/app/services/enterprise_builder.rb +++ b/engines/dfc_provider/app/services/enterprise_builder.rb @@ -7,11 +7,12 @@ class EnterpriseBuilder < DfcBuilder supplied_products = catalog_items.map(&:product) DataFoodConsortium::Connector::Enterprise.new( - enterprise.name - ).tap do |e| - e.semanticId = urls.enterprise_url(enterprise.id) - e.suppliedProducts = supplied_products - e.catalogItems = catalog_items - end + urls.enterprise_url(enterprise.id), + name: enterprise.name, + description: enterprise.description, + vatNumber: enterprise.abn, + suppliedProducts: supplied_products, + catalogItems: catalog_items + ) end end diff --git a/engines/dfc_provider/spec/services/enterprise_builder_spec.rb b/engines/dfc_provider/spec/services/enterprise_builder_spec.rb index d6c7c2f3c8..18186ed431 100644 --- a/engines/dfc_provider/spec/services/enterprise_builder_spec.rb +++ b/engines/dfc_provider/spec/services/enterprise_builder_spec.rb @@ -16,6 +16,24 @@ describe EnterpriseBuilder do ) end + it "assigns a name" do + result = builder.enterprise(enterprise) + + expect(result.name).to eq(enterprise.name) + end + + it "assigns a description" do + result = builder.enterprise(enterprise) + + expect(result.description).to eq(enterprise.description) + end + + it "assigns a VAT Number (ABN in australia)" do + result = builder.enterprise(enterprise) + + expect(result.vatNumber).to eq(enterprise.abn) + end + it "assignes products" do result = builder.enterprise(enterprise)