skillbox/server/users/tests/mock_hep_data_factory.py

48 lines
1.4 KiB
Python

import json
import os
from datetime import datetime, timedelta
class MockResponse:
def __init__(self, status_code, data={}):
self.status_code = status_code
self.data = data
def json(self):
return self.data
## Setup json data
def make_orders_valid(orders):
for order in orders:
if 'created_at' in order:
yesterday = datetime.now() - timedelta(1)
order['created_at'] = datetime.strftime(yesterday, '%Y-%m-%d %H:%M:%S')
return orders
# Load data
dir_path = os.path.dirname(os.path.realpath(__file__))
with open('{}/mock_data/valid_teacher_orders.json'.format(dir_path), 'r') as file:
valid_teacher_order_data = file.read()
with open('{}/mock_data/valid_student_orders.json'.format(dir_path), 'r') as file:
valid_student_order_data = file.read()
with open('{}/mock_data/me_data.json'.format(dir_path), 'r') as file:
me_data = file.read()
with open('{}/mock_data/email_not_confirmed_me.json'.format(dir_path), 'r') as file:
not_confirmed_email_me_data = file.read()
ME_DATA = json.loads(me_data)
NOT_CONFIRMED_ME = json.loads(not_confirmed_email_me_data)
valid_teacher_orders = json.loads(valid_teacher_order_data)
VALID_TEACHERS_ORDERS = make_orders_valid(valid_teacher_orders)
valid_student_orders = json.loads(valid_student_order_data)
VALID_STUDENT_ORDERS = make_orders_valid(valid_student_orders)