def test_user_stocks(self):
self.create_user(username='******', password='******')
token = Helper.token(1, 'newuser')
headers = dict(Token=token)
response = self.client.get('/users/stocks', headers=headers)
self.assertEqual(response.status_code, 200)
self.assertTrue('stocks' in response.json)
def test_verify_token(self):
token = Helper.token(1, 'newuser') # not user with token created using 'newuser'
headers = dict(Token=token)
data = dict(amount=10)
response = self.client.post('/wallets/deposit', data=data, headers=headers)
self.assertEqual(response.status_code, 401)
self.assertTrue('error' in response.json)
self.assertEqual(response.json['error'], 'Expects a valid token to be provided')
def test_deposit(self):
self.create_user(username='******', password='******')
token = Helper.token(1, 'newuser')
headers = dict(Token=token)
data = dict(amount=1000)
response = self.client.post('/wallets/deposit', data=data, headers=headers)
self.assertEqual(response.status_code, 200)
self.assertTrue('total' in response.json)
self.assertEqual(response.json['total'], 1000)
def test_stock_purchase(self):
self.create_user(username='******', password='******', amount=5000)
token = Helper.token(1, 'newuser')
headers = dict(Token=token)
with requests_mock.mock() as mock:
symbol = 'aapl'
volume = 10
url = IEXCloud.url(f'/stock/{symbol}/price')
mock.get(url, text='100.0')
# make request actions
response = self.client.post(f'/stocks/{symbol}/purchase/{volume}',
headers=headers)
self.assertEqual(response.status_code, 200)
with self.app.app_context():
amount = User.query.get(1).wallet.amount
self.assertEqual(amount, 4000)
def test_token(self):
expected = 'eyJ0eXAiOiJKV1QiLCJhbGciOiJIUzI1NiJ9.eyJ1c2VybmFtZSI6ImR1bW15IiwiaWQiOjF9.O-Lz9IAv58Soa4Lk0Cr0ccpnMUa-060kvsC6budrH10'
token = Helper.token(1, 'dummy')
self.assertEqual(token, expected)