Implement mock.patch.dict

This commit is contained in:
Ignacio Orlandini 2020-11-24 10:45:41 -03:00
parent 26b95100d0
commit 8b961fbc1e

View File

@ -568,15 +568,11 @@ def test_passes_request_into_context_request(client):
assert response_json(response) == {"data": {"request": "testing"}}
@patch("graphene_django.settings.graphene_settings.ATOMIC_MUTATIONS", False)
@patch.dict(
connection.settings_dict, {"ATOMIC_MUTATIONS": False, "ATOMIC_REQUESTS": True}
)
def test_form_mutation_multiple_creation_invalid_atomic_request(client):
old_atomic_mutations = connection.settings_dict.get("ATOMIC_MUTATIONS", False)
old_atomic_requests = connection.settings_dict["ATOMIC_REQUESTS"]
old_graphene_atomic_mutations = graphene_settings.ATOMIC_MUTATIONS
try:
connection.settings_dict["ATOMIC_MUTATIONS"] = False
connection.settings_dict["ATOMIC_REQUESTS"] = True
graphene_settings.ATOMIC_MUTATIONS = False
query = """
mutation PetMutations {
petFormMutation1: petFormMutation(input: { name: "Mia", age: 99 }) {
@ -607,21 +603,12 @@ def test_form_mutation_multiple_creation_invalid_atomic_request(client):
assert Pet.objects.count() == 0
finally:
connection.settings_dict["ATOMIC_MUTATIONS"] = old_atomic_mutations
connection.settings_dict["ATOMIC_REQUESTS"] = old_atomic_requests
graphene_settings.ATOMIC_MUTATIONS = old_graphene_atomic_mutations
@patch("graphene_django.settings.graphene_settings.ATOMIC_MUTATIONS", False)
@patch.dict(
connection.settings_dict, {"ATOMIC_MUTATIONS": True, "ATOMIC_REQUESTS": False}
)
def test_form_mutation_multiple_creation_invalid_atomic_mutation_1(client):
old_atomic_mutations = connection.settings_dict.get("ATOMIC_MUTATIONS", False)
old_atomic_requests = connection.settings_dict["ATOMIC_REQUESTS"]
old_graphene_atomic_mutations = graphene_settings.ATOMIC_MUTATIONS
try:
connection.settings_dict["ATOMIC_MUTATIONS"] = True
connection.settings_dict["ATOMIC_REQUESTS"] = False
graphene_settings.ATOMIC_MUTATIONS = False
query = """
mutation PetMutations {
petFormMutation1: petFormMutation(input: { name: "Mia", age: 99 }) {
@ -652,21 +639,12 @@ def test_form_mutation_multiple_creation_invalid_atomic_mutation_1(client):
assert Pet.objects.count() == 0
finally:
connection.settings_dict["ATOMIC_MUTATIONS"] = old_atomic_mutations
connection.settings_dict["ATOMIC_REQUESTS"] = old_atomic_requests
graphene_settings.ATOMIC_MUTATIONS = old_graphene_atomic_mutations
@patch("graphene_django.settings.graphene_settings.ATOMIC_MUTATIONS", True)
@patch.dict(
connection.settings_dict, {"ATOMIC_MUTATIONS": False, "ATOMIC_REQUESTS": False}
)
def test_form_mutation_multiple_creation_invalid_atomic_mutation_2(client):
old_atomic_mutations = connection.settings_dict.get("ATOMIC_MUTATIONS", False)
old_atomic_requests = connection.settings_dict["ATOMIC_REQUESTS"]
old_graphene_atomic_mutations = graphene_settings.ATOMIC_MUTATIONS
try:
connection.settings_dict["ATOMIC_MUTATIONS"] = False
connection.settings_dict["ATOMIC_REQUESTS"] = False
graphene_settings.ATOMIC_MUTATIONS = True
query = """
mutation PetMutations {
petFormMutation1: petFormMutation(input: { name: "Mia", age: 99 }) {
@ -697,21 +675,12 @@ def test_form_mutation_multiple_creation_invalid_atomic_mutation_2(client):
assert Pet.objects.count() == 0
finally:
connection.settings_dict["ATOMIC_MUTATIONS"] = old_atomic_mutations
connection.settings_dict["ATOMIC_REQUESTS"] = old_atomic_requests
graphene_settings.ATOMIC_MUTATIONS = old_graphene_atomic_mutations
@patch("graphene_django.settings.graphene_settings.ATOMIC_MUTATIONS", False)
@patch.dict(
connection.settings_dict, {"ATOMIC_MUTATIONS": False, "ATOMIC_REQUESTS": False}
)
def test_form_mutation_multiple_creation_invalid_non_atomic(client):
old_atomic_mutations = connection.settings_dict.get("ATOMIC_MUTATIONS", False)
old_atomic_requests = connection.settings_dict["ATOMIC_REQUESTS"]
old_graphene_atomic_mutations = graphene_settings.ATOMIC_MUTATIONS
try:
connection.settings_dict["ATOMIC_MUTATIONS"] = False
connection.settings_dict["ATOMIC_REQUESTS"] = False
graphene_settings.ATOMIC_MUTATIONS = False
query = """
mutation PetMutations {
petFormMutation1: petFormMutation(input: { name: "Mia", age: 99 }) {
@ -746,21 +715,12 @@ def test_form_mutation_multiple_creation_invalid_non_atomic(client):
assert pet.name == "Enzo"
assert pet.age == 0
finally:
connection.settings_dict["ATOMIC_MUTATIONS"] = old_atomic_mutations
connection.settings_dict["ATOMIC_REQUESTS"] = old_atomic_requests
graphene_settings.ATOMIC_MUTATIONS = old_graphene_atomic_mutations
@patch("graphene_django.settings.graphene_settings.ATOMIC_MUTATIONS", False)
@patch.dict(
connection.settings_dict, {"ATOMIC_MUTATIONS": False, "ATOMIC_REQUESTS": True}
)
def test_model_form_mutation_multiple_creation_invalid_atomic_request(client):
old_atomic_mutations = connection.settings_dict.get("ATOMIC_MUTATIONS", False)
old_atomic_requests = connection.settings_dict["ATOMIC_REQUESTS"]
old_graphene_atomic_mutations = graphene_settings.ATOMIC_MUTATIONS
try:
connection.settings_dict["ATOMIC_MUTATIONS"] = False
connection.settings_dict["ATOMIC_REQUESTS"] = True
graphene_settings.ATOMIC_MUTATIONS = False
query = """
mutation PetMutations {
petMutation1: petMutation(input: { name: "Mia", age: 99 }) {
@ -800,21 +760,12 @@ def test_model_form_mutation_multiple_creation_invalid_atomic_request(client):
assert Pet.objects.count() == 0
finally:
connection.settings_dict["ATOMIC_MUTATIONS"] = old_atomic_mutations
connection.settings_dict["ATOMIC_REQUESTS"] = old_atomic_requests
graphene_settings.ATOMIC_MUTATIONS = old_graphene_atomic_mutations
@patch("graphene_django.settings.graphene_settings.ATOMIC_MUTATIONS", False)
@patch.dict(
connection.settings_dict, {"ATOMIC_MUTATIONS": False, "ATOMIC_REQUESTS": False}
)
def test_model_form_mutation_multiple_creation_invalid_non_atomic(client):
old_atomic_mutations = connection.settings_dict.get("ATOMIC_MUTATIONS", False)
old_atomic_requests = connection.settings_dict["ATOMIC_REQUESTS"]
old_graphene_atomic_mutations = graphene_settings.ATOMIC_MUTATIONS
try:
connection.settings_dict["ATOMIC_MUTATIONS"] = False
connection.settings_dict["ATOMIC_REQUESTS"] = False
graphene_settings.ATOMIC_MUTATIONS = False
query = """
mutation PetMutations {
petMutation1: petMutation(input: { name: "Mia", age: 99 }) {
@ -858,45 +809,22 @@ def test_model_form_mutation_multiple_creation_invalid_non_atomic(client):
assert pet.name == "Enzo"
assert pet.age == 0
finally:
connection.settings_dict["ATOMIC_MUTATIONS"] = old_atomic_mutations
connection.settings_dict["ATOMIC_REQUESTS"] = old_atomic_requests
graphene_settings.ATOMIC_MUTATIONS = old_graphene_atomic_mutations
@patch("graphene_django.utils.utils.transaction.set_rollback")
@patch("graphene_django.settings.graphene_settings.ATOMIC_MUTATIONS", False)
@patch.dict(
connection.settings_dict, {"ATOMIC_MUTATIONS": False, "ATOMIC_REQUESTS": True}
)
def test_query_errors_atomic_request(set_rollback_mock, client):
old_atomic_mutations = connection.settings_dict.get("ATOMIC_MUTATIONS", False)
old_atomic_requests = connection.settings_dict["ATOMIC_REQUESTS"]
old_graphene_atomic_mutations = graphene_settings.ATOMIC_MUTATIONS
try:
connection.settings_dict["ATOMIC_MUTATIONS"] = False
connection.settings_dict["ATOMIC_REQUESTS"] = True
graphene_settings.ATOMIC_MUTATIONS = False
client.get(url_string(query="force error"))
set_rollback_mock.assert_called_once_with(True)
finally:
connection.settings_dict["ATOMIC_MUTATIONS"] = old_atomic_mutations
connection.settings_dict["ATOMIC_REQUESTS"] = old_atomic_requests
graphene_settings.ATOMIC_MUTATIONS = old_graphene_atomic_mutations
@patch("graphene_django.utils.utils.transaction.set_rollback")
@patch("graphene_django.settings.graphene_settings.ATOMIC_MUTATIONS", False)
@patch.dict(
connection.settings_dict, {"ATOMIC_MUTATIONS": False, "ATOMIC_REQUESTS": False}
)
def test_query_errors_non_atomic(set_rollback_mock, client):
old_atomic_mutations = connection.settings_dict.get("ATOMIC_MUTATIONS", False)
old_atomic_requests = connection.settings_dict["ATOMIC_REQUESTS"]
old_graphene_atomic_mutations = graphene_settings.ATOMIC_MUTATIONS
try:
connection.settings_dict["ATOMIC_MUTATIONS"] = False
connection.settings_dict["ATOMIC_REQUESTS"] = False
graphene_settings.ATOMIC_MUTATIONS = False
client.get(url_string(query="force error"))
set_rollback_mock.assert_not_called()
finally:
connection.settings_dict["ATOMIC_MUTATIONS"] = old_atomic_mutations
connection.settings_dict["ATOMIC_REQUESTS"] = old_atomic_requests
graphene_settings.ATOMIC_MUTATIONS = old_graphene_atomic_mutations