Migrate tests to pytest (#23544)
* Migrate tests to pytest * Fixup * Use loop fixture in test_check_config * Lint
This commit is contained in:
parent
d71424f285
commit
407e0c58f9
25 changed files with 4744 additions and 4910 deletions
|
@ -1,6 +1,5 @@
|
|||
"""Test Home Assistant ruamel.yaml loader."""
|
||||
import os
|
||||
import unittest
|
||||
from tempfile import mkdtemp
|
||||
import pytest
|
||||
|
||||
|
@ -114,45 +113,51 @@ views:
|
|||
cards: !include cards.yaml
|
||||
"""
|
||||
|
||||
TMP_DIR = None
|
||||
|
||||
class TestYAML(unittest.TestCase):
|
||||
"""Test lovelace.yaml save and load."""
|
||||
|
||||
def setUp(self):
|
||||
"""Set up for tests."""
|
||||
self.tmp_dir = mkdtemp()
|
||||
self.yaml = YAML(typ='rt')
|
||||
def setup():
|
||||
"""Set up for tests."""
|
||||
global TMP_DIR
|
||||
TMP_DIR = mkdtemp()
|
||||
|
||||
def tearDown(self):
|
||||
"""Clean up after tests."""
|
||||
for fname in os.listdir(self.tmp_dir):
|
||||
os.remove(os.path.join(self.tmp_dir, fname))
|
||||
os.rmdir(self.tmp_dir)
|
||||
|
||||
def _path_for(self, leaf_name):
|
||||
return os.path.join(self.tmp_dir, leaf_name+".yaml")
|
||||
def teardown():
|
||||
"""Clean up after tests."""
|
||||
for fname in os.listdir(TMP_DIR):
|
||||
os.remove(os.path.join(TMP_DIR, fname))
|
||||
os.rmdir(TMP_DIR)
|
||||
|
||||
def test_save_and_load(self):
|
||||
"""Test saving and loading back."""
|
||||
fname = self._path_for("test1")
|
||||
open(fname, "w+").close()
|
||||
util_yaml.save_yaml(fname, self.yaml.load(TEST_YAML_A))
|
||||
data = util_yaml.load_yaml(fname, True)
|
||||
assert data == self.yaml.load(TEST_YAML_A)
|
||||
|
||||
def test_overwrite_and_reload(self):
|
||||
"""Test that we can overwrite an existing file and read back."""
|
||||
fname = self._path_for("test2")
|
||||
open(fname, "w+").close()
|
||||
util_yaml.save_yaml(fname, self.yaml.load(TEST_YAML_A))
|
||||
util_yaml.save_yaml(fname, self.yaml.load(TEST_YAML_B))
|
||||
data = util_yaml.load_yaml(fname, True)
|
||||
assert data == self.yaml.load(TEST_YAML_B)
|
||||
def _path_for(leaf_name):
|
||||
return os.path.join(TMP_DIR, leaf_name+".yaml")
|
||||
|
||||
def test_load_bad_data(self):
|
||||
"""Test error from trying to load unserialisable data."""
|
||||
fname = self._path_for("test3")
|
||||
with open(fname, "w") as fh:
|
||||
fh.write(TEST_BAD_YAML)
|
||||
with pytest.raises(HomeAssistantError):
|
||||
util_yaml.load_yaml(fname, True)
|
||||
|
||||
def test_save_and_load():
|
||||
"""Test saving and loading back."""
|
||||
yaml = YAML(typ='rt')
|
||||
fname = _path_for("test1")
|
||||
open(fname, "w+").close()
|
||||
util_yaml.save_yaml(fname, yaml.load(TEST_YAML_A))
|
||||
data = util_yaml.load_yaml(fname, True)
|
||||
assert data == yaml.load(TEST_YAML_A)
|
||||
|
||||
|
||||
def test_overwrite_and_reload():
|
||||
"""Test that we can overwrite an existing file and read back."""
|
||||
yaml = YAML(typ='rt')
|
||||
fname = _path_for("test2")
|
||||
open(fname, "w+").close()
|
||||
util_yaml.save_yaml(fname, yaml.load(TEST_YAML_A))
|
||||
util_yaml.save_yaml(fname, yaml.load(TEST_YAML_B))
|
||||
data = util_yaml.load_yaml(fname, True)
|
||||
assert data == yaml.load(TEST_YAML_B)
|
||||
|
||||
|
||||
def test_load_bad_data():
|
||||
"""Test error from trying to load unserialisable data."""
|
||||
fname = _path_for("test3")
|
||||
with open(fname, "w") as fh:
|
||||
fh.write(TEST_BAD_YAML)
|
||||
with pytest.raises(HomeAssistantError):
|
||||
util_yaml.load_yaml(fname, True)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue