diff --git a/app/lib/Firefly/Helper/Controllers/Budget.php b/app/lib/Firefly/Helper/Controllers/Budget.php index d2a9884718..4cab145e0f 100644 --- a/app/lib/Firefly/Helper/Controllers/Budget.php +++ b/app/lib/Firefly/Helper/Controllers/Budget.php @@ -1,10 +1,4 @@ _accounts = $this->mock('Firefly\Storage\Account\AccountRepositoryInterface'); + $this->_categories = $this->mock('Firefly\Storage\Category\CategoryRepositoryInterface'); + } + + public function tearDown() + { + Mockery::close(); + } + + public function testBeneficiaries() + { + $beneficiary = f::create('Account'); + + $this->_accounts->shouldReceive('getBeneficiaries')->once()->andReturn([$beneficiary]); + $this->action('GET', 'JsonController@beneficiaries'); + $this->assertResponseOk(); + } + + public function testCategories() + { + $category = f::create('Category'); + $this->_categories->shouldReceive('get')->once()->andReturn([$category]); + $this->action('GET', 'JsonController@categories'); + $this->assertResponseOk(); + } +} \ No newline at end of file