public function testGetTotalAmount() { $this->assertEquals(1500, $this->subclass->getTotalAmount()); /** @var Subclass $emptySubclass */ $emptySubclass = $this->getMock('Model\\WealthbotRebalancer\\Subclass', null); $this->assertEquals(0, $emptySubclass->getTotalAmount()); /** @var Subclass $subclassWithSecurity */ $subclassWithSecurity = $this->getMock('Model\\WealthbotRebalancer\\Subclass', null); $security = new Security(); $security->setAmount(200); $subclassWithSecurity->setSecurity($security); $this->assertEquals(200, $subclassWithSecurity->getTotalAmount()); /** @var Subclass $subclassWithMuni */ $subclassWithMuni = $this->getMock('Model\\WealthbotRebalancer\\Subclass', null); $muni = new Security(); $muni->setAmount(722); $subclassWithMuni->setMuniSecurity($muni); $this->assertEquals(722, $subclassWithMuni->getTotalAmount()); }