Increase test coverage

.coveragerc

@@ -0,0 +1,2 @@
+[run]
+omit = charlatan/_compat.py

charlatan/file_format.py

@@ -71,6 +71,6 @@ def load_file(filename):
         configure_yaml()
         content = yaml.load(content)
     else:
-        raise KeyError("Unsupported filetype: '%s'" % filename)
+        raise ValueError("Unsupported filetype: '%s'" % filename)
 
     return content

charlatan/tests/data/dependencies.yaml

@@ -2,7 +2,6 @@ fixture1:
     fields:
         foo: bar
 
-
 fixture2:
     fields:
         foo: !rel fixture1

charlatan/tests/test_file_format.py

@@ -0,0 +1,9 @@
+import pytest
+
+from charlatan import file_format
+
+
+def test_non_yaml_file():
+    """Verify that we can't open a non-YAML file."""
+    with pytest.raises(ValueError):
+        file_format.load_file("./charlatan/tests/data/test.json")

charlatan/tests/test_fixture_collection.py

@@ -0,0 +1,31 @@
+import pytest
+
+from charlatan import FixturesManager
+
+
+@pytest.fixture(scope="module")
+def collection():
+    """Return FixtureCollection."""
+
+    manager = FixturesManager()
+    manager.load("docs/examples/collection.yaml")
+    return manager.fixture_collection.get("toasters")[0]
+
+
+def test_repr(collection):
+    """Verify the repr."""
+    assert repr(collection) == "<DictFixtureCollection 'toasters'>"
+
+
+def test_cant_get_invalid_format(collection):
+    """Verify that we can't get an invalid format."""
+
+    with pytest.raises(ValueError):
+        collection.get_all_instances(format="invalid")
+
+
+def test_cant_get_missing_fixture(collection):
+    """Verify that we can't get a missing fixture."""
+
+    with pytest.raises(KeyError):
+        collection.get("missing")

charlatan/tests/test_fixtures_manager.py

@@ -1,5 +1,7 @@
 from __future__ import absolute_import
 
+import pytest
+
 from charlatan import testing
 from charlatan import depgraph
 from charlatan import FixturesManager
@@ -108,3 +110,15 @@ def test_constructs_ancestors(self):
         fm.get_fixture('fixture3')
         self.assertIn('fixture1', fm.cache)
         self.assertIn('fixture4', fm.cache)
+
+    def test_invalid_hook(self):
+        """Verify that can't set an invalid hook."""
+
+        manager = FixturesManager()
+        with pytest.raises(KeyError):
+            manager.set_hook("invalid", lambda p: p)
+
+    def test_set_hook(self):
+        """Verify that we can set a hook."""
+        manager = FixturesManager()
+        manager.set_hook("before_save", lambda p: p)