def test_adjust_sell_amount_without_enable(self):
follower = XueQiuFollower()
mock_user = mock.MagicMock()
follower._users = [mock_user]
follower._adjust_sell = False
amount = follower._adjust_sell_amount("169101", 1000)
self.assertEqual(amount, amount)
def test_adjust_sell_amount_without_enable(self):
follower = XueQiuFollower()
mock_user = mock.MagicMock()
follower._users = [mock_user]
follower._adjust_sell = False
amount = follower._adjust_sell_amount("169101", 1000)
self.assertEqual(amount, amount)
def test_adjust_sell_amount(self):
follower = XueQiuFollower()
mock_user = mock.MagicMock()
follower._users = [mock_user]
mock_user.position = TEST_POSITION
follower._adjust_sell = True
test_cases = [
("169101", 600, 600),
("169101", 700, 600),
("000000", 100, 100),
("sh169101", 700, 600),
]
for stock_code, sell_amount, excepted_amount in test_cases:
amount = follower._adjust_sell_amount(stock_code, sell_amount)
self.assertEqual(amount, excepted_amount)
def test_adjust_sell_amount(self):
follower = XueQiuFollower()
mock_user = mock.MagicMock()
follower._users = [mock_user]
mock_user.position = TEST_POSITION
follower._adjust_sell = True
test_cases = [
("169101", 600, 600),
("169101", 700, 600),
("000000", 100, 100),
("sh169101", 700, 600),
]
for stock_code, sell_amount, excepted_amount in test_cases:
amount = follower._adjust_sell_amount(stock_code, sell_amount)
self.assertEqual(amount, excepted_amount)