fix authenticated order book and add test

Author: csko

gdax/orderbook.py

@@ -131,7 +131,7 @@ async def _subscribe(self):
             message['key'] = self.api_key
             message['passphrase'] = self.passphrase
 
-        return await self._send(type='subscribe', product_ids=self.product_ids)
+        return await self._send(**message)
 
     async def handle_message(self):
         try:

tests/test_orderbook.py

@@ -1,12 +1,15 @@
-import pytest
-import gdax
-import gdax.orderbook
 import json
-import asyncio
-from asynctest import MagicMock, patch, CoroutineMock, call
 import uuid
+import base64
 from decimal import Decimal
 
+import asyncio
+import pytest
+from asynctest import MagicMock, patch, CoroutineMock, call
+
+import gdax
+import gdax.orderbook
+
 
 class AsyncContextManagerMock(MagicMock):
     async def __aenter__(self):
@@ -204,3 +207,27 @@ async def test_heartbeat(self, mock_book, mock_connect):
             heartbeat_msg = {'type': 'heartbeat', 'on': True}
             calls = [call(subscribe_msg), call(heartbeat_msg)]
             mock_connect.return_value.aenter.send_json.assert_has_calls(calls)
+
+    @patch('gdax.trader.Trader.get_product_order_book')
+    async def test_authentication(self, mock_book, mock_connect, mocker):
+        mock_connect.return_value.aenter.send_json = CoroutineMock()
+        mock_book.return_value = {'bids': [], 'asks': [], 'sequence': 1}
+        timestamp = '1493343391.076892'
+        mocker.patch('time.time', return_value=timestamp)
+        product_ids = ['ETH-USD']
+        async with gdax.orderbook.OrderBook(
+            product_ids,
+            api_key='a',
+            api_secret=base64.b64encode(b'a' * 64),
+            passphrase='b',
+        ) as orderbook:
+            msg = {
+                'type': 'subscribe',
+                'product_ids': product_ids,
+                'signature': '5qne58tAXSW3OJlU/GoC+/mTLF1xgT8vucjJWFZzhsU=',
+                'timestamp': timestamp,
+                'key': 'a',
+                'passphrase': 'b',
+                }
+            assert orderbook._authenticated
+            mock_connect.return_value.aenter.send_json.assert_called_with(msg)