From 1bdceb351b50b805262978bc6cb2e0e0dbcdb089 Mon Sep 17 00:00:00 2001
From: Jon Bernard <jbernard@tuxion.com>
Date: Fri, 15 Jan 2016 16:07:24 -0500
Subject: Improve custom exceptions

---
 test_dotfiles.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'test_dotfiles.py')

diff --git a/test_dotfiles.py b/test_dotfiles.py
index caff325..7cfb276 100644
--- a/test_dotfiles.py
+++ b/test_dotfiles.py
@@ -1,6 +1,7 @@
 import pytest
 
 from dotfiles import Repository, Dotfile, cli
+from dotfiles import IsSymlink
 
 
 class TestCli(object):
@@ -116,8 +117,7 @@ class TestDotfile(object):
         assert name.samefile(target)
 
         for x in range(2, times):
-            with pytest.raises(OSError):
-                # TODO: verify exception type once those exists
+            with pytest.raises(IsSymlink):
                 Dotfile(name, target).add()
             assert target.check(file=1, link=0)
             assert name.check(file=1, link=1)
-- 
cgit v1.2.3