This commit is contained in:
Eoin 2017-09-28 08:02:26 +00:00 committed by GitHub
commit c2576f0909
2 changed files with 2 additions and 0 deletions

View File

@ -83,6 +83,7 @@ class Trade(Base):
self.close_profit = profit
self.close_date = datetime.utcnow()
self.open_order_id = order_id
self.is_open = False
# Flush changes
Trade.session.flush()

View File

@ -22,6 +22,7 @@ class TestTrade(unittest.TestCase):
self.assertEqual(trade.close_rate, 1.0)
self.assertEqual(trade.close_profit, profit)
self.assertIsNotNone(trade.close_date)
self.assertFalse(trade.is_open)
if __name__ == '__main__':