def mockup(cls):
cls.mockup_attachment1 = join(STUFF_DIR, 'test-image-1.jpg')
cls.mockup_attachment2 = join(STUFF_DIR, 'test-image-2.png')
admin1 = Admin()
admin1.email = '*****@*****.**'
admin1.password = '******'
admin1.is_active = True
cls.session.add(admin1)
client1 = Client()
client1.email = '*****@*****.**'
client1.password = '******'
client1.is_active = True
cls.session.add(client1)
client2 = Client()
client2.email = '*****@*****.**'
client2.password = '******'
client2.is_active = True
client2.is_email_verified = True
cls.session.add(client2)
department1 = TicketDepartment(title='department-1')
department2 = TicketDepartment(title='department-2')
cls.session.add(department1)
cls.session.add(department2)
cls.session.commit()
cls.mockup_department1_id = department1.id
cls.mockup_admin1_id = admin1.id
cls.mockup_client1_id = client1.id
cls.mockup_client2_id = client2.id
def mockup(cls):
admin1 = Admin()
admin1.email = '*****@*****.**'
admin1.password = '******'
admin1.is_active = True
cls.session.add(admin1)
testnet = Cryptocurrency(symbol='TESTNET3',
name='TESTNET3',
wallet_id='TESTNET3')
rinkeby = Cryptocurrency(symbol='RINKEBY',
name='RINKEBY',
wallet_id='RINKEBY')
cls.session.add(testnet)
cls.session.add(rinkeby)
cls.session.commit()
class MockStexchangeClient(StexchangeClient):
def __init__(self, headers=None):
super().__init__("", headers)
def asset_list(self):
return ujson.loads(
'[{"name": "RINKEBY", "prec": 8}, {"name": "TESTNET3", "prec": 8}]'
)
def asset_summary(self):
return ujson.loads(
'[{"freeze_count": 0, "name": "RINKEBY", "total_balance": "1996.3", "available_count": 1, '
'"freeze_balance": "0", "available_balance": "1996.3"}, {"freeze_count": 0, "name": "TESTNET3", '
'"total_balance": "9137.8", "available_count": 1, "freeze_balance": "0", "available_balance": '
'"9137.8"}]')
stexchange_client._set_instance(MockStexchangeClient())
def mockup(cls):
cls.mockup_id_card = join(STUFF_DIR, 'test-image-1.jpg')
cls.mockup_id_card_secondary = join(STUFF_DIR, 'test-image-2.png')
iran = Country(name='Iran', code='ir', phone_prefix=98)
tehran_state = State(name='Tehran', country=iran)
tehran_city = City(name='Tehran', state=tehran_state)
cls.session.add(tehran_city)
admin1 = Admin()
admin1.email = '*****@*****.**'
admin1.password = '******'
admin1.is_active = True
cls.session.add(admin1)
client1 = Client()
client1.email = '*****@*****.**'
client1.password = '******'
client1.is_active = True
client1.is_email_verified = True
client1.evidence.mobile_phone = '+77777777777'
client1.evidence.fixed_phone = '+888888888'
cls.session.add(client1)
cls.session.commit()
cls.city1_id = tehran_city.id
cls.client1_id = client1.id
def mockup(cls):
admin1 = Admin()
admin1.email = '*****@*****.**'
admin1.password = '******'
admin1.is_active = True
cls.session.add(admin1)
cls.session.add(Fiat(symbol='USD', name='USA Dollar'))
cls.session.add(
Cryptocurrency(symbol='BTC', name='Bitcoin', wallet_id='BTC'))
cls.session.commit()
def mockup(cls):
admin1 = Admin()
admin1.email = '*****@*****.**'
admin1.password = '******'
admin1.is_active = True
cls.session.add(admin1)
invitation1 = Invitation()
invitation1.code = 'test-code-1'
invitation1.total_sits = 100
invitation1.filled_sits = 50
cls.session.add(invitation1)
cls.session.commit()
def mockup(cls):
admin1 = Admin()
admin1.email = '*****@*****.**'
admin1.password = '******'
admin1.is_active = True
cls.session.add(admin1)
client1 = Client()
client1.email = '*****@*****.**'
client1.password = '******'
client1.is_active = True
cls.session.add(client1)
cls.session.commit()
cls.client1_id = client1.id
def mockup(cls):
admin1 = Admin()
admin1.email = '*****@*****.**'
admin1.password = '******'
admin1.is_active = True
cls.session.add(admin1)
client1 = Client()
client1.email = '*****@*****.**'
client1.password = '******'
client1.is_active = True
cls.session.add(client1)
testnet = Cryptocurrency(symbol='TESTNET3',
name='TESTNET3',
wallet_id='TESTNET3')
rinkeby = Cryptocurrency(symbol='RINKEBY',
name='RINKEBY',
wallet_id='RINKEBY')
cls.session.add(testnet)
cls.session.add(rinkeby)
cls.session.commit()
class MockStexchangeClient(StexchangeClient):
def __init__(self, headers=None):
super().__init__("", headers)
def asset_list(self):
return ujson.loads(
'[{"name": "RINKEBY", "prec": 8}, {"name": "TESTNET3", "prec": 8}]'
)
def balance_query(self, *args, **kwargs):
return ujson.loads(
'{"TESTNET3": {"available": "9137.8", "freeze": "0"}, "RINKEBY": {"available": "100.65", '
'"freeze": "10.01"}}')
stexchange_client._set_instance(MockStexchangeClient())
def mockup(cls):
admin1 = Admin()
admin1.email = '*****@*****.**'
admin1.password = '******'
admin1.is_active = True
client1 = Client()
client1.email = '*****@*****.**'
client1.password = '******'
client1.is_active = True
client1.is_email_verified = True
client1.is_evidence_verified = True
cls.session.add(admin1)
cls.session.add(client1)
usd = Fiat(symbol='USD', name='USA Dollar')
cls.session.add(usd)
cls.session.commit()
cls.client1_id = client1.id
def mockup(cls):
admin1 = Admin()
admin1.email = '*****@*****.**'
admin1.password = '******'
admin1.is_active = True
cls.session.add(admin1)
client1 = Client()
client1.email = '*****@*****.**'
client1.password = '******'
client1.is_active = True
client1.is_email_verified = True
client1.is_evidence_verified = True
cls.session.add(client1)
client2 = Client()
client2.email = '*****@*****.**'
client2.password = '******'
client2.is_active = True
client2.is_email_verified = True
client2.is_evidence_verified = True
cls.session.add(client2)
irr = Fiat(
symbol='IRR',
name='Iran Rial',
)
cls.session.add(irr)
# Adding a payment gateway
shaparak = PaymentGateway()
shaparak.name = "shaparak"
shaparak.fiat_symbol = "IRR"
shaparak.cashout_min = cashout_min
shaparak.cashout_max = cashout_max
shaparak.cashout_static_commission = cashout_static_commission
shaparak.cashout_commission_rate = cashout_commission_rate
shaparak.cashout_max_commission = cashout_max_commission
cls.session.add(shaparak)
# Mine, verified:
sheba_address_1 = BankAccount()
sheba_address_1.iban = 'IR123456789123456789123456'
sheba_address_1.owner = "Client One"
sheba_address_1.client = client1
sheba_address_1.fiat_symbol = "IRR"
sheba_address_1.is_verified = True
# Mine, unverified:
sheba_address_2 = BankAccount()
sheba_address_2.iban = 'IR837498056254698443242343'
sheba_address_2.owner = "Client One"
sheba_address_2.client = client1
sheba_address_2.fiat_symbol = "IRR"
sheba_address_2.is_verified = False
# Other's, verified:
sheba_address_3 = BankAccount()
sheba_address_3.iban = 'IR837498056254698443242343'
sheba_address_3.owner = "Client Two"
sheba_address_3.client = client2
sheba_address_3.fiat_symbol = "IRR"
sheba_address_3.is_verified = True
for address in [sheba_address_1, sheba_address_2, sheba_address_3]:
cls.session.add(address)
cls.session.commit()
cls.mockup_client_1_id = client1.id
cls.mockup_sheba_address_verified_id = sheba_address_1.id
cls.mockup_sheba_address_unverified_id = sheba_address_2.id
cls.mockup_sheba_address_others_id = sheba_address_3.id
cls.mockup_payment_gateway_name = shaparak.name
class MockStexchangeClient(StexchangeClient):
def __init__(self, headers=None):
super().__init__("", headers)
self.mock_balance = ["3001", "0"]
def asset_list(self):
return ujson.loads('[{"name": "IRR", "prec": 2}]')
def balance_update(self, user_id, asset, business, business_id, change, detail):
if user_id == cls.mockup_client_1_id and business in ['cashout', 'cashback'] and asset == 'IRR':
self.mock_balance[0] = '{:.8f}'.format(Decimal(change) + Decimal(self.mock_balance[0]))
return ujson.loads(
'{"IRR": {"available": "' +
self.mock_balance[0] +
'", "freeze": "' +
self.mock_balance[1] +
'"}}'
)
def balance_query(self, *args, **kwargs):
return ujson.loads(
'{"IRR": {"available": "' +
self.mock_balance[0] +
'", "freeze": "' +
self.mock_balance[1] +
'"}}'
)
stexchange_client._set_instance(MockStexchangeClient())