python code to test function raise exceptions if called incorrectly

neeraj1909 · neeraj1909 · commit 20125f34651a · 2018-07-03T18:52:02.000+05:30
diff --git a/Python_testing_with_pytest/Chapter-2 Writing Test Functions/tasks_proj/tests/func/test_api_exceptions.py b/Python_testing_with_pytest/Chapter-2 Writing Test Functions/tasks_proj/tests/func/test_api_exceptions.py
@@ -0,0 +1,54 @@
+"""Test for expected exceptions from using the API wrong."""
+
+import pytest
+import tasks
+
+
+def test_add_raises():
+    """add() should raise an exception with wrong type param."""
+    with pytest.raises(TypeError):
+        tasks.add(task='not a Task object')
+
+
+@pytest.mark.smoke
+def test_list_raises():
+    """list() should raise an exception with wrong type param."""
+    with pytest.raises(TypeError):
+        tasks.list_tasks(owner=123)
+
+
+@pytest.mark.get
+@pytest.mark.smoke
+def test_get_raises():
+    """get() should raise an exception with wrong type param."""
+    with pytest.raises(TypeError):
+        tasks.get(task_id='123')
+
+
+class TestUpdate():
+    """Test expected exceptions with tasks.update()."""
+
+    def test_bad_id(self):
+        """A non-int id should raise an excption."""
+        with pytest.raises(TypeError):
+            tasks.update(task_id={'dict instead': 1},
+                         task=tasks.Task())
+
+    def test_bad_task(self):
+        """A non-Task task should raise an excption."""
+        with pytest.raises(TypeError):
+            tasks.update(task_id=1, task='not a task')
+
+
+def test_delete_raises():
+    """delete() should raise an exception with wrong type param."""
+    with pytest.raises(TypeError):
+        tasks.delete(task_id=(1, 2, 3))
+
+
+def test_start_tasks_db_raises():
+    """Make sure unsupported db raises an exception."""
+    with pytest.raises(ValueError) as excinfo:
+        tasks.start_tasks_db('some/great/path', 'mysql')
+    exception_msg = excinfo.value.args[0]
+    assert exception_msg == "db_type must be a 'tiny' or 'mongo'"
