def test_exists_yes(self, mock_open): mock_f = mock_open.return_value.__enter__.return_value mock_f.read.return_value = Mock() mock_f.read.return_value.decode.return_value = '{"a":1}' out = self.filesystem_file_url.exists() mock_open.assert_called_with('/topdir/bottomdir/file', 'rb') self.assertEqual(out, True)
def test_upload_fileobj(self, mock_copyfileobj, mock_open): mock_fileobj = Mock(name='fileobj') self.filesystem_file_url.upload_fileobj(mock_fileobj) mock_copyfileobj.assert_called_with(mock_fileobj, mock_open.return_value) mock_open.assert_called_with('/topdir/bottomdir/file', 'wb')
def test_gladius_get_lines(self, mock_open): Event = namedtuple("Event", ['src_path']) src_path = 'test_src_path' event = Event(src_path) self.handler.get_lines(event) mock_open.assert_called_with(event.src_path, 'r')
def test_store_string(self, mock_open): mock_string = Mock(name='string') self.filesystem_file_url.store_string(mock_string) file_loc = Mock(name='file_loc') file_loc.local_file_path = '/my/file' mock_open.assert_called_with('/topdir/bottomdir/file', 'wb') mock_open.return_value.write.assert_called_with(mock_string.encode.return_value)
def test_gladius_handler_on_created_directory_true(self, mock_os, mock_open, mock_md5): mock_os.path.isdir.return_value = False Directory = namedtuple("Directory", ['src_path']) src_path = 'test_src_path' my_dir = Directory(src_path) self.handler.on_created(my_dir) mock_open.assert_called_with(src_path, 'r')
def test_displaySm(monkeypatch): mock_open = mock.Mock() mock_json = mock.Mock() mock_file = mock.Mock() mock_open.return_value = mock_file default_file = './appControl/socialMediaControl/smName.txt' smObject = socialMediaControl.socialMediaController() monkeypatch.setattr(__builtin__, 'open', mock_open) monkeypatch.setattr(json, 'load', mock_json) retValue = smObject.displaySm() mock_open.assert_called_with(default_file) mock_json.assert_called_with(mock_file) monkeypatch.undo()
def test_displaySm(monkeypatch): mock_open = mock.Mock() mock_json = mock.Mock() mock_file = mock.Mock() mock_open.return_value = mock_file default_file = './appControl/socialMediaControl/smName.txt' smObject = socialMediaControl.socialMediaController() monkeypatch.setattr(__builtin__,'open',mock_open) monkeypatch.setattr(json,'load',mock_json) retValue = smObject.displaySm() mock_open.assert_called_with(default_file) mock_json.assert_called_with(mock_file) monkeypatch.undo()
def test_rmSm_1(monkeypatch): mock_json = mock.Mock() mock_open = mock.Mock() default_file = './appControl/socialMediaControl/smName.txt' mock_display = mock.Mock() mock_display.return_value = ['facebook'] smObject = socialMediaControl.socialMediaController() mock_file = mock.Mock() mock_open.return_value = mock_file monkeypatch.setattr(smObject, 'displaySm', mock_display) monkeypatch.setattr(__builtin__, 'open', mock_open) monkeypatch.setattr(json, 'dump', mock_json) assert smObject.displaySm() == ['facebook'] retValue = smObject.rmSm(['facebook']) mock_open.assert_called_with(default_file, 'w') mock_json.assert_called_with([], mock_file) monkeypatch.undo()
def test_rmSm_1(monkeypatch): mock_json = mock.Mock() mock_open = mock.Mock() default_file = './appControl/socialMediaControl/smName.txt' mock_display = mock.Mock() mock_display.return_value = ['facebook'] smObject = socialMediaControl.socialMediaController() mock_file = mock.Mock() mock_open.return_value = mock_file monkeypatch.setattr(smObject,'displaySm',mock_display) monkeypatch.setattr(__builtin__,'open',mock_open) monkeypatch.setattr(json,'dump',mock_json) assert smObject.displaySm() == ['facebook'] retValue = smObject.rmSm(['facebook']) mock_open.assert_called_with(default_file,'w') mock_json.assert_called_with([],mock_file) monkeypatch.undo()
def test_exists_no(self, mock_open): mock_open.side_effect = FileNotFoundError out = self.filesystem_file_url.exists() mock_open.assert_called_with('/topdir/bottomdir/file', 'rb') self.assertEqual(out, False)