From ae2c52edb4a9aabe940506af4cd458c21d4072b7 Mon Sep 17 00:00:00 2001
From: kosiakkatrina <54268893+kosiakkatrina@users.noreply.github.com>
Date: Mon, 18 Dec 2023 13:23:07 +0100
Subject: [PATCH] Update seeds (#2095)

---
 db/seeds.rb | 92 +++++++++++++++++++++++++++++++++++++++++++++--------
 1 file changed, 79 insertions(+), 13 deletions(-)

diff --git a/db/seeds.rb b/db/seeds.rb
index 76da9ad35..ad2172734 100644
--- a/db/seeds.rb
+++ b/db/seeds.rb
@@ -40,14 +40,14 @@ unless Rails.env.test?
     provider_type: "LA",
   )
   managing_agent1 = Organisation.find_or_create_by!(
-    name: "Managing Agent 1",
+    name: "Managing Agent 1 (PRP)",
     address_line1: "2 Marsham Street",
     address_line2: "London",
     postcode: "SW1P 4DF",
     holds_own_stock: true,
     other_stock_owners: "None",
     managing_agents_label: "None",
-    provider_type: "LA",
+    provider_type: "PRP",
   )
   managing_agent2 = Organisation.find_or_create_by!(
     name: "Managing Agent 2",
@@ -89,11 +89,23 @@ unless Rails.env.test?
     provider_type: "LA",
   )
 
+  standalone_no_stock = Organisation.find_or_create_by!(
+    name: "Standalone No Stock 1 Ltd",
+    address_line1: "2 Marsham Street",
+    address_line2: "London",
+    postcode: "SW1P 4DF",
+    holds_own_stock: false,
+    other_stock_owners: "None",
+    managing_agents_label: "None",
+    provider_type: "LA",
+  )
+
   User.find_or_create_by!(
     name: "Provider Owns Stock",
     email: "provider.owner1@example.com",
     organisation: standalone_owns_stock,
     role: "data_provider",
+    is_dpo: true,
   ) do |user|
     user.password = "password"
     user.confirmed_at = Time.zone.now
@@ -105,31 +117,23 @@ unless Rails.env.test?
     email: "coordinator.owner1@example.com",
     organisation: standalone_owns_stock,
     role: "data_coordinator",
+    is_dpo: true,
   ) do |user|
     user.password = "password"
     user.confirmed_at = Time.zone.now
     create_data_protection_confirmation(user)
   end
 
-  standalone_no_stock = Organisation.find_or_create_by!(
-    name: "Standalone No Stock 1 Ltd",
-    address_line1: "2 Marsham Street",
-    address_line2: "London",
-    postcode: "SW1P 4DF",
-    holds_own_stock: false,
-    other_stock_owners: "None",
-    managing_agents_label: "None",
-    provider_type: "LA",
-  )
-
   User.find_or_create_by!(
     name: "Provider No Stock",
     email: "provider.nostock@example.com",
     organisation: standalone_no_stock,
     role: "data_provider",
+    is_dpo: true,
   ) do |user|
     user.password = "password"
     user.confirmed_at = Time.zone.now
+    create_data_protection_confirmation(user)
   end
 
   User.find_or_create_by!(
@@ -142,6 +146,54 @@ unless Rails.env.test?
     user.confirmed_at = Time.zone.now
   end
 
+  User.find_or_create_by!(
+    name: "Stock owner 1",
+    email: "stock_owner1_dpo@example.com",
+    organisation: stock_owner1,
+    role: "data_coordinator",
+    is_dpo: true,
+  ) do |user|
+    user.password = "password"
+    user.confirmed_at = Time.zone.now
+    create_data_protection_confirmation(user)
+  end
+
+  User.find_or_create_by!(
+    name: "Stock owner 2",
+    email: "stock_owner2_dpo@example.com",
+    organisation: stock_owner2,
+    role: "data_coordinator",
+    is_dpo: true,
+  ) do |user|
+    user.password = "password"
+    user.confirmed_at = Time.zone.now
+    create_data_protection_confirmation(user)
+  end
+
+  User.find_or_create_by!(
+    name: "Managing agent 1",
+    email: "managing_agent1_dpo@example.com",
+    organisation: managing_agent1,
+    role: "data_coordinator",
+    is_dpo: true,
+  ) do |user|
+    user.password = "password"
+    user.confirmed_at = Time.zone.now
+    create_data_protection_confirmation(user)
+  end
+
+  User.find_or_create_by!(
+    name: "Managing agent 2",
+    email: "managing_agent2_dpo@example.com",
+    organisation: managing_agent2,
+    role: "data_coordinator",
+    is_dpo: true,
+  ) do |user|
+    user.password = "password"
+    user.confirmed_at = Time.zone.now
+    create_data_protection_confirmation(user)
+  end
+
   OrganisationRelationship.find_or_create_by!(
     parent_organisation: stock_owner1,
     child_organisation: org,
@@ -175,6 +227,7 @@ unless Rails.env.test?
       email: "coordinator@example.com",
       organisation: org,
       role: "data_coordinator",
+      is_dpo: true,
     ) do |user|
       user.password = "password"
       user.confirmed_at = Time.zone.now
@@ -187,6 +240,7 @@ unless Rails.env.test?
       email: "support@example.com",
       organisation: org,
       role: "support",
+      is_dpo: true,
     ) do |user|
       user.password = "password"
       user.confirmed_at = Time.zone.now
@@ -248,6 +302,18 @@ unless Rails.env.test?
     )
 
     pp "Seeded dummy FooBar LTD organisation"
+
+    User.find_or_create_by!(
+      name: "Dummy user",
+      email: "dummy_org@example.com",
+      organisation: dummy_org,
+      role: "data_provider",
+      is_dpo: true,
+    ) do |user|
+      user.password = "password"
+      user.confirmed_at = Time.zone.now
+      create_data_protection_confirmation(user)
+    end
   end
 
   if (Rails.env.development? || Rails.env.review?) && Scheme.count.zero?