PB-159: update xain-{proto,sdk} dependencies to the right branch (#301)

* PB-159: update xain-{proto,sdk} dependencies to the right branch

Follow-up of #298

* PB-159: fix broken tests

This test broke due to a change in xain-sdk.

Ref: https://github.com/xainag/xain-sdk/pull/91/

setup.py

@@ -27,7 +27,7 @@
     "numpy==1.15",  # BSD
     "grpcio==1.23",  # Apache License 2.0
     "structlog==19.2.0",  # Apache License 2.0
-    "xain-proto @ git+https://github.com/xainag/xain-proto.git@PB-159-use-s3-for-transfering-weights#egg=xain_proto-0.6.0&subdirectory=python",  # Apache License 2.0
+    "xain-proto @ git+https://github.com/xainag/xain-proto.git@development#egg=xain_proto-0.6.0&subdirectory=python",  # Apache License 2.0
     "boto3==1.10.48",  # Apache License 2.0
     "toml==0.10.0",  # MIT
     "schema~=0.7",  # MIT
@@ -53,7 +53,7 @@
     "pytest-cov==2.8.1",  # MIT
     "pytest-watch==4.2.0",  # MIT
     "pytest-mock==2.0.0",  # MIT
-    "xain-sdk@ git+https://github.com/xainag/xain-sdk.git@PB-159-use-s3-for-transfering-weights#egg=xain_sdk-0.6.0",  # Apache License 2.0
+    "xain-sdk@ git+https://github.com/xainag/xain-sdk.git@development#egg=xain_sdk-0.6.0",  # Apache License 2.0
 ]
 
 docs_require = [

tests/test_grpc.py

@@ -265,7 +265,8 @@ def test_message_loop(mock_heartbeat_request, _mock_sleep, _mock_event):
     message_loop(channel, state_record, terminate_event)
 
     # check that the heartbeat is sent exactly twice
-    mock_heartbeat_request.assert_has_calls([mock.call(), mock.call()])
+    expected_call = mock.call(round=-1, state=State.READY)
+    mock_heartbeat_request.assert_has_calls([expected_call, expected_call])
 
 
 @pytest.mark.integration