TestBtcdeAPIDocu

Complexity

Total Complexity

10

Size/Duplication

Total Lines	98
Duplicated Lines	0 %

Importance

Changes	4
Bugs	0	Features	2

7 Methods

Rating	Name	Duplication	Size	Complexity
A	sortParams()	0	11	3
A	verifySignature()	0	13	2
A	tearDown()	0	5	1
A	setUp()	0	6	1
B	test_signature_get()	0	24	1
A	test_signature_post()	0	15	1
A	test_signature_delete()	0	13	1

from unittest import TestCase 
from mock import patch
import time
import hashlib
import hmac
import requests
import requests_mock
import btcde

@requests_mock.Mocker()
class TestBtcdeAPIDocu(TestCase):
    '''Tests are as in bitcoin.de API documentation. 
    https://www.bitcoin.de/de/api/tapi/v2/docu'''
    
    def sortParams(self, url, params={}):
        '''To sort params for url string.'''
        self.encoded_string = ''
        if len(params) > 0:
            for key, value in sorted(params.items()):
                self.encoded_string += str(key) + '=' + str(value) + '&'
            self.encoded_string = self.encoded_string[:-1]
            self.url = url + '?' + self.encoded_string
        else:
            self.url = url
        return self.url, self.encoded_string
    
    def verifySignature(self, url, method, params):
        '''To verify API Signature.'''
        self.XAPINONCE += 1
        self.url, self.encoded_string = self.sortParams(url, params)
        if method == 'POST':
            md5_encoded_query_string = hashlib.md5(self.encoded_string.encode()).hexdigest()
        else:
            md5_encoded_query_string = hashlib.md5(b'').hexdigest()
        hmac_data = method + '#' + \
                    self.url + '#' + self.XAPIKEY + \
                    '#' + str(self.XAPINONCE) + '#' + md5_encoded_query_string
        hmac_signed = hmac.new(bytearray(self.XAPISECRET.encode()), msg=hmac_data.encode(), digestmod=hashlib.sha256).hexdigest()
        return hmac_signed
        
    def setUp(self):
        self.XAPIKEY = 'f00b4r'
        self.XAPISECRET = 'b4rf00'
        self.XAPINONCE = int(time.time())
        btcde.nonce = self.XAPINONCE
        self.conn = btcde.Connection(self.XAPIKEY, self.XAPISECRET)
        
    def tearDown(self):
        del self.XAPIKEY
        del self.XAPISECRET
        del self.XAPINONCE
        del self.conn
     
    def test_signature_post(self, m):
        '''Test the signature on a post request.'''
        params = {'type': 'buy',
                  'trading_pair': 'btceur',
                  'max_amount': 10,
                  'price': 1337}
        response = {"order_id": "A1234BC",
                    "errors": [],
                    "credits": 8,}
        m.post(requests_mock.ANY, json=response, status_code=201)
        btcde.createOrder(self.conn, params.get('type'), params.get('trading_pair'), params.get('max_amount'), params.get('price'))
        history = m.request_history
        request_signature = history[0].headers.get('X-API-SIGNATURE')
        verified_signature = self.verifySignature(btcde.orderuri, 'POST', params)
        self.assertEqual(request_signature, verified_signature)
             
    def test_signature_get(self, m):
        '''Test the signature on a get request.'''
        params = {'type': 'buy',
                  'trading_pair': 'btceur',}
        response = {"orders":{"order_id": "A1B2D3", 
            "trading_pair": "btceur", "type": "buy",
            "max_amount":0.5, "min_amount":0.1,
            "price":230.55, "max_volume":115.28, "min_volume":23.06, 
            "order_requirements_fullfilled":True, 
            "trading_partner_information":{
            "username":"bla", "is_kyc_full":True,
            "trust_level":"gold", "bank_name":"Sparkasse",
            "bic":"HASPDEHHXXX", "seat_of_bank":"DE", "rating": 99,
            "amount_trades": 52}, 
            "order_requirements":{"min_trust_level":"gold",
            "only_kyc_full":True, "seat_of_bank":["DE", "NL"],"payment_option":1, }
            }, "errors":[],
            "credits":12 }
        m.get(requests_mock.ANY, json=response, status_code=200)
        btcde.showOrderbook(self.conn, params.get('type'), params.get('trading_pair'))
        history = m.request_history
        request_signature = history[0].headers.get('X-API-SIGNATURE')
        verified_signature = self.verifySignature(btcde.orderuri, 'GET', params)
        self.assertEqual(request_signature, verified_signature)
        
    def test_signature_delete(self, m):
        '''Test the signature on a delete request.'''
        order_id = 'A1234BC'
        trading_pair = 'btceur'
        params = {'order_id': order_id,
                  'trading_pair': trading_pair}
        m.delete(requests_mock.ANY, json={}, status_code=200)
        btcde.deleteOrder(self.conn, order_id, trading_pair)
        history = m.request_history
        request_signature = history[0].headers.get('X-API-SIGNATURE')
        url = btcde.orderuri + "/" + order_id + "/" + trading_pair
        verified_signature = self.verifySignature(url, 'DELETE', params)
        self.assertEqual(request_signature, verified_signature)