class PortfolioValueTest(unittest.TestCase): def fake_current_prices(self): return {'IBM': 140.0, 'HPQ': 32.0} def setUp(self): self.p = Portfolio() self.p.buy("IBM", 100, 120.0) self.p.buy("HPQ", 100, 30.0) self.p.current_prices = self.fake_current_prices def test_value(self): self.assertEqual(self.p.value(), 17200)
class PortfolioValueTest(unittest.TestCase): def setUp(self): # Save the real urllib, and install our fake. self.old_urllib = portfolio3.urllib portfolio3.urllib = FakeUrllib() self.p = Portfolio() self.p.buy("IBM", 100, 120.0) self.p.buy("HPQ", 100, 30.0) def test_value(self): self.assertEqual(self.p.value(), 17200) def tearDown(self): # Restore the real urllib. portfolio3.urllib = self.old_urllib
class PortfolioValueTest(unittest.TestCase): def setUp(self): # Save the real requests, and install our fake. self.old_requests = portfolio3.requests portfolio3.requests = FakeRequests() self.p = Portfolio() self.p.buy("IBM", 100, 120.0) self.p.buy("HPQ", 100, 30.0) def test_value(self): self.assertEqual(self.p.value(), 17200) def tearDown(self): # Restore the real requests. portfolio3.requests = self.old_requests
class PortfolioValueTest(unittest.TestCase): def setUp(self): self.p = Portfolio() self.p.buy("IBM", 100, 120.0) self.p.buy("HPQ", 100, 30.0) def test_value(self): # Create a mock urllib.urlopen. with mock.patch('urllib.urlopen') as urlopen: # When called, it will return this value: fake_yahoo = StringIO('"IBM",140\n"HPQ",32\n') urlopen.return_value = fake_yahoo # Run the test! self.assertEqual(self.p.value(), 17200) # We can ask the mock what its arguments were. urlopen.assert_called_with("http://finance.yahoo.com/d/quotes.csv" "?f=sl1&s=HPQ,IBM")
class PortfolioValueTest(unittest.TestCase): def setUp(self): self.p = Portfolio() self.p.buy("IBM", 100, 120.0) self.p.buy("HPQ", 100, 30.0) def test_value(self): # Create a mock requests.get. with mock.patch('portfolio3.requests.get') as req_get: # When called, it will return this value: req_get.return_value = SimpleNamespace( text='\nIBM,,,140\nHPQ,,,32\n') # Run the test! self.assertEqual(self.p.value(), 17200) # We can ask the mock what its arguments were. self.assertEqual(len(req_get.call_args_list), 1) called_url = req_get.call_args_list[0][0][0] self.assertIn("api.worldtradingdata.com/api/v1/stock", called_url) self.assertIn("symbol=HPQ,IBM", called_url)
class PortfolioValueTest(unittest.TestCase): def setUp(self): self.p = Portfolio() self.p.buy("IBM", 100, 120.0) self.p.buy("HPQ", 100, 30.0) def test_value(self): # Create a mock urllib.urlopen. with mock.patch('urllib.urlopen') as urlopen: # When called, it will return this value: fake_yahoo = StringIO('"IBM",140\n"HPQ",32\n') urlopen.return_value = fake_yahoo # Run the test! self.assertEqual(self.p.value(), 17200) # We can ask the mock what its arguments were. urlopen.assert_called_with( "http://finance.yahoo.com/d/quotes.csv" "?f=sl1&s=HPQ,IBM" )