Admiral/admiral-worker/tests/test_repos/test_OptimizationRepo.py
2025-06-24 14:22:50 +02:00

36 lines
1.0 KiB
Python

import unittest
from app.App import App
from core.domain.optimization.OptimizationState import OptimizationState
class test_OptimizationRepo(unittest.TestCase):
@classmethod
def setUpClass(cls):
App.init()
def test_all(self):
optimizations = App.repos.optimizationRepo.getAll()
self.assertGreater(len(optimizations), 0)
optimization = App.repos.optimizationRepo.get(id=optimizations[0].id)
self.assertIn(optimization, optimizations)
def test_getWithState(self):
for opti in App.repos.optimizationRepo.getAll():
print(opti.id, opti.state)
App.repos.optimizationRepo.setState(opti.id, toState=OptimizationState.COMPLETED)
print("=====================")
for opti in App.repos.optimizationRepo.getAll():
print(opti.id, opti.state)
def test_updateFirst(self):
opti = App.repos.optimizationRepo.updateFirst(fromState=OptimizationState.CREATED, toState=OptimizationState.ACTIVE)
print(opti)
def test_getLatest(self):
opt = App.repos.optimizationRepo.getLatestConfirmedByPosta(posta = 9710)
print(opt)