import unittest

from app.App import App


class test_OptimizationVehicleRepo(unittest.TestCase):

	@classmethod
	def setUpClass(cls):
		App.init()

	def test_all(self):
		optimizationVehicles = App.repos.optimizationVehicleRepo.getAll()
		print(optimizationVehicles)
		self.assertGreater(len(optimizationVehicles), 0)

		optimization = App.repos.optimizationVehicleRepo.get(id=optimizationVehicles[0].id)
		self.assertIn(optimization, optimizationVehicles)