def test_get_events(self):
"""Getting event test."""
package_members = self.prepare_package_members()
db.create_package(package_members['escrow'][0],
package_members['launcher'][0],
package_members['recipient'][0], 50000000, 100000000,
time.time(), None, None, None, None)
new_package_members = self.prepare_package_members()
db.create_package(new_package_members['escrow'][0],
new_package_members['launcher'][0],
new_package_members['recipient'][0], 50000000,
100000000, time.time(), None, None, None, None)
for user in new_package_members:
db.add_event(new_package_members['escrow'][0], user[0],
'new event', None)
events = db.get_package_events(package_members['escrow'][0])
self.assertEqual(
len(events), 1,
"expected 1 event for package: '{}', but '{}' got instead".format(
package_members['escrow'][0], len(events)))
events = db.get_package_events(new_package_members['escrow'][0])
self.assertEqual(
len(events), 5,
"expected 5 events for package: '{}', but '{}' got instead".format(
package_members['escrow'][0], len(events)))
def test_get_packages(self):
"""Getting packages test."""
for i in range(5):
package_members = self.prepare_package_members()
db.create_package(
package_members['escrow'][0], package_members['launcher'][0], package_members['recipient'][0],
'+490857461783', '+4904597863891', i * 10 ** 7, i * 2 * 10 ** 7, time.time(),
'Package description', '12.970686,77.595590', '41.156193,-8.637541',
'India Bengaluru', 'Spain Porto', '12.970686,77.595590', None)
packages = db.get_packages()
self.assertEqual(len(packages), 5, "expected 5 packages, {} got instead".format(len(packages)))
def test_get_packages(self):
"""Getting packages test."""
for i in range(5):
package_members = self.prepare_package_members()
db.create_package(package_members['escrow'][0],
package_members['launcher'][0],
package_members['recipient'][0], i * 10**7,
i * 2 * 10**7, time.time(), None, None, None,
None)
packages = db.get_packages()
self.assertEqual(
len(packages), 5,
"expected 5 packages, {} got instead".format(len(packages)))
def test_add_event(self):
"""Adding event test."""
package_members = self.prepare_package_members()
db.create_package(
package_members['escrow'][0], package_members['launcher'][0], package_members['recipient'][0],
'+490857461783', '+4904597863891', 50000000, 100000000, time.time(), 'Package description',
'12.970686,77.595590', '41.156193,-8.637541', 'India Bengaluru', 'Spain Porto', '12.970686,77.595590', None)
db.add_event(package_members['courier'][0], 'couriered', '12.970686,77.595590', package_members['escrow'][0])
events = db.get_package_events(package_members['escrow'][0])
self.assertEqual(len(events), 2, "2 events expected, but {} got instead".format(len(events)))
couriered_event = next((event for event in events if event['event_type'] == 'couriered'), None)
self.assertIsNotNone(couriered_event, "expected event with event_type: 'couriered', None got instead")
self.assertEqual(
couriered_event['user_pubkey'], package_members['courier'][0],
"expected event with user_pubkey: {}, but {} got instead".format(
package_members['courier'][0], couriered_event['user_pubkey']))
def create_mock_package_handler(
escrow_pubkey, launcher_pubkey, recipient_pubkey,
payment_buls, collateral_buls, deadline_timestamp):
"""
Create a mock package - for debug only.
---
:return:
"""
return {'status': 201, 'package': db.create_package(
escrow_pubkey, launcher_pubkey, recipient_pubkey, payment_buls, collateral_buls, deadline_timestamp,
'mock_setopts', 'mock_refund', 'mock merge', 'mock payment')}
def test_get_package(self):
"""Getting package test."""
package_members = self.prepare_package_members()
db.create_package(
package_members['escrow'][0], package_members['launcher'][0], package_members['recipient'][0],
'+490857461783', '+4904597863891', 50000000, 100000000, time.time(), 'Package description',
'12.970686,77.595590', '41.156193,-8.637541', 'India Bengaluru', 'Spain Porto', '12.970686,77.595590', None)
package = db.get_package(package_members['escrow'][0])
self.assertEqual(
package['escrow_pubkey'], package_members['escrow'][0],
"returned package with escrow_pubkey: {}, but {} expected".format(
package['escrow_pubkey'], package_members['escrow'][0]))
self.assertEqual(
package['launcher_pubkey'], package_members['launcher'][0],
"returned package with launcher_pubkey: {}, but {} expected".format(
package['launcher_pubkey'], package_members['launcher'][0]))
self.assertEqual(
package['recipient_pubkey'], package_members['recipient'][0],
"returned package with recipient_pubkey: {}, but {} expected".format(
package['recipient_pubkey'], package_members['recipient'][0]))
def test_get_package(self):
"""Getting package test."""
package_members = self.prepare_package_members()
db.create_package(package_members['escrow'][0],
package_members['launcher'][0],
package_members['recipient'][0], 50000000, 100000000,
time.time(), None, None, None, None)
package = db.get_package(package_members['escrow'][0])
self.assertEqual(
package['escrow_pubkey'], package_members['escrow'][0],
"returned package with escrow_pubkey: {}, but {} expected".format(
package['escrow_pubkey'], package_members['escrow'][0]))
self.assertEqual(
package['launcher_pubkey'], package_members['launcher'][0],
"returned package with launcher_pubkey: {}, but {} expected".
format(package['launcher_pubkey'], package_members['launcher'][0]))
self.assertEqual(
package['recipient_pubkey'], package_members['recipient'][0],
"returned package with recipient_pubkey: {}, but {} expected".
format(package['recipient_pubkey'],
package_members['recipient'][0]))
def test_get_escrow_events(self):
"""Getting escrow events test."""
packages_members = [self.prepare_package_members() for _ in range(3)]
for members in packages_members:
db.create_package(members['escrow'][0], members['launcher'][0], members['recipient'][0],
'+490857461783', '+4904597863891', 50000000, 100000000, time.time(),
'Package description', '12.970686,77.595590', '41.156193,-8.637541',
'India Bengaluru', 'Spain Porto', '12.970686,77.595590', None)
for members in packages_members:
events = db.get_package_events(members['escrow'][0])
self.assertEqual(len(events), 1,
"1 event expected for escrow: {}, but {} got instead".format(
members['escrow'][0], len(events)))
for index, members in enumerate(packages_members):
for _ in range(index):
db.add_event(members['courier'][0], 'new event', '12.970686,77.595590', members['escrow'][0])
for index, members in enumerate(packages_members):
events = db.get_package_events(members['escrow'][0])
self.assertEqual(len(events), index+1, "{} event expected for escrow: {}, but {} got instead".format(
index + 1, members['escrow'][0], len(events)))
def test_get_events(self):
"""Getting event test."""
package_members = self.prepare_package_members()
db.create_package(
package_members['escrow'][0], package_members['launcher'][0], package_members['recipient'][0],
'+490857461783', '+4904597863891', 50000000, 100000000, time.time(), 'Package description',
'12.970686,77.595590', '41.156193,-8.637541', 'India Bengaluru', 'Spain Porto', '12.970686,77.595590', None)
new_package_members = self.prepare_package_members()
db.create_package(
new_package_members['escrow'][0], new_package_members['launcher'][0], new_package_members['recipient'][0],
'+490857461783', '+4904597863891', 50000000, 100000000, time.time(), 'Package description',
'12.970686,77.595590', '41.156193,-8.637541', 'India Bengaluru', 'Spain Porto', '12.970686,77.595590', None)
for user in new_package_members:
db.add_event(new_package_members[user][0], 'new event',
'12.970686,77.595590', new_package_members['escrow'][0])
events = db.get_package_events(package_members['escrow'][0])
self.assertEqual(len(events), 1, "expected 1 event for package: '{}', but '{}' got instead".format(
package_members['escrow'][0], len(events)))
events = db.get_package_events(new_package_members['escrow'][0])
self.assertEqual(len(events), 5, "expected 5 events for package: '{}', but '{}' got instead".format(
package_members['escrow'][0], len(events)))
def prepare_escrow_handler(
user_pubkey, launcher_pubkey, courier_pubkey, recipient_pubkey,
payment_buls, collateral_buls, deadline_timestamp, location=None):
"""
Launch a package.
Use this call to create a new package for delivery.
---
:param user_pubkey: the escrow pubkey
:param launcher_pubkey:
:param courier_pubkey:
:param recipient_pubkey:
:param payment_buls:
:param collateral_buls:
:param deadline_timestamp:
:param location:
:return:
"""
package_details = paket_stellar.prepare_escrow(
user_pubkey, launcher_pubkey, courier_pubkey, recipient_pubkey,
payment_buls, collateral_buls, deadline_timestamp)
db.create_package(**dict(package_details, location=location))
return dict(status=201, **package_details)
def test_add_event(self):
"""Adding event test."""
package_members = self.prepare_package_members()
db.create_package(package_members['escrow'][0],
package_members['launcher'][0],
package_members['recipient'][0], 50000000, 100000000,
time.time(), None, None, None, None)
db.add_event(package_members['escrow'][0],
package_members['courier'][0], 'couriered', None)
events = db.get_package_events(package_members['escrow'][0])
self.assertEqual(
len(events), 2,
"2 events expected, but {} got instead".format(len(events)))
couriered_event = next(
(event for event in events if event['event_type'] == 'couriered'),
None)
self.assertIsNotNone(
couriered_event,
"expected event with event_type: 'couriered', None got instead")
self.assertEqual(
couriered_event['user_pubkey'], package_members['courier'][0],
"expected event with user_pubkey: {}, but {} got instead".format(
package_members['courier'][0], couriered_event['user_pubkey']))
def test_create_package(self):
"""Creating package test."""
package_members = self.prepare_package_members()
db.create_package(
package_members['escrow'][0], package_members['launcher'][0], package_members['recipient'][0],
'+490857461783', '+4904597863891', 50000000, 100000000, time.time(), 'Package description',
'12.970686,77.595590', '41.156193,-8.637541', 'India Bengaluru', 'Spain Porto', '12.970686,77.595590', None)
events = db.get_package_events(package_members['escrow'][0])
packages = db.get_packages(package_members['launcher'][0])
self.assertEqual(len(packages), 1,
"should be 1 created package, {} got instead".format(len(packages)))
self.assertEqual(len(events), 1,
"created package should contain 1 event, {} got instead".format(len(events)))
self.assertEqual(
packages[0]['escrow_pubkey'], package_members['escrow'][0],
"stored escrow_pubkey is: {}, but package was creating with: {}".format(
packages[0]['escrow_pubkey'], package_members['escrow'][0]))
self.assertEqual(
packages[0]['launcher_pubkey'], package_members['launcher'][0],
"stored launcher_pubkey is: {}, but package was creating with: {}".format(
packages[0]['launcher_pubkey'], package_members['launcher'][0]))
self.assertEqual(
packages[0]['recipient_pubkey'], package_members['recipient'][0],
"stored recipient_pubkey is: {}, but package was creating with: {}".format(
packages[0]['recipient_pubkey'], package_members['recipient'][0]))
self.assertEqual(
events[0]['event_type'], 'launched',
"created event '{}', but must be 'launched'".format(
events[0]['event_type']))
self.assertEqual(
packages[0]['launch_date'], events[0]['timestamp'],
"package has launch date {}, but was launched at {}".format(
packages[0]['launch_date'], events[0]['timestamp']))
self.assertEqual(
events[0]['user_pubkey'], package_members['launcher'][0],
"created event for user: {}, but must be for: {}".format(
events[0]['user_pubkey'], package_members['launcher'][0]))
def test_create_package(self):
"""Creating package test."""
package_members = self.prepare_package_members()
db.create_package(package_members['escrow'][0],
package_members['launcher'][0],
package_members['recipient'][0], 50000000, 100000000,
time.time(), None, None, None, None)
events = db.get_package_events(package_members['escrow'][0])
packages = db.get_packages(package_members['launcher'][0])
self.assertEqual(len(packages), 1, '')
self.assertEqual(len(events), 1, '')
self.assertEqual(
packages[0]['escrow_pubkey'], package_members['escrow'][0],
"stored escrow_pubkey is: {}, but package was creating with: {}".
format(packages[0]['escrow_pubkey'], package_members['escrow'][0]))
self.assertEqual(
packages[0]['launcher_pubkey'], package_members['launcher'][0],
"stored launcher_pubkey is: {}, but package was creating with: {}".
format(packages[0]['launcher_pubkey'],
package_members['launcher'][0]))
self.assertEqual(
packages[0]['recipient_pubkey'], package_members['recipient'][0],
"stored recipient_pubkey is: {}, but package was creating with: {}"
.format(packages[0]['recipient_pubkey'],
package_members['recipient'][0]))
self.assertEqual(
events[0]['event_type'], 'launched',
"created event '{}', but must be 'launched'".format(
events[0]['event_type']))
self.assertEqual(
packages[0]['launch_date'], events[0]['timestamp'],
"package has launch date {}, but was launched at {}".format(
packages[0]['launch_date'], events[0]['timestamp']))
self.assertEqual(
events[0]['user_pubkey'], package_members['launcher'][0],
"created event for user: {}, but must be for: {}".format(
events[0]['user_pubkey'], package_members['launcher'][0]))
def test_get_escrow_events(self):
"""Getting escrow events test."""
packages_members = [self.prepare_package_members() for _ in range(3)]
for members in packages_members:
db.create_package(members['escrow'][0], members['launcher'][0],
members['recipient'][0], 50000000, 100000000,
time.time(), None, None, None, None)
for members in packages_members:
events = db.get_package_events(members['escrow'][0])
self.assertEqual(
len(events), 1,
"1 event expected for escrow: {}, but {} got instead".format(
members['escrow'][0], len(events)))
for index, members in enumerate(packages_members):
for _ in range(index):
db.add_event(members['escrow'][0], members['courier'][0],
'couriered', None)
for index, members in enumerate(packages_members):
events = db.get_package_events(members['escrow'][0])
self.assertEqual(
len(events), index + 1,
"{} event expected for escrow: {}, but {} got instead".format(
index + 1, members['escrow'][0], len(events)))
def create_mock_package_handler(escrow_pubkey,
launcher_pubkey,
recipient_pubkey,
launcher_phone_number,
recipient_phone_number,
payment_buls,
collateral_buls,
deadline_timestamp,
description='mock_description',
from_location='mock_location',
to_location='mock_location',
from_address='mock_address',
to_address='mock_address',
event_location='mock_location',
photo=None):
"""
Create a mock package - for debug only.
---
:param escrow_pubkey:
:param launcher_pubkey:
:param recipient_pubkey:
:param launcher_phone_number:
:param recipient_phone_number:
:param payment_buls:
:param collateral_buls:
:param deadline_timestamp:
:param description:
:param from_location:
:param to_location:
:param from_address:
:param to_address:
:param event_location:
:param photo:
:return:
"""
return {
'status':
201,
'package':
db.create_package(escrow_pubkey, launcher_pubkey, recipient_pubkey,
launcher_phone_number, recipient_phone_number,
payment_buls, collateral_buls, deadline_timestamp,
description, from_location, to_location,
from_address, to_address, event_location, photo)
}
def create_package_handler(user_pubkey,
escrow_pubkey,
recipient_pubkey,
launcher_phone_number,
recipient_phone_number,
payment_buls,
collateral_buls,
deadline_timestamp,
description,
from_location,
to_location,
from_address,
to_address,
event_location,
photo=None):
"""
Create a package.
Use this call to create a new package for delivery.
---
:param user_pubkey:
:param escrow_pubkey:
:param recipient_pubkey:
:param launcher_phone_number:
:param recipient_phone_number:
:param payment_buls:
:param collateral_buls:
:param deadline_timestamp:
:param description:
:param from_location:
:param to_location:
:param from_address:
:param to_address:
:param event_location:
:param photo:
:return:
"""
package_details = db.create_package(
escrow_pubkey, user_pubkey, recipient_pubkey, launcher_phone_number,
recipient_phone_number, payment_buls, collateral_buls,
deadline_timestamp, description, from_location, to_location,
from_address, to_address, event_location, photo)
return {'status': 201, 'package': package_details}
def test_get_user_packages(self):
"""Getting user packages test."""
user = self.generate_keypair()
first_package_members = self.prepare_package_members()
LOGGER.info('creating package with user role: launcher')
db.create_package(
first_package_members['escrow'][0], user[0], first_package_members['recipient'][0],
'+490857461783', '+4904597863891', 50000000, 100000000, time.time(), 'Package description',
'12.970686,77.595590', '41.156193,-8.637541', 'India Bengaluru', 'Spain Porto', '12.970686,77.595590', None)
second_package_members = self.prepare_package_members()
LOGGER.info('creating package with user role: recipient')
db.create_package(
second_package_members['escrow'][0], second_package_members['launcher'][0], user[0],
'+490857461783', '+4904597863891', 50000000, 100000000, time.time(), 'Package description',
'12.970686,77.595590', '41.156193,-8.637541', 'India Bengaluru', 'Spain Porto', '12.970686,77.595590', None)
third_package_members = self.prepare_package_members()
LOGGER.info('creating package with user role: courier')
db.create_package(
third_package_members['escrow'][0], third_package_members['launcher'][0],
third_package_members['recipient'][0], '+490857461783', '+4904597863891',
50000000, 100000000, time.time(), 'Package description', '12.970686,77.595590', '41.156193,-8.637541',
'India Bengaluru', 'Spain Porto', '12.970686,77.595590', None)
db.add_event(user[0], 'couriered', '12.970686,77.595590', third_package_members['escrow'][0])
packages = db.get_packages(user[0])
self.assertEqual(len(packages), 3, "3 packages expected, {} got instead".format(len(packages)))
package = next((
package for package in packages if package['launcher_pubkey'] == user[0]), None)
self.assertEqual(package['user_role'], 'launcher',
"expected role: 'launcher', '{}' got instead".format(package['user_role']))
self.assertEqual(package['custodian_pubkey'], user[0],
"{} expected as custodian, {} got instead".format(user[0], package['custodian_pubkey']))
package = next((
package for package in packages if package['recipient_pubkey'] == user[0]), None)
self.assertEqual(package['user_role'], 'recipient',
"expected role: 'recipient', '{}' got instead".format(package['user_role']))
self.assertEqual(package['custodian_pubkey'], second_package_members['launcher'][0],
"{} expected as custodian, {} got instead".format(
second_package_members['launcher'][0], package['custodian_pubkey']))
package = next((
package for package in packages
if user[0] not in (package['recipient_pubkey'], package['launcher_pubkey'])), None)
self.assertEqual(package['user_role'], 'courier',
"expected role: 'courier', '{}' got instead".format(package['user_role']))
self.assertEqual(package['custodian_pubkey'], user[0],
"{} expected as custodian, {} got instead".format(user[0], package['custodian_pubkey']))
def test_get_user_packages(self):
"""Getting user packages test."""
user = self.generate_keypair()
first_package_members = self.prepare_package_members()
LOGGER.info('creating package with user role: launcher')
db.create_package(first_package_members['escrow'][0], user[0],
first_package_members['recipient'][0], 50000000,
100000000, time.time(), None, None, None, None)
second_package_members = self.prepare_package_members()
LOGGER.info('creating package with user role: recipient')
db.create_package(second_package_members['escrow'][0],
second_package_members['launcher'][0], user[0],
50000000, 100000000, time.time(), None, None, None,
None)
third_package_members = self.prepare_package_members()
LOGGER.info('creating package with user role: courier')
db.create_package(third_package_members['escrow'][0],
third_package_members['launcher'][0],
third_package_members['recipient'][0], 50000000,
100000000, time.time(), None, None, None, None)
db.add_event(third_package_members['escrow'][0], user[0], 'couriered',
None)
packages = db.get_packages(user[0])
self.assertEqual(
len(packages), 3,
"3 packages expected, {} got instead".format(len(packages)))
package = next(
(package
for package in packages if package['launcher_pubkey'] == user[0]),
None)
self.assertEqual(
package['user_role'], 'launcher',
"expected role: 'launcher', '{}' got instead".format(
package['user_role']))
self.assertEqual(
package['custodian_pubkey'], user[0],
"{} expected as custodian, {} got instead".format(
user[0], package['custodian_pubkey']))
package = next((package for package in packages
if package['recipient_pubkey'] == user[0]), None)
self.assertEqual(
package['user_role'], 'recipient',
"expected role: 'recipient', '{}' got instead".format(
package['user_role']))
self.assertEqual(
package['custodian_pubkey'], second_package_members['launcher'][0],
"{} expected as custodian, {} got instead".format(
second_package_members['launcher'][0],
package['custodian_pubkey']))
package = next((package for package in packages
if user[0] not in (package['recipient_pubkey'],
package['launcher_pubkey'])), None)
self.assertEqual(
package['user_role'], 'courier',
"expected role: 'courier', '{}' got instead".format(
package['user_role']))
self.assertEqual(
package['custodian_pubkey'], user[0],
"{} expected as custodian, {} got instead".format(
user[0], package['custodian_pubkey']))
