Skip to content

Commit

Permalink
Increase coverage a bit
Browse files Browse the repository at this point in the history
  • Loading branch information
Rob Harrigan committed Feb 27, 2014
1 parent 1e6dd97 commit 0025302
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 0 deletions.
10 changes: 10 additions & 0 deletions tests/mock_ftp.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,12 +33,16 @@ def rmd(self, dirname):
return

def delete(self, filename):
if '//' in filename:
raise Exception("Doesn't exist")
return True

def rename(self, fromname, toname):
return

def cwd(self, pathname):
if '//' in pathname:
raise Exception("Doesn't exist")
self._current = pathname

def size(self, filename):
Expand All @@ -54,6 +58,12 @@ def sendcmd(self, command):
def set_pasv(self, passive):
return passive

def quit(self):
raise Exception('Fake a a problem with quit')

def close(self):
return True

def _set_files(self, files):
self._files = files

Expand Down
4 changes: 4 additions & 0 deletions tests/test_ftpretty.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@ def setUp(self):
def test_cd(self):
self.pretty.cd('photos/nature/mountains')
self.assertEquals(self.pretty.pwd(), 'photos/nature/mountains')
self.assertRaises(Exception, self.pretty.cd('photos//nature/mountains'))

def test_list(self):
self.mock_ftp._set_files(['a.txt', 'b.txt'])
Expand All @@ -29,6 +30,7 @@ def test_get(self):

def test_delete(self):
self.assertTrue(self.pretty.delete('remote_file.txt'))
self.assertRaises(Exception, self.pretty.delete('photos//nature/remote.txt'))

def test_dir_parse(self):
self.mock_ftp._set_dirlist("-rw-rw-r-- 1 rharrigan www 47 Feb 20 11:39 Cool.txt\n" +
Expand All @@ -53,6 +55,8 @@ def test_fallthrough(self):
def test_set_pasv(self):
pretty = ftpretty(None, None, None, ftp_conn=self.mock_ftp, passive=False)

def test_close(self):
self.pretty.close()

def suite():
loader = unittest.TestLoader()
Expand Down

0 comments on commit 0025302

Please sign in to comment.