Bug 7180: Add UT for C4::Budgets::GetBudgetByCode

[srvgit] / t / db_dependent / Budgets.t

diff --git a/t/db_dependent/Budgets.t b/t/db_dependent/Budgets.t
index bfb910b..54beddf 100755 (executable)
--- a/t/db_dependent/Budgets.t
+++ b/t/db_dependent/Budgets.t
@@ -1,6 +1,6 @@
 use strict;
 use warnings;
-use Test::More tests=>20;
+use Test::More tests => 22;
 
 BEGIN {use_ok('C4::Budgets') }
 use C4::Dates;
@@ -94,6 +94,7 @@ ok($budget_id=AddBudget(
 
 my $budget;
 ok($budget=GetBudget($budget_id) ,"GetBudget OK");
+$budget_id = $budget->{budget_id};
 $$budget{budget_permission}=1;
 ok($mod_status=ModBudget($budget),"ModBudget OK");
 ok(GetBudgets()>0,
@@ -109,6 +110,11 @@ ok(GetBudgets({budget_period_id=>GetBudgetPeriod($bpid)->{budget_period_id}},[{"
 my $budget_name = GetBudgetName( $budget_id );
 is($budget_name, $budget->{budget_name}, "Test the GetBudgetName routine");
 
+my $budget_code = $budget->{budget_code};
+my $budget_by_code = GetBudgetByCode( $budget_code );
+is($budget_by_code->{budget_id}, $budget_id, "GetBudgetByCode, check id");
+is($budget_by_code->{budget_notes}, 'This is a note', "GetBudgetByCode, check notes");
+
 my $second_budget_id;
 ok($second_budget_id=AddBudget(
                         {   budget_code         => "ZZZZ",