Coverage for tests/django_tests/db/simulations/test_simulation_queue.py: 100%

1from datetime import datetime 

2 

3from pytz import UTC 

4 

5from django_napse.core.models import Bot, Controller, EmptyBotConfig, EmptyStrategy, LBOPlugin, MBPPlugin, SBVPlugin, SinglePairArchitecture 

6from django_napse.simulations.models import SimulationQueue 

7from django_napse.utils.model_test_case import ModelTestCase 

8 

9""" 

10python tests/test_app/manage.py test tests.django_tests.db.simulations.test_simulation_queue -v2 --keepdb --parallel 

11""" 

12 

13 

14class SimulationQueueTestCase: 

15 model = SimulationQueue 

16 

17 def simple_create(self): 

18 config = EmptyBotConfig.objects.create(space=self.space, settings={"empty": True}) 

19 architecture = SinglePairArchitecture.objects.create( 

20 constants={ 

21 "controller": Controller.get( 

22 exchange_account=self.exchange_account, 

23 base="BTC", 

24 quote="USDT", 

25 interval="1d", 

26 ), 

27 }, 

28 ) 

29 strategy = EmptyStrategy.objects.create(config=config, architecture=architecture) 

30 MBPPlugin.objects.create(strategy=strategy) 

31 LBOPlugin.objects.create(strategy=strategy) 

32 SBVPlugin.objects.create(strategy=strategy) 

33 bot = Bot.objects.create(name="Test Bot", strategy=strategy) 

34 return SimulationQueue.objects.create( 

35 space=self.space, 

36 bot=bot, 

37 start_date=datetime(2020, 1, 1, tzinfo=UTC), 

38 end_date=datetime(2020, 1, 5, tzinfo=UTC), 

39 investments={"USDT": 1000}, 

40 ) 

41 

42 def test_quick_simulation(self): 

43 simulation_queue = self.simple_create() 

44 simulation_queue.run_quick_simulation(verbose=False) 

45 

46 def test_irl_simulation(self): 

47 simulation_queue = self.simple_create() 

48 simulation_queue.run_irl_simulation(verbose=False) 

49 

50 

51class SimulationQueueBINANCETestCase(SimulationQueueTestCase, ModelTestCase): 

52 exchange = "BINANCE"