From 5cf51600ada0462a8409a139b6c8325f3c766ae4 Mon Sep 17 00:00:00 2001
From: Emmi Takkinen <emmi.takkinen@outlook.com>
Date: Thu, 20 Feb 2020 12:45:27 +0200
Subject: [PATCH] Bug 17499: Define koha_object(s)_class where needed

This patch adds koha_object(s)_class for needed modules
to fix issues with tests.

Sponsored-by: Koha-Suomi Oy
Signed-off-by: David Nind <david@davidnind.com>
Signed-off-by: Martin Renvoize <martin.renvoize@ptfs-europe.com>
---
 Koha/Schema/Result/BorrowerMessagePreference.pm          | 8 ++++++++
 Koha/Schema/Result/BorrowerMessageTransportPreference.pm | 8 ++++++++
 Koha/Schema/Result/MessageAttribute.pm                   | 8 ++++++++
 Koha/Schema/Result/MessageTransport.pm                   | 8 ++++++++
 Koha/Schema/Result/MessageTransportType.pm               | 8 ++++++++
 5 files changed, 40 insertions(+)

diff --git a/Koha/Schema/Result/BorrowerMessagePreference.pm b/Koha/Schema/Result/BorrowerMessagePreference.pm
index 3069df50f0..34f21f9e12 100644
--- a/Koha/Schema/Result/BorrowerMessagePreference.pm
+++ b/Koha/Schema/Result/BorrowerMessagePreference.pm
@@ -194,4 +194,12 @@ __PACKAGE__->many_to_many(
 
 
 # You can replace this text with custom content, and it will be preserved on regeneration
+
+sub koha_object_class {
+  'Koha::Patron::Message::Preference';
+}
+sub koha_objects_class {
+  'Koha::Patron::Message::Preferences';
+}
+
 1;
diff --git a/Koha/Schema/Result/BorrowerMessageTransportPreference.pm b/Koha/Schema/Result/BorrowerMessageTransportPreference.pm
index 10f3f460d0..be3751c6da 100644
--- a/Koha/Schema/Result/BorrowerMessageTransportPreference.pm
+++ b/Koha/Schema/Result/BorrowerMessageTransportPreference.pm
@@ -112,4 +112,12 @@ __PACKAGE__->belongs_to(
 
 
 # You can replace this text with custom content, and it will be preserved on regeneration
+
+sub koha_object_class {
+  'Koha::Patron::Message::Transport::Preference';
+}
+sub koha_objects_class {
+  'Koha::Patron::Message::Transport::Preferences';
+}
+
 1;
diff --git a/Koha/Schema/Result/MessageAttribute.pm b/Koha/Schema/Result/MessageAttribute.pm
index 8fc01b3f7e..e1357ac2b2 100644
--- a/Koha/Schema/Result/MessageAttribute.pm
+++ b/Koha/Schema/Result/MessageAttribute.pm
@@ -117,4 +117,12 @@ __PACKAGE__->has_many(
 
 
 # You can replace this text with custom content, and it will be preserved on regeneration
+
+sub koha_object_class {
+  'Koha::Patron::Message::Attribute';
+}
+sub koha_objects_class {
+  'Koha::Patron::Message::Attributes';
+}
+
 1;
diff --git a/Koha/Schema/Result/MessageTransport.pm b/Koha/Schema/Result/MessageTransport.pm
index 82e8ab2805..ee7bc79450 100644
--- a/Koha/Schema/Result/MessageTransport.pm
+++ b/Koha/Schema/Result/MessageTransport.pm
@@ -134,4 +134,12 @@ __PACKAGE__->belongs_to(
 
 
 # You can replace this text with custom content, and it will be preserved on regeneration
+
+sub koha_object_class {
+  'Koha::Patron::Message::Transport';
+}
+sub koha_objects_class {
+  'Koha::Patron::Message::Transports';
+}
+
 1;
diff --git a/Koha/Schema/Result/MessageTransportType.pm b/Koha/Schema/Result/MessageTransportType.pm
index 775d9a250e..060336977f 100644
--- a/Koha/Schema/Result/MessageTransportType.pm
+++ b/Koha/Schema/Result/MessageTransportType.pm
@@ -155,4 +155,12 @@ __PACKAGE__->many_to_many(
 
 
 # You can replace this text with custom content, and it will be preserved on regeneration
+
+sub koha_object_class {
+  'Koha::Patron::Message::Transport::Type';
+}
+sub koha_objects_class {
+  'Koha::Patron::Message::Transport::Types';
+}
+
 1;
-- 
2.20.1