Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- @patch('__builtin__.open')
- @patch.object(os.path, 'join')
- @patch.object(os, 'getcwd')
- def test_002_readfile_exist_with_local_provided\
- (self, mk_getcwd, mk_join, mk_open):
- """
- Tests reading a file and verify its content with local
- provided.
- """
- # setup mocks
- mk_getcwd.return_value = None
- mk_join.return_value = '/tmp/test/dummy.py'
- mk_open.return_value = MagicMock(spec=file)
- m_file = mk_open.return_value.__enter__.return_value
- m_file.read.return_value = 'hello world'
- # assert function output is 'hello world'
- self.assertEqual(
- readfile(self.source_f, **self.kwargs),
- 'hello world')
- # assert mk_getcwd
- self.assertEqual(mk_getcwd.called, True)
- self.assertEqual(mk_getcwd.call_args_list, [call()])
- # assert mk_join
- self.assertEqual(mk_join.call_count, 1)
- join_call_list = [call(self.kwargs['local'], 'dummy.py')]
- self.assertEqual(mk_join.call_args_list, join_call_list)
- # assert mk_open
- self.assertEqual(mk_open.call_count, 1)
- open_call_list = [call('/tmp/test/dummy.py', 'r')]
- self.assertEqual(mk_open.call_args_list, open_call_list)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement