39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
import pytest
|
|
|
|
from rtv.clipboard import copy_linux, copy_osx
|
|
from rtv.exceptions import ProgramError
|
|
|
|
|
|
try:
|
|
from unittest import mock
|
|
except ImportError:
|
|
import mock
|
|
|
|
|
|
def test_copy():
|
|
|
|
with mock.patch('subprocess.Popen') as Popen, \
|
|
mock.patch('subprocess.call') as call:
|
|
|
|
# Mock out the subprocess calls
|
|
p = mock.Mock()
|
|
p.communicate = mock.Mock()
|
|
Popen.return_value = p
|
|
|
|
# If the `which` command can't find a program to use
|
|
call.return_value = 1 # Returns an error code
|
|
with pytest.raises(ProgramError):
|
|
copy_linux('test')
|
|
|
|
call.return_value = 0
|
|
copy_linux('test')
|
|
assert Popen.call_args[0][0] == ['xsel', '-b', '-i']
|
|
p.communicate.assert_called_with(input='test'.encode('utf-8'))
|
|
copy_linux('test ❤')
|
|
p.communicate.assert_called_with(input='test ❤'.encode('utf-8'))
|
|
|
|
copy_osx('test')
|
|
assert Popen.call_args[0][0] == ['pbcopy', 'w']
|
|
p.communicate.assert_called_with(input='test'.encode('utf-8'))
|
|
copy_osx('test ❤')
|
|
p.communicate.assert_called_with(input='test ❤'.encode('utf-8')) |