pull/31/merge
Eoin 9 years ago committed by GitHub
commit c2576f0909

@ -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()

@ -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__':

Loading…
Cancel
Save