71513: Card tokenization [v3 test tweak]

[freeside.git] / FS / FS / sales_pkg_class.pm

diff --git a/FS/FS/sales_pkg_class.pm b/FS/FS/sales_pkg_class.pm
index 90867a8..84ba683 100644 (file)
--- a/FS/FS/sales_pkg_class.pm
+++ b/FS/FS/sales_pkg_class.pm
@@ -1,5 +1,5 @@
 package FS::sales_pkg_class;
-use base qw( FS::Record );
+use base qw( FS::Commission_Mixin FS::Record );
 
 use strict;
 use FS::Record qw( qsearchs ); # qsearch qsearchs );
@@ -121,6 +121,18 @@ sub classname {
   $pkg_class ? $pkg_class->classname : '(no package class)';
 }
 
+sub cust_credit_search {
+  my $self = shift;
+  my $sales = FS::sales->by_key($self->salesnum);
+  $sales->cust_credit_search(@_, commission_classnum => $self->classnum);
+}
+
+sub cust_bill_pkg_search {
+  my $self = shift;
+  my $sales = FS::sales->by_key($self->salesnum);
+  $sales->cust_bill_pkg_search(@_, classnum => $self->classnum);
+}
+
 =back
 
 =head1 BUGS