From 18565f4a85c46c366c82d0fabe8d717e2635d1cf Mon Sep 17 00:00:00 2001
From: Gaetan Craig-Riou <gaetan.riou@gmail.com>
Date: Mon, 14 Aug 2023 15:51:04 +1000
Subject: [PATCH] Add all the enterpise attributes supported by DFC connector

---
 .../app/services/enterprise_builder.rb         | 13 +++++++------
 .../spec/services/enterprise_builder_spec.rb   | 18 ++++++++++++++++++
 2 files changed, 25 insertions(+), 6 deletions(-)

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)