107 lines
2.7 KiB
Python
107 lines
2.7 KiB
Python
|
import mock
|
||
|
|
|
||
|
|
class ScpSenderTest(mock.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
self.setupModules([])
|
||
|
|
self.fs = mock.DiskIO()
|
||
|
|
|
||
|
|
def tearDown(self):
|
||
|
|
self.tearDownModules()
|
||
|
|
|
||
|
|
def set_host_1_test(self):
|
||
|
|
import log_picker.sending.scpsender as scp_s
|
||
|
|
|
||
|
|
HOST = "localhost"
|
||
|
|
|
||
|
|
obj = scp_s.ScpSender()
|
||
|
|
obj.set_host(HOST)
|
||
|
|
|
||
|
|
self.assertEqual(HOST, obj.host)
|
||
|
|
|
||
|
|
def set_host_2_test(self):
|
||
|
|
import log_picker.sending.scpsender as scp_s
|
||
|
|
|
||
|
|
HOST = "localhost"
|
||
|
|
PORT = 12345
|
||
|
|
ADDRESS = "%s:%s" % (HOST, PORT)
|
||
|
|
|
||
|
|
obj = scp_s.ScpSender()
|
||
|
|
obj.set_host(ADDRESS)
|
||
|
|
|
||
|
|
self.assertEqual(HOST, obj.host)
|
||
|
|
self.assertEqual(PORT, obj.port)
|
||
|
|
|
||
|
|
def set_host_3_test(self):
|
||
|
|
import log_picker.sending.scpsender as scp_s
|
||
|
|
|
||
|
|
HOST = "localhost"
|
||
|
|
PORT = "abc"
|
||
|
|
ADDRESS = "%s:%s" % (HOST, PORT)
|
||
|
|
|
||
|
|
obj = scp_s.ScpSender()
|
||
|
|
obj.set_host(ADDRESS)
|
||
|
|
|
||
|
|
self.assertEqual(HOST, obj.host)
|
||
|
|
self.assertEqual(None, obj.port)
|
||
|
|
|
||
|
|
def set_login_test(self):
|
||
|
|
import log_picker.sending.scpsender as scp_s
|
||
|
|
|
||
|
|
USERNAME = "foo"
|
||
|
|
|
||
|
|
obj = scp_s.ScpSender()
|
||
|
|
obj.set_login(USERNAME)
|
||
|
|
|
||
|
|
self.assertEqual(USERNAME, obj.username)
|
||
|
|
|
||
|
|
def set_path_1_test(self):
|
||
|
|
import log_picker.sending.scpsender as scp_s
|
||
|
|
|
||
|
|
PATH = "/tmp/something"
|
||
|
|
|
||
|
|
obj = scp_s.ScpSender()
|
||
|
|
obj.set_path(PATH)
|
||
|
|
|
||
|
|
self.assertEqual(PATH, obj.path)
|
||
|
|
|
||
|
|
def set_path_2_test(self):
|
||
|
|
import log_picker.sending.scpsender as scp_s
|
||
|
|
|
||
|
|
obj = scp_s.ScpSender()
|
||
|
|
obj.set_path(None)
|
||
|
|
|
||
|
|
self.assertEqual(".", obj.path)
|
||
|
|
|
||
|
|
def sendfile_test(self):
|
||
|
|
import log_picker.sending.scpsender as scp_s
|
||
|
|
|
||
|
|
FILE = "/tmp/file"
|
||
|
|
MIMETYPE = "text/plain"
|
||
|
|
USERNAME = "foo"
|
||
|
|
HOST = "localhost"
|
||
|
|
PATH = "/home/foo"
|
||
|
|
|
||
|
|
scp_s.subprocess = mock.Mock()
|
||
|
|
scp_s.subprocess.Popen().returncode = False
|
||
|
|
|
||
|
|
obj = scp_s.ScpSender()
|
||
|
|
obj.set_host(HOST)
|
||
|
|
obj.set_login(USERNAME)
|
||
|
|
obj.set_path(PATH)
|
||
|
|
obj.sendfile(FILE, MIMETYPE)
|
||
|
|
|
||
|
|
TARGET = "%s@%s:%s" % (USERNAME, HOST, PATH)
|
||
|
|
self.assertTrue(TARGET in scp_s.subprocess.Popen.call_args[0][0])
|
||
|
|
|
||
|
|
def sendfile_fail_1_test(self):
|
||
|
|
import log_picker.sending.scpsender as scp_s
|
||
|
|
|
||
|
|
FILE = "/tmp/file"
|
||
|
|
MIMETYPE = "text/plain"
|
||
|
|
|
||
|
|
scp_s.subprocess = mock.Mock()
|
||
|
|
|
||
|
|
obj = scp_s.ScpSender()
|
||
|
|
self.assertRaises(scp_s.SenderError, obj.sendfile, FILE, MIMETYPE)
|
||
|
|
|