Merge multiple context managers in tests (#48146)
This commit is contained in:
parent
79af18a8ab
commit
ad13a9295e
33 changed files with 590 additions and 627 deletions
|
@ -13,29 +13,30 @@ def test_config_google_home_entity_id_to_number():
|
|||
with patch(
|
||||
"homeassistant.components.emulated_hue.load_json",
|
||||
return_value={"1": "light.test2"},
|
||||
) as json_loader:
|
||||
with patch("homeassistant.components.emulated_hue.save_json") as json_saver:
|
||||
number = conf.entity_id_to_number("light.test")
|
||||
assert number == "2"
|
||||
) as json_loader, patch(
|
||||
"homeassistant.components.emulated_hue.save_json"
|
||||
) as json_saver:
|
||||
number = conf.entity_id_to_number("light.test")
|
||||
assert number == "2"
|
||||
|
||||
assert json_saver.mock_calls[0][1][1] == {
|
||||
"1": "light.test2",
|
||||
"2": "light.test",
|
||||
}
|
||||
assert json_saver.mock_calls[0][1][1] == {
|
||||
"1": "light.test2",
|
||||
"2": "light.test",
|
||||
}
|
||||
|
||||
assert json_saver.call_count == 1
|
||||
assert json_loader.call_count == 1
|
||||
assert json_saver.call_count == 1
|
||||
assert json_loader.call_count == 1
|
||||
|
||||
number = conf.entity_id_to_number("light.test")
|
||||
assert number == "2"
|
||||
assert json_saver.call_count == 1
|
||||
number = conf.entity_id_to_number("light.test")
|
||||
assert number == "2"
|
||||
assert json_saver.call_count == 1
|
||||
|
||||
number = conf.entity_id_to_number("light.test2")
|
||||
assert number == "1"
|
||||
assert json_saver.call_count == 1
|
||||
number = conf.entity_id_to_number("light.test2")
|
||||
assert number == "1"
|
||||
assert json_saver.call_count == 1
|
||||
|
||||
entity_id = conf.number_to_entity_id("1")
|
||||
assert entity_id == "light.test2"
|
||||
entity_id = conf.number_to_entity_id("1")
|
||||
assert entity_id == "light.test2"
|
||||
|
||||
|
||||
def test_config_google_home_entity_id_to_number_altered():
|
||||
|
@ -47,28 +48,29 @@ def test_config_google_home_entity_id_to_number_altered():
|
|||
with patch(
|
||||
"homeassistant.components.emulated_hue.load_json",
|
||||
return_value={"21": "light.test2"},
|
||||
) as json_loader:
|
||||
with patch("homeassistant.components.emulated_hue.save_json") as json_saver:
|
||||
number = conf.entity_id_to_number("light.test")
|
||||
assert number == "22"
|
||||
assert json_saver.call_count == 1
|
||||
assert json_loader.call_count == 1
|
||||
) as json_loader, patch(
|
||||
"homeassistant.components.emulated_hue.save_json"
|
||||
) as json_saver:
|
||||
number = conf.entity_id_to_number("light.test")
|
||||
assert number == "22"
|
||||
assert json_saver.call_count == 1
|
||||
assert json_loader.call_count == 1
|
||||
|
||||
assert json_saver.mock_calls[0][1][1] == {
|
||||
"21": "light.test2",
|
||||
"22": "light.test",
|
||||
}
|
||||
assert json_saver.mock_calls[0][1][1] == {
|
||||
"21": "light.test2",
|
||||
"22": "light.test",
|
||||
}
|
||||
|
||||
number = conf.entity_id_to_number("light.test")
|
||||
assert number == "22"
|
||||
assert json_saver.call_count == 1
|
||||
number = conf.entity_id_to_number("light.test")
|
||||
assert number == "22"
|
||||
assert json_saver.call_count == 1
|
||||
|
||||
number = conf.entity_id_to_number("light.test2")
|
||||
assert number == "21"
|
||||
assert json_saver.call_count == 1
|
||||
number = conf.entity_id_to_number("light.test2")
|
||||
assert number == "21"
|
||||
assert json_saver.call_count == 1
|
||||
|
||||
entity_id = conf.number_to_entity_id("21")
|
||||
assert entity_id == "light.test2"
|
||||
entity_id = conf.number_to_entity_id("21")
|
||||
assert entity_id == "light.test2"
|
||||
|
||||
|
||||
def test_config_google_home_entity_id_to_number_empty():
|
||||
|
@ -79,25 +81,26 @@ def test_config_google_home_entity_id_to_number_empty():
|
|||
|
||||
with patch(
|
||||
"homeassistant.components.emulated_hue.load_json", return_value={}
|
||||
) as json_loader:
|
||||
with patch("homeassistant.components.emulated_hue.save_json") as json_saver:
|
||||
number = conf.entity_id_to_number("light.test")
|
||||
assert number == "1"
|
||||
assert json_saver.call_count == 1
|
||||
assert json_loader.call_count == 1
|
||||
) as json_loader, patch(
|
||||
"homeassistant.components.emulated_hue.save_json"
|
||||
) as json_saver:
|
||||
number = conf.entity_id_to_number("light.test")
|
||||
assert number == "1"
|
||||
assert json_saver.call_count == 1
|
||||
assert json_loader.call_count == 1
|
||||
|
||||
assert json_saver.mock_calls[0][1][1] == {"1": "light.test"}
|
||||
assert json_saver.mock_calls[0][1][1] == {"1": "light.test"}
|
||||
|
||||
number = conf.entity_id_to_number("light.test")
|
||||
assert number == "1"
|
||||
assert json_saver.call_count == 1
|
||||
number = conf.entity_id_to_number("light.test")
|
||||
assert number == "1"
|
||||
assert json_saver.call_count == 1
|
||||
|
||||
number = conf.entity_id_to_number("light.test2")
|
||||
assert number == "2"
|
||||
assert json_saver.call_count == 2
|
||||
number = conf.entity_id_to_number("light.test2")
|
||||
assert number == "2"
|
||||
assert json_saver.call_count == 2
|
||||
|
||||
entity_id = conf.number_to_entity_id("2")
|
||||
assert entity_id == "light.test2"
|
||||
entity_id = conf.number_to_entity_id("2")
|
||||
assert entity_id == "light.test2"
|
||||
|
||||
|
||||
def test_config_alexa_entity_id_to_number():
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue