def test_execute_multiple_dofirst_error(): with mock.patch( "umapi_client.connection.requests.Session.post") as mock_post: mock_post.return_value = MockResponse( 200, { "result": "error", "completed": 1, "notCompleted": 1, "errors": [{ "index": 1, "step": 1, "errorCode": "test.error", "message": "Test error message" }] }) conn = Connection(**mock_connection_params) action0 = Action(top="top0").append(a="a0") action1 = Action(top="top1").append(a="a1").insert(b="b") assert conn.execute_multiple([action0, action1]) == (0, 2, 1) assert action0.execution_errors() == [] assert action1.execution_errors() == [{ "command": { "a": "a1" }, "target": { "top": "top1" }, "errorCode": "test.error", "message": "Test error message" }]
def test_execute_multiple_dofirst_success(): with mock.patch( "umapi_client.connection.requests.Session.post") as mock_post: mock_post.return_value = MockResponse(200, {"result": "success"}) conn = Connection(**mock_connection_params) action0 = Action(top="top0").append(a="a0").insert(b="b") action1 = Action(top="top1").append(a="a1") assert conn.execute_multiple([action0, action1]) == (0, 2, 2)
def test_execute_multiple_single_queued_throttle_actions(): with mock.patch( "umapi_client.connection.requests.Session.post") as mock_post: mock_post.side_effect = [ MockResponse(200, {"result": "success"}), MockResponse( 200, { "result": "partial", "completed": 1, "notCompleted": 1, "errors": [{ "index": 0, "step": 0, "errorCode": "test" }] }) ] conn = Connection(throttle_actions=2, **mock_connection_params) action0 = Action(top="top0").append(a="a0") action1 = Action(top="top1").append(a="a1") action2 = Action(top="top2").append(a="a2") action3 = Action(top="top3").append(a="a3") assert conn.execute_multiple([action0, action1, action2], immediate=False) == (1, 2, 2) local_status, server_status = conn.status(remote=False) assert server_status == { "status": "Never contacted", "endpoint": conn.endpoint } assert local_status == { "multiple-query-count": 0, "single-query-count": 0, "actions-sent": 2, "actions-completed": 2, "actions-queued": 1 } assert conn.execute_single(action3) == (0, 2, 1) local_status, _ = conn.status(remote=False) assert local_status == { "multiple-query-count": 0, "single-query-count": 0, "actions-sent": 4, "actions-completed": 3, "actions-queued": 0 } assert action0.execution_errors() == [] assert action1.execution_errors() == [] assert action2.execution_errors() == [{ "command": { "a": "a2" }, "target": { "top": "top2" }, "errorCode": "test" }] assert action3.execution_errors() == []
def test_execute_single_success_immediate(): with mock.patch( "umapi_client.connection.requests.Session.post") as mock_post: mock_post.return_value = MockResponse(200, {"result": "success"}) conn = Connection(**mock_connection_params) action = Action(top="top").append(a="a") assert conn.execute_single(action, immediate=True) == (0, 1, 1)
def test_execute_single_dofirst_error_immediate(): with mock.patch( "umapi_client.connection.requests.Session.post") as mock_post: mock_post.return_value = MockResponse( 200, { "result": "error", "errors": [{ "index": 0, "step": 0, "errorCode": "test.error", "message": "Test error message" }] }) conn = Connection(**mock_connection_params) action = Action(top="top").insert(a="a") assert conn.execute_single(action, immediate=True) == (0, 1, 0) assert action.execution_errors() == [{ "command": { "a": "a" }, "target": { "top": "top" }, "errorCode": "test.error", "message": "Test error message" }]
def test_execute_single_error_immediate_throttled(): with mock.patch( "umapi_client.connection.requests.Session.post") as mock_post: mock_post.return_value = MockResponse( 200, { "result": "partial", "completed": 1, "notCompleted": 1, "errors": [{ "index": 1, "step": 0, "errorCode": "test" }] }) conn = Connection(throttle_commands=2, **mock_connection_params) action = Action(top="top0").append(a="a0").append(a="a1").append( a="a2") assert conn.execute_single(action, immediate=True) == (0, 2, 1) assert action.execution_errors() == [{ "command": { "a": "a2" }, "target": { "top": "top0" }, "errorCode": "test" }]
def test_execute_single_success_queued(): with mock.patch( "umapi_client.connection.requests.Session.post") as mock_post: mock_post.return_value = MockResponse(200, {"result": "success"}) conn = Connection(throttle_actions=2, **mock_connection_params) action = Action(top="top").append(a="a") assert conn.execute_single(action) == (1, 0, 0) assert conn.execute_single(action) == (0, 2, 2)
def test_execute_multiple_queued_throttle_actions_error(): with mock.patch( "umapi_client.connection.requests.Session.post") as mock_post: mock_post.side_effect = [ MockResponse(500), MockResponse(200, {"result": "success"}), MockResponse(200, {"result": "success"}) ] conn = Connection(throttle_actions=2, **mock_connection_params) action0 = Action(top="top0").append(a="a0") action1 = Action(top="top1").append(a="a1") action2 = Action(top="top2").append(a="a2") action3 = Action(top="top3").append(a="a3") action4 = Action(top="top4").append(a="a4") action5 = Action(top="top5").append(a="a5") pytest.raises(BatchError, conn.execute_multiple, [action0, action1, action2, action3, action4, action5], immediate=False) local_status, _ = conn.status(remote=False) assert local_status == { "multiple-query-count": 0, "single-query-count": 0, "actions-sent": 6, "actions-completed": 4, "actions-queued": 0 }
def test_execute_single_error_queued_throttled(): with mock.patch( "umapi_client.connection.requests.Session.post") as mock_post: mock_post.side_effect = [ MockResponse(200, {"result": "success"}), MockResponse( 200, { "result": "partial", "completed": 1, "notCompleted": 1, "errors": [{ "index": 1, "step": 0, "errorCode": "test.error", "message": "Test error message" }] }) ] conn = Connection(throttle_actions=2, throttle_commands=1, **mock_connection_params) action = Action(top="top").append(a="a").append(b="b").append( c="c").append(d="d") assert conn.execute_single(action) == (0, 4, 3) assert action.execution_errors() == [{ "command": { "d": "d" }, "target": { "top": "top" }, "errorCode": "test.error", "message": "Test error message" }]
def test_action_create_two_dofirst(): action = Action(a1="a1 text", z1="z1 text").append(com1={ "com1k": "com1v" }).insert(com2={"com2k": "com2v"}) assert json.dumps(action.wire_dict(), sort_keys=True) == \ '{"a1": "a1 text", "do": [{"com2": {"com2k": "com2v"}}, {"com1": {"com1k": "com1v"}}], "z1": "z1 text"}'
def test_action_create_one_dofirst(): action = Action(z1="z1 text").insert(com1={"com1k": "com1v"}) assert json.dumps(action.wire_dict(), sort_keys=True) == \ '{"do": [{"com1": {"com1k": "com1v"}}], "z1": "z1 text"}'
def test_action_create(): action = Action(frame_name="frame text") assert json.dumps(action.wire_dict(), sort_keys=True) == \ '{"do": [], "frame_name": "frame text"}'