def setUp(self):
with app.app_context():
test_user1 = User(
"*****@*****.**",
"testpw",
'US/Central',
'sub_user1',
)
test_user2 = User(
"*****@*****.**",
"testpw",
'US/Central',
'sub_user2',
)
db.session.add(test_user1)
db.session.add(test_user2)
db.session.commit()
test_customer = Customer(
'test_id',
'test_cc',
'first_name',
'last_name',
test_user2.public_id,
)
db.session.add(test_customer)
db.session.commit()
self.test_id_wo_customer = test_user1.public_id
self.test_id_with_customer = test_user2.public_id
def setUpClass(cls):
with app.app_context():
db.drop_all()
db.create_all()
cls.test_dao = ScheduleDAO()
test_user = User(
"*****@*****.**",
"testpw",
'US/Central',
'scheduleputfactest',
)
test2_user = User(
"*****@*****.**",
"testpw",
'US/Central',
'scheduleputotheruser',
)
cls.test_uid = test_user.public_id
cls.other_uid = test2_user.public_id
db.session.add(test_user)
db.session.add(test2_user)
db.session.commit()
def setUpClass(cls):
with app.app_context():
db.drop_all()
db.create_all()
cls.test_client = app.test_client()
cls.test_dao = UserDAO()
scheduling = User(
"*****@*****.**",
"testpw",
'US/Central',
'scheduling_user',
)
scheduled = User(
"*****@*****.**",
"testpw",
'US/Central',
'scheduled_user',
)
cls.scheduling_uid = scheduling.public_id
cls.scheduled_uid = scheduled.public_id
db.session.add(scheduling)
db.session.add(scheduled)
db.session.commit()
User.query.filter_by(
public_id=cls.scheduled_uid
).update({
'sixty_min_price': 15
})
db.session.add(Submerchant(
cls.scheduled_uid,
'testaccountid',
'firstName',
'LastName',
'email',
datetime.utcnow() + timedelta(days=-365*20),
'address_street',
'address_locality',
'address_region',
'address_zip',
))
new_schedule = Schedule(
datetime.utcnow(),
datetime.utcnow() + timedelta(minutes=60),
cls.scheduled_uid,
'US/Central'
)
db.session.add(new_schedule)
db.session.commit()
cls.new_schedule_id = new_schedule.id
def setUpClass(cls):
with app.app_context():
db.drop_all()
db.create_all()
cls.test_dao = EventDAO()
cls.scheduling_user = User(
"*****@*****.**",
"testpw",
'US/Central',
'scheduling_user',
)
cls.scheduled_user = User(
"*****@*****.**",
"testpw",
'US/Central',
'scheduled_user',
)
db.session.add(cls.scheduling_user)
db.session.add(cls.scheduled_user)
db.session.commit()
cls.scheduled_user = User.query.filter_by(
public_id=cls.scheduled_user.public_id).first().public_id
cls.scheduling_user = User.query.filter_by(
public_id=cls.scheduling_user.public_id).first().public_id
User.query.filter_by(public_id=cls.scheduled_user).update(
{'sixty_min_price': 15})
db.session.add(
Submerchant(
cls.scheduled_user,
'testaccountid',
'firstName',
'LastName',
'email',
datetime.utcnow() + timedelta(days=-365 * 20),
'address_street',
'address_locality',
'address_region',
'address_zip',
))
db.session.commit()
new_schedule = Schedule(
datetime.utcnow().replace(hour=0) + timedelta(days=1),
datetime.utcnow().replace(hour=12) + timedelta(days=1),
cls.scheduled_user, 'US/Central')
db.session.add(new_schedule)
db.session.commit()
def setUp(self):
with app.app_context():
test_user = User(
'*****@*****.**',
'testpw',
'US/Central',
'test1',
)
test_user2 = User(
'*****@*****.**',
'testpw',
'US/Central',
'test2',
)
self.test_uid = test_user.public_id
self.test_uid2 = test_user2.public_id
self.test_user_email = test_user.email
db.session.add(test_user)
db.session.add(test_user2)
db.session.commit()
test_customer = Customer(
'test_customer_id',
'cc_token',
'first_name',
'last_name',
self.test_uid,
)
test_customer2 = Customer(
'test_customer_id2',
'cc_token2',
'first_name',
'last_name',
self.test_uid2,
)
test_sub = Subscription(
test_customer2.public_id,
self.test_uid2,
'UUID-id'
)
self.sub_id = test_sub.public_id
db.session.add(test_customer)
db.session.add(test_customer2)
db.session.commit()
db.session.add(test_sub)
db.session.commit()
def create_new_user(self,
email,
password,
local_tz,
username,
*,
skip_commit=False):
"""
Handles the creation of a new user.
:param str email: The user's email to be registered with the account.
:param str password: The desired password (not yet hashed).
:param str local_tz: The local TZ of the user
:param str username: The desired username for the user.
:param bool skip_commit: An optional flag used to indicate if we want
to create the object and add to DB delta, but not commit it just yet.
Used mostly for facade methods which roll back if the event doesn't
fully complete (transaction doesn't finalize, &c.).
:rtype: UserTable
:return: The newly created user table.
"""
new_user = User(
email,
password,
local_tz,
username,
)
exec_and_commit(db.session.add, new_user, skip_commit=skip_commit)
return new_user
def setUpClass(cls):
with app.app_context():
db.drop_all()
db.create_all()
cls.scheduling_user = User(
"*****@*****.**",
"testpw",
'US/Central',
'scheduling_user',
)
cls.scheduled_user = User(
"*****@*****.**",
"testpw",
'US/Central',
'scheduled_user',
)
db.session.add(cls.scheduling_user)
db.session.add(cls.scheduled_user)
db.session.commit()
cls.scheduled_user = User.query.filter_by(
public_id=cls.scheduled_user.public_id).first().public_id
cls.scheduling_user = User.query.filter_by(
public_id=cls.scheduling_user.public_id).first().public_id
db.session.add(
Submerchant(
cls.scheduled_user,
'testaccountid',
'firstName',
'LastName',
'email',
datetime.utcnow() + timedelta(days=-365 * 20),
'address_street',
'address_locality',
'address_region',
'address_zip',
))
db.session.commit()
def setUpClass(cls):
with app.app_context():
db.drop_all()
db.create_all()
cls.test_client = app.test_client()
cls.test_dao = UserDAO()
test_user = User(
"*****@*****.**",
"testpw",
'US/Central',
'test-verify',
)
test_user2 = User(
"*****@*****.**",
"testpw",
'US/Central',
'test-verify2',
)
test_user3 = User(
"*****@*****.**",
"testpw",
'US/Central',
'test-verify3',
)
cls.test_user_token = test_user.verify_token
cls.test_uid = test_user.public_id
cls.test_user2_token = test_user2.verify_token
cls.test_uid2 = test_user2.public_id
cls.test_user_email3 = test_user3.email
cls.test_uid3 = test_user3.public_id
db.session.add(test_user)
db.session.add(test_user2)
db.session.add(test_user3)
db.session.commit()
def setUpClass(cls):
with app.app_context():
db.drop_all()
db.create_all()
cls.test_client = app.test_client()
cls.test_dao = UserDAO()
test_user = User(
"*****@*****.**",
"testpw",
'US/Central',
'test',
)
deleted_user = User(
"*****@*****.**",
"testpassword",
'US/Central',
'delete-test',
)
deleted_user.is_deleted = True
user_to_delete = User(
"*****@*****.**",
"testpw",
'US/Central',
'user-to-delete',
)
cls.test_uid = test_user.public_id
cls.deleted_uid = deleted_user.public_id
cls.to_delete_uid = user_to_delete.public_id
db.session.add(MasterMerchant('clarksoftware'))
db.session.add(test_user)
db.session.add(deleted_user)
db.session.add(user_to_delete)
db.session.commit()
def setUpClass(cls):
with app.app_context():
db.create_all()
test_user = User(
"*****@*****.**",
"testpw",
'US/Central',
'passwordtest',
)
cls.test_uid = test_user.public_id
db.session.add(test_user)
db.session.commit()
def setUp(self):
with app.app_context():
test_user = User(
'*****@*****.**',
'testpw',
'US/Central',
'test6',
)
self.test_uid = test_user.public_id
self.test_user_email = test_user.email
db.session.add(test_user)
db.session.commit()
def setUp(self):
with app.app_context():
test_user = User(
"*****@*****.**",
"testpw",
'US/Central',
'scheduling_usersubmerch',
)
db.session.add(test_user)
db.session.commit()
self.test_id = test_user.public_id
self.test_email = test_user.email
def setUpClass(cls):
with app.app_context():
db.drop_all()
db.create_all()
cls.test_facade = UserFacade()
test_user = User(
"*****@*****.**",
"testpw",
'US/Central',
'test1',
)
deleted_user = User(
"*****@*****.**",
"testpassword",
'US/Central',
'test2',
)
deleted_user.is_deleted = True
user_to_delete = User(
"*****@*****.**",
"testpw",
'US/Central',
'test3'
)
cls.test_uid = test_user.public_id
cls.deleted_uid = deleted_user.public_id
cls.to_delete_uid = user_to_delete.public_id
db.session.add(test_user)
db.session.add(deleted_user)
db.session.add(user_to_delete)
db.session.commit()
def setUpClass(cls):
with app.app_context():
db.drop_all()
db.create_all()
cls.test_facade = CustomerFacade()
test_user = User(
"*****@*****.**",
"testpw",
'US/Central',
'test1',
)
cls.test_uid = test_user.public_id
db.session.add(test_user)
db.session.commit()
def setUpClass(cls):
with app.app_context():
db.drop_all()
db.create_all()
cls.test_client = app.test_client()
cls.test_dao = UserDAO()
test_user = User(
'*****@*****.**',
'testpw',
'US/Central',
'auth_test_user',
)
db.session.add(test_user)
db.session.commit()
cls.scheduled_uid = test_user.public_id
def test_get_schedules(self):
with app.app_context():
test_user = User(
"*****@*****.**",
"testpw",
'US/Central',
'splitSchedules',
)
db.session.add(test_user)
db.session.commit()
sched1 = Schedule(
datetime(12,
datetime.today().month,
datetime.today().day, 0, 00, 00),
datetime(12,
datetime.today().month,
datetime.today().day, 1, 00, 0), test_user.public_id,
test_user.local_tz)
sched2 = Schedule(
datetime(12,
datetime.today().month,
datetime.today().day, 2, 00, 0),
datetime(12,
datetime.today().month,
datetime.today().day + 1, 3, 00, 00),
test_user.public_id, test_user.local_tz)
db.session.add(sched1)
db.session.add(sched2)
db.session.commit()
response = self.test_client.get('/schedules/{0}'.format(
test_user.public_id))
self.assertEqual(response.status_code, HTTPStatus.OK)
self.assertIsNotNone(response)
schedules = json.loads(response.data.decode('utf-8'))['schedules']
self.assertEqual(len(schedules), 2)
def setUpClass(cls):
with app.app_context():
db.drop_all()
db.create_all()
cls.test_client = app.test_client()
cls.test_dao = UserDAO()
test_user = User(
'*****@*****.**',
'testpw',
'US/Central',
'test5',
)
cls.test_user_token = test_user.verify_token
cls.test_uid = test_user.public_id
db.session.add(test_user)
db.session.commit()
def setUpClass(cls):
with app.app_context():
app.config['TESTING'] = True
db.drop_all()
db.create_all()
cls.test_client = app.test_client()
cls.test_dao = UserDAO()
test_user = User(
"*****@*****.**",
"testpw",
'US/Central',
'test',
)
cls.test_uid = test_user.public_id
db.session.add(test_user)
db.session.commit()
cls.schedule_ids = []
for i in range(10):
start_time = datetime.utcnow()
end_time = datetime.utcnow() + timedelta(minutes=60)
if i >= 7:
start_time = datetime.utcnow() + timedelta(minutes=120)
end_time = datetime.utcnow() + timedelta(minutes=180)
new_schedule = Schedule(start_time, end_time, cls.test_uid,
'US/Central')
db.session.add(new_schedule)
db.session.commit()
cls.schedule_ids.append(new_schedule.public_id)
def setUpClass(cls):
with app.app_context():
db.drop_all()
db.create_all()
cls.scheduling_user = User(
"*****@*****.**",
"testpw",
'US/Central',
'scheduling_user',
)
cls.scheduled_user = User(
"*****@*****.**",
"testpw",
'US/Central',
'scheduled_user',
)
db.session.add(cls.scheduling_user)
db.session.add(cls.scheduled_user)
new_default_address = Address(
cls.scheduling_user.public_id,
'test-first',
'test-last',
'test-street-address',
'test-locality',
'test-region',
'53703',
'US',
is_default=True,
)
db.session.add(new_default_address)
db.session.commit()
db.session.add(
Submerchant(
cls.scheduled_user.public_id,
'testaccountid',
'firstName',
'LastName',
'email',
datetime.utcnow() + timedelta(days=-365 * 20),
'address_street',
'address_locality',
'address_region',
'address_zip',
))
new_schedule = Schedule(
datetime.utcnow(),
datetime.utcnow() + timedelta(minutes=3 * 60),
cls.scheduled_user.public_id, 'US/Central')
db.session.add(new_schedule)
db.session.commit()
cls.scheduled_user = User.query.filter_by(
public_id=cls.scheduled_user.public_id).first().public_id
cls.scheduling_user = User.query.filter_by(
public_id=cls.scheduling_user.public_id).first().public_id
User.query.filter_by(public_id=cls.scheduled_user).update(
{'sixty_min_price': 15})
db.session.commit()
def setUpClass(cls):
with app.app_context():
db.drop_all()
db.create_all()
cls.test_client = app.test_client()
cls.test_dao = UserDAO()
scheduling = User(
"*****@*****.**",
"testpw",
'UTC',
'scheduling_user1',
)
scheduled = User(
"*****@*****.**",
"testpw",
'UTC',
'scheduled_user2',
)
cls.scheduling_uid = scheduling.public_id
cls.scheduled_uid = scheduled.public_id
db.session.add(scheduling)
db.session.add(scheduled)
db.session.commit()
default_address = Address(
cls.scheduling_uid,
'first-default',
'last-default',
'street address1',
'Madison',
'WI',
'53703-def',
'US',
is_default=True,
extended_address=None
)
db.session.add(default_address)
db.session.commit()
User.query.filter_by(
public_id=cls.scheduled_uid
).update({
'sixty_min_price': 15
})
db.session.commit()
db.session.add(Submerchant(
cls.scheduled_uid,
'3bX9oSVaocidZMxWGGMybj',
'firstName',
'LastName',
'email',
datetime.utcnow() + timedelta(days=-365*20),
'address_street',
'address_locality',
'address_region',
'address_zip',
))
new_schedule = Schedule(
datetime.utcnow(),
datetime.utcnow() + timedelta(minutes=60),
cls.scheduled_uid,
'UTC'
)
db.session.add(new_schedule)
db.session.commit()
cls.new_schedule_id = new_schedule.id
