def _test_email_default_behavior(self, smtp_mock):
#with patch('sock.SMTP', autospec=True) as mock_smtp:
#define mock behavior
mock_smtp_object = MagicMock()
smtp_mock.side_effect = mock_smtp_object
t = thread_wrap_windmill('EmailWindmill', argv=[])
assert t
try:
t.start()
self.assertTrue(t.is_alive(),
'The EmailWindmill instance should have started.')
self.sock_map['PUSH'].send(
'{"type":"email_request","payload" : {"msg" : "Hi, this is a message that has been delivered through email service as a test.\nNeeded '
'multi '
'recipient test.\n\n share and enjoy", "subject" : "A message from email send service.", "sender" : "*****@*****.**",'
'"to" : ["*****@*****.**"]}}')
time.sleep(1)
finally:
t.windmill.kill()
t.join(3)
self.assertFalse(
t.is_alive(),
'The EmailWindmill instance should have shut down.')
def test_conrnerstone_default_behavior(self):
foo = Cornerstone()
t = thread_wrap_windmill('Cornerstone', argv=None)
t.start()
assert t.is_alive()
# send kill command
kill_cornerstone()
t.join(3)
assert not t.is_alive()
def _deliver_the_message(self, msg=None, args=list(), sock_type='PUSH'):
t = thread_wrap_windmill('CliListener', args)
try:
t.start()
self.assertTrue(t.is_alive())
self.sock_map[sock_type].send(msg)
time.sleep(1)
finally:
t.windmill.kill()
t.join(3)
self.assertFalse(t.is_alive(),
'CliListener instance should have shutdown.')
def _deliver_the_message(self, msg=None, args=list(), sock_type='PUSH'):
t = thread_wrap_windmill('CliListener', args)
try:
t.start()
self.assertTrue(t.is_alive())
self.sock_map[sock_type].send(msg)
time.sleep(1)
finally:
t.windmill.kill()
t.join(3)
self.assertFalse(t.is_alive(),
'CliListener instance should have shutdown.')
def test_cli_listener_default_behavior(self):
t = thread_wrap_windmill('CliListener')
try:
t.start()
self.assertTrue(t.is_alive(),
'The CliEmitter instance should have started.')
self.sock_map['PUSH'].send('Hello')
time.sleep(1)
finally:
t.windmill.kill()
t.join(3)
self.assertFalse(t.is_alive(),
'The CliEmitter instance should have shut down.')
def test_cli_listener_default_behavior(self):
t = thread_wrap_windmill('CliListener')
try:
t.start()
self.assertTrue(t.is_alive(),
'The CliEmitter instance should have started.')
self.sock_map['PUSH'].send('Hello')
time.sleep(1)
finally:
t.windmill.kill()
t.join(3)
self.assertFalse(t.is_alive(),
'The CliEmitter instance should have shut down.')
def test_cli_emitter_default_hehaviour(self):
archive_file, output_file = gen_archive_output_pair(
'cli_emitter_default_behavior')
emitter = CliEmitter()
assert emitter
# create the listener and direct it's output to out_file
t = thread_wrap_windmill('CliListener', argv=[
'-f', output_file,
'--input_sock_url', 'tcp://localhost:6677'])
assert t
self.emit_message(emitter, t)
self.assertFiles(archive_file, output_file)
def executor(self, test_name=None, emitter_args=None, listener_args=None):
assert test_name
archive_file, output_file = gen_archive_output_pair(test_name)
emitter = CliEmitter(argv=emitter_args)
assert emitter
# add the output file to the listener arguments
listener_args = ['-f', output_file] + listener_args
# create the listener and direct it's output to out_file
t = thread_wrap_windmill('CliListener', argv=listener_args)
assert t
self.emit_message(emitter, t)
self.assertFiles(archive_file, output_file)
def test_cli_emitter_default_hehaviour(self):
archive_file, output_file = gen_archive_output_pair(
'cli_emitter_default_behavior')
emitter = CliEmitter()
assert emitter
# create the listener and direct it's output to out_file
t = thread_wrap_windmill('CliListener',
argv=[
'-f', output_file, '--input_sock_url',
'tcp://localhost:6677'
])
assert t
self.emit_message(emitter, t)
self.assertFiles(archive_file, output_file)
def executor(self,
test_name=None,
emitter_args=None,
listener_args=None):
assert test_name
archive_file, output_file = gen_archive_output_pair(test_name)
emitter = CliEmitter(argv=emitter_args)
assert emitter
# add the output file to the listener arguments
listener_args = ['-f', output_file] + listener_args
# create the listener and direct it's output to out_file
t = thread_wrap_windmill('CliListener', argv=listener_args)
assert t
self.emit_message(emitter, t)
self.assertFiles(archive_file, output_file)
def test_echo_service_options(self):
req_out_sock = self.zmq_ctx.socket(REQ)
req_out_sock.bind('tcp://*:8899')
argv = ['-m', 'pong', '--reply_sock_url', 'tcp://localhost:8899']
t = thread_wrap_windmill('EchoService', argv=argv)
try:
t.start()
self.assertTrue(t.is_alive(),
'The EchoService instance should have started.')
req_out_sock.send('ping')
msg = req_out_sock.recv()
self.assertEqual('pong', msg)
finally:
t.windmill.kill()
t.join(3)
self.assertFalse(t.is_alive(),
'The EchoService instance shold have shutdown.')
req_out_sock.close()
def test_echo_service_default_behavior(self):
req_out_sock = self.zmq_ctx.socket(REQ)
req_out_sock.bind('tcp://*:8889')
t = thread_wrap_windmill('EchoService')
try:
t.start()
self.assertTrue(t.is_alive,
'The EchoService instance should have started.')
req_out_sock.send("echo, echo, echo")
msg = req_out_sock.recv()
self.assertEqual("echo, echo, echo", msg)
finally:
t.windmill.kill()
t.join(3)
self.assertFalse(t.is_alive(),
'The EchoService instance should have shutdown.')
req_out_sock.close()
def _test_email_sub_behavior(self):
#with patch('sock.SMTP', autospec=True) as mock_smtp:
#define mock behavior
mock_smtp_object = MagicMock()
#smtp_mock.SMTP.side_effect = mock_smtp_object
args = '--input_sock_url tcp://*:6679 --input_bind --input_sock_type SUB --verbose --monitor_stream'
t = thread_wrap_windmill('EmailWindmill', argv=args.split())
assert t
try:
t.start()
self.assertTrue(t.is_alive(),
'The EmailWindmill instance should have started.')
self.sock_map['PUB'].send(
'{"type":"email_request","payload" : {"msg" : "Hi, this is a message that has been delivered through email service as a test. \\nNeeded multi recipient test. share and enjoy", "subject" : "A message from email send service.", "sender" : "*****@*****.**","to" : ["*****@*****.**"]}}')
time.sleep(1)
finally:
t.windmill.kill()
t.join(3)
self.assertFalse(t.is_alive(),
'The EmailWindmill instance should have shut down.')
def _test_email_sub_behavior(self):
#with patch('sock.SMTP', autospec=True) as mock_smtp:
#define mock behavior
mock_smtp_object = MagicMock()
#smtp_mock.SMTP.side_effect = mock_smtp_object
args = '--input_sock_url tcp://*:6679 --input_bind --input_sock_type SUB --verbose --monitor_stream'
t = thread_wrap_windmill('EmailWindmill', argv=args.split())
assert t
try:
t.start()
self.assertTrue(t.is_alive(),
'The EmailWindmill instance should have started.')
self.sock_map['PUB'].send(
'{"type":"email_request","payload" : {"msg" : "Hi, this is a message that has been delivered through email service as a test. \\nNeeded multi recipient test. share and enjoy", "subject" : "A message from email send service.", "sender" : "*****@*****.**","to" : ["*****@*****.**"]}}'
)
time.sleep(1)
finally:
t.windmill.kill()
t.join(3)
self.assertFalse(
t.is_alive(),
'The EmailWindmill instance should have shut down.')
def _test_email_default_behavior(self, smtp_mock):
#with patch('sock.SMTP', autospec=True) as mock_smtp:
#define mock behavior
mock_smtp_object = MagicMock()
smtp_mock.side_effect = mock_smtp_object
t = thread_wrap_windmill('EmailWindmill', argv=[])
assert t
try:
t.start()
self.assertTrue(t.is_alive(),
'The EmailWindmill instance should have started.')
self.sock_map['PUSH'].send(
'{"type":"email_request","payload" : {"msg" : "Hi, this is a message that has been delivered through email service as a test.\nNeeded '
'multi '
'recipient test.\n\n share and enjoy", "subject" : "A message from email send service.", "sender" : "*****@*****.**",'
'"to" : ["*****@*****.**"]}}')
time.sleep(1)
finally:
t.windmill.kill()
t.join(3)
self.assertFalse(t.is_alive(),
'The EmailWindmill instance should have shut down.')
