class ConnectOperationTestConfig(object):
@pytest.fixture
def cls_type(self):
return pipeline_ops_base.ConnectOperation
@pytest.fixture
def init_kwargs(self, mocker):
kwargs = {"callback": mocker.MagicMock()}
return kwargs
pipeline_ops_test.add_operation_tests(
test_module=this_module,
op_class_under_test=pipeline_ops_base.ConnectOperation,
op_test_config_class=ConnectOperationTestConfig,
)
class DisconnectOperationTestConfig(object):
@pytest.fixture
def cls_type(self):
return pipeline_ops_base.DisconnectOperation
@pytest.fixture
def init_kwargs(self, mocker):
kwargs = {"callback": mocker.MagicMock()}
return kwargs
kwargs = {"auth_provider": mocker.MagicMock(), "callback": mocker.MagicMock()}
return kwargs
class SetAuthProviderOperationInstantiationTests(SetAuthProviderOperationTestConfig):
@pytest.mark.it(
"Initializes 'auth_provider' attribute with the provided 'auth_provider' parameter"
)
def test_auth_provider(self, cls_type, init_kwargs):
op = cls_type(**init_kwargs)
assert op.auth_provider is init_kwargs["auth_provider"]
pipeline_ops_test.add_operation_tests(
test_module=this_module,
op_class_under_test=pipeline_ops_iothub.SetAuthProviderOperation,
op_test_config_class=SetAuthProviderOperationTestConfig,
extended_op_instantiation_test_class=SetAuthProviderOperationInstantiationTests,
)
class SetX509AuthProviderOperationTestConfig(object):
@pytest.fixture
def cls_type(self):
return pipeline_ops_iothub.SetX509AuthProviderOperation
@pytest.fixture
def init_kwargs(self, mocker):
kwargs = {"auth_provider": mocker.MagicMock(), "callback": mocker.MagicMock()}
return kwargs
op = cls_type(**init_kwargs)
assert op.sas_token == init_kwargs["sas_token"]
@pytest.mark.it(
"Initializes 'sas_token' attribute to None if no 'sas_token' parameter is provided"
)
def test_sas_token_default(self, cls_type, init_kwargs):
del init_kwargs["sas_token"]
op = cls_type(**init_kwargs)
assert op.sas_token is None
pipeline_ops_test.add_operation_tests(
test_module=this_module,
op_class_under_test=pipeline_ops_mqtt.SetMQTTConnectionArgsOperation,
op_test_config_class=SetMQTTConnectionArgsOperationTestConfig,
extended_op_instantiation_test_class=
SetMQTTConnectionArgsOperationInstantiationTests,
)
class MQTTPublishOperationTestConfig(object):
@pytest.fixture
def cls_type(self):
return pipeline_ops_mqtt.MQTTPublishOperation
@pytest.fixture
def init_kwargs(self, mocker):
kwargs = {
"topic": "some_topic",
"payload": "some_payload",
class SetSymmetricKeySecurityClientOperationInstantiationTests(
SetSymmetricKeySecurityClientOperationTestConfig
):
@pytest.mark.it(
"Initializes 'security_client' attribute with the provided 'security_client' parameter"
)
def test_security_client(self, cls_type, init_kwargs):
op = cls_type(**init_kwargs)
assert op.security_client is init_kwargs["security_client"]
pipeline_ops_test.add_operation_tests(
test_module=this_module,
op_class_under_test=pipeline_ops_provisioning.SetSymmetricKeySecurityClientOperation,
op_test_config_class=SetSymmetricKeySecurityClientOperationTestConfig,
extended_op_instantiation_test_class=SetSymmetricKeySecurityClientOperationInstantiationTests,
)
class SetX509SecurityClientOperationTestConfig(object):
@pytest.fixture
def cls_type(self):
return pipeline_ops_provisioning.SetX509SecurityClientOperation
@pytest.fixture
def init_kwargs(self, mocker):
kwargs = {"security_client": mocker.MagicMock(), "callback": mocker.MagicMock()}
return kwargs
@pytest.mark.it(
"Initializes 'query_params' attribute with the provided 'query_params' parameter"
)
def test_query_params(self, cls_type, init_kwargs):
op = cls_type(**init_kwargs)
assert op.query_params == init_kwargs["query_params"]
@pytest.mark.it("Initializes 'status_code' attribute as None")
def test_status_code(self, cls_type, init_kwargs):
op = cls_type(**init_kwargs)
assert op.status_code is None
@pytest.mark.it("Initializes 'response_body' attribute as None")
def test_response_body(self, cls_type, init_kwargs):
op = cls_type(**init_kwargs)
assert op.response_body is None
@pytest.mark.it("Initializes 'reason' attribute as None")
def test_reason(self, cls_type, init_kwargs):
op = cls_type(**init_kwargs)
assert op.reason is None
pipeline_ops_test.add_operation_tests(
test_module=this_module,
op_class_under_test=pipeline_ops_http.HTTPRequestAndResponseOperation,
op_test_config_class=HTTPRequestAndResponseOperationTestConfig,
extended_op_instantiation_test_class=
HTTPRequestAndResponseOperationInstantiationTests,
)
"Initializes 'method_params' attribute with the provided 'method_params' parameter"
)
def test_method_params(self, cls_type, init_kwargs):
op = cls_type(**init_kwargs)
assert op.method_params is init_kwargs["method_params"]
@pytest.mark.it("Initializes 'method_response' attribute as None")
def test_method_response(self, cls_type, init_kwargs):
op = cls_type(**init_kwargs)
assert op.method_response is None
pipeline_ops_test.add_operation_tests(
test_module=this_module,
op_class_under_test=pipeline_ops_iothub_http.MethodInvokeOperation,
op_test_config_class=MethodInvokeOperationTestConfig,
extended_op_instantiation_test_class=
MethodInvokeOperationInstantiationTests,
)
class GetStorageInfoOperationTestConfig(object):
@pytest.fixture
def cls_type(self):
return pipeline_ops_iothub_http.GetStorageInfoOperation
@pytest.fixture
def init_kwargs(self, mocker):
kwargs = {
"blob_name": "__fake_blob_name__",
"callback": mocker.MagicMock()
class InitializePipelineOperationTestConfig(object):
@pytest.fixture
def cls_type(self):
return pipeline_ops_base.InitializePipelineOperation
@pytest.fixture
def init_kwargs(self, mocker):
kwargs = {"callback": mocker.MagicMock()}
return kwargs
pipeline_ops_test.add_operation_tests(
test_module=this_module,
op_class_under_test=pipeline_ops_base.InitializePipelineOperation,
op_test_config_class=InitializePipelineOperationTestConfig,
)
class ConnectOperationTestConfig(object):
@pytest.fixture
def cls_type(self):
return pipeline_ops_base.ConnectOperation
@pytest.fixture
def init_kwargs(self, mocker):
kwargs = {"callback": mocker.MagicMock()}
return kwargs
@pytest.mark.it("Initializes 'polling_timer' attribute to None")
def test_polling_timer(self, cls_type, init_kwargs):
op = cls_type(**init_kwargs)
assert op.polling_timer is None
@pytest.mark.it(
"Initializes 'provisioning_timeout_timer' attribute to None")
def test_provisioning_timeout_timer(self, cls_type, init_kwargs):
op = cls_type(**init_kwargs)
assert op.provisioning_timeout_timer is None
pipeline_ops_test.add_operation_tests(
test_module=this_module,
op_class_under_test=pipeline_ops_provisioning.RegisterOperation,
op_test_config_class=RegisterOperationTestConfig,
extended_op_instantiation_test_class=RegisterOperationInstantiationTests,
)
class PollStatusOperationTestConfig(object):
@pytest.fixture
def cls_type(self):
return pipeline_ops_provisioning.PollStatusOperation
@pytest.fixture
def init_kwargs(self, mocker):
kwargs = {
"operation_id": "some_operation_id",
"request_payload": "some_request_payload",
"callback": mocker.MagicMock(),
