enable CardFortress in test database, #71513

[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..0f85ac4 100644 (file)
--- a/FS/FS/sales_pkg_class.pm
+++ b/FS/FS/sales_pkg_class.pm
@@ -1,10 +1,7 @@
 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 );
-use FS::sales;
-use FS::pkg_class;
 
 =head1 NAME
 
@@ -110,17 +107,24 @@ sub check {
   $self->SUPER::check;
 }
 
-sub pkg_class {
-  my $self = shift;
-  qsearchs('pkg_class', { 'classnum' => $self->classnum });
-}
-
 sub classname {
   my $self = shift;
   my $pkg_class = $self->pkg_class;
   $pkg_class ? $pkg_class->classname : '(no package class)';
 }
 
+sub cust_credit_search {
+  my $self = shift;
+  my $sales = $self->sales;
+  $sales->cust_credit_search(@_, commission_classnum => $self->classnum);
+}
+
+sub cust_bill_pkg_search {
+  my $self = shift;
+  my $sales = $self->sales;
+  $sales->cust_bill_pkg_search(@_, classnum => $self->classnum);
+}
+
 =back
 
 =head1 BUGS