def test_create_order(self):
from orders_manager.utils.generate_data_helper import populate_database
from orders_manager.roles import init_roles
init_roles()
populate_database(skipped_items=('orders', ))
order = Order.objects.create(**self.order_info)
self.assertEqual(12, len(order.code))
self.assertEqual(self.order_info.get('author_id'),
order.author.user.id)
self.assertEqual(self.order_info.get('client_id'), order.client.id)
self.assertEqual(len(self.order_info.get('client_children_id')),
len(order.client_children.all()))
self.assertEqual(self.order_info.get('celebrate_date'),
order.celebrate_date)
self.assertEqual(self.order_info.get('celebrate_time'),
order.celebrate_time)
self.assertEqual(self.order_info.get('celebrate_place'),
order.celebrate_place)
self.assertEqual(self.order_info.get('address'), order.address)
self.assertEqual(self.order_info.get('program_id'), order.program.id)
self.assertEqual(self.order_info.get('duration'), order.duration)
self.assertEqual(self.order_info.get('price'), order.price)
self.assertEqual(self.order_info.get('details'), order.details)
self.assertEqual(self.order_info.get('executor_comment'),
order.executor_comment)
self.assertEqual(self.order_info.get('discount_id'), order.discount.id)
self.assertEqual(self.order_info.get('total_price'), order.total_price)
self.assertEqual(self.order_info.get('total_price_with_discounts'),
order.total_price_with_discounts)
order_services = order.additional_services_executors.all()
for i in order_services:
print('dunya')
def test_generate_data_helper(self):
init_roles()
UserProfileGenerator().generate(UserProfileGenerator.MANAGER, num=2)
UserProfileGenerator().generate(UserProfileGenerator.ANIMATOR, num=3)
UserProfileGenerator().generate(UserProfileGenerator.PHOTOGRAPHER,
num=2)
ProgramGenerator().generate(5)
additional_services = AdditionalServicesGenerator().generate()
DiscountsGenerator().generate()
ClientGenerator().generate(5)
OrderGenerator().generate(num_events=10, num_days=45)
orders = Order.objects.all()
id_to_service = {}
for s in additional_services:
id_to_service.update({s.id: s})
for order in orders:
services_executors = order.additional_services_executors.all()
for line in services_executors:
self.assertTrue(line.executor.user.id in [
i.user.id for i in id_to_service[
line.additional_service.id].possible_executors.all()
])
self.assertTrue(len(order.program_executors.all()) > 0)
def test_generate_data_helper(self):
init_roles()
UserProfileGenerator().generate(UserProfileGenerator.MANAGER, num=2)
UserProfileGenerator().generate(UserProfileGenerator.ANIMATOR, num=3)
UserProfileGenerator().generate(UserProfileGenerator.PHOTOGRAPHER,
num=2)
ProgramGenerator().generate(5)
additional_services = AdditionalServicesGenerator().generate()
DiscountsGenerator().generate()
ClientGenerator().generate(5)
OrderGenerator().generate(num_events=10, num_days=45)
orders = Order.objects.all()
id_to_service = {}
for s in additional_services:
id_to_service.update({s.id: s})
for order in orders:
services_executors = order.additional_services_executors.all()
for line in services_executors:
self.assertTrue(
line.executor.user.id in [i.user.id for i in id_to_service[
line.additional_service.id].possible_executors.all()]
)
self.assertTrue(len(order.program_executors.all()) > 0)
def test_create_order(self):
from orders_manager.utils.generate_data_helper import populate_database
from orders_manager.roles import init_roles
init_roles()
populate_database(skipped_items=('orders',))
order = Order.objects.create(**self.order_info)
self.assertEqual(12, len(order.code))
self.assertEqual(self.order_info.get('author_id'), order.author.user.id)
self.assertEqual(self.order_info.get('client_id'), order.client.id)
self.assertEqual(len(self.order_info.get('client_children_id')),
len(order.client_children.all()))
self.assertEqual(self.order_info.get('celebrate_date'),
order.celebrate_date)
self.assertEqual(self.order_info.get('celebrate_time'),
order.celebrate_time)
self.assertEqual(self.order_info.get('celebrate_place'),
order.celebrate_place)
self.assertEqual(self.order_info.get('address'), order.address)
self.assertEqual(self.order_info.get('program_id'), order.program.id)
self.assertEqual(self.order_info.get('duration'), order.duration)
self.assertEqual(self.order_info.get('price'), order.price)
self.assertEqual(self.order_info.get('details'), order.details)
self.assertEqual(self.order_info.get('executor_comment'),
order.executor_comment)
self.assertEqual(self.order_info.get('discount_id'), order.discount.id)
self.assertEqual(self.order_info.get('total_price'), order.total_price)
self.assertEqual(self.order_info.get('total_price_with_discounts'),
order.total_price_with_discounts)
order_services = order.additional_services_executors.all()
for i in order_services:
print('dunya')
def get(self, request, *args, **kwargs):
from orders_manager.utils.generate_data_helper import populate_database
from orders_manager.roles import init_roles
init_roles()
User.objects.create_superuser('admin', '*****@*****.**',
'12345')
Discount.objects.create(name='Нет скидки', value=0)
populate_database()
return HttpResponse('Done')
def setUp(self):
from orders_manager.models import UserProfile
from orders_manager.roles import init_roles
init_roles()
gen_us = UserProfileGenerator()
gen_us.generate(UserProfileGenerator.ANIMATOR, 4)
user_profiles = UserProfile.objects.all()
user_profile = UserProfile.objects.first()
self.day_off_data = {
'user_id': user_profile.user.id,
'date': '2015-12-03',
'time_start': '10:00:00',
'time_end': '15:00:00'
}
def setUp(self):
from orders_manager.models import UserProfile
from orders_manager.roles import init_roles
init_roles()
gen_us = UserProfileGenerator()
gen_us.generate(UserProfileGenerator.ANIMATOR, 4)
user_profiles = UserProfile.objects.all()
user_profile = UserProfile.objects.first()
self.day_off_data = {
'user_id': user_profile.user.id,
'date': '2015-12-03',
'time_start': '10:00:00',
'time_end': '15:00:00'
}
def set_up(self):
from orders_manager.utils.generate_data_helper import populate_database
from orders_manager.roles import init_roles
init_roles()
populate_database()
def setUp(self):
init_roles()
us_gen = UserProfileGenerator()
us_gen.generate(role=UserProfileGenerator.ANIMATOR, num=5)
us_gen.generate(role=UserProfileGenerator.MANAGER, num=3)
us_gen.generate(role=UserProfileGenerator.PHOTOGRAPHER, num=1)
def set_up(self):
from orders_manager.utils.generate_data_helper import populate_database
from orders_manager.roles import init_roles
init_roles()
populate_database()
def setUp(self):
init_roles()
us_gen = UserProfileGenerator()
us_gen.generate(role=UserProfileGenerator.ANIMATOR, num=5)
us_gen.generate(role=UserProfileGenerator.MANAGER, num=3)
us_gen.generate(role=UserProfileGenerator.PHOTOGRAPHER, num=1)
