def test_tail_really_pretty(self, stdout, argv):
msgs = []
def mock_tail(self, topic="", passive=False, **kw):
msg = dict(
msg=dict(hello="world"),
msg_id='2ad5aaf8-68af-4a6d-9196-2a8b43a73238',
timestamp=1354563717.472648, # Once upon a time...
topic="org.threebean.prod.testing",
)
yield ("name", "endpoint", "topic", msg)
config = {}
with mock.patch("fedmsg.__local", self.local):
with mock.patch("fedmsg.config.__cache", config):
with mock.patch("fedmsg.core.FedMsgContext.tail_messages",
mock_tail):
command = fedmsg.commands.tail.TailCommand()
command.execute()
output = stdout.getvalue()
expected = \
'\x1b[33m"hello"\x1b[39;49;00m:\x1b[39;49;00m \x1b[39;49;00m' + \
'\x1b[33m"world"\x1b[39;49;00m'
assert(expected in output)
def test_logger_json(self, stdout, argv):
test_input_dict = {"hello": "world"}
test_input = json.dumps(test_input_dict)
if six.PY3:
stdin = lambda: six.StringIO(test_input)
else:
stdin = lambda: six.StringIO(test_input.encode('utf-8'))
msgs = []
def mock_publish(context, topic=None, msg=None, modname=None):
msgs.append(msg)
config = {}
with mock.patch("fedmsg.__local", self.local):
with mock.patch("fedmsg.config.__cache", config):
with mock.patch("fedmsg.core.FedMsgContext.publish",
mock_publish):
with mock.patch("sys.stdin", new_callable=stdin):
command = LoggerCommand()
command.execute()
eq_(msgs, [test_input_dict])
def test_config_query(self, stdout, argv):
with mock.patch('fedmsg.config.__cache', {}):
config_command()
output = stdout.getvalue()
output_conf = json.loads(output)
with mock.patch('fedmsg.config.__cache', {}):
fedmsg_conf = fedmsg.config.load_config()
eq_(output_conf, fedmsg_conf["endpoints"])
def test_config_single_file(self, stdout, argv):
with mock.patch('fedmsg.config.__cache', {}):
config_command()
output = stdout.getvalue()
output_conf = json.loads(output)
with mock.patch('fedmsg.config.__cache', {}):
fedmsg_conf = fedmsg.config.load_config(
filenames=[CONF_FILE],
disable_defaults=True,
)
eq_(output_conf, fedmsg_conf)
def test_tail_basic(self, stdout, argv):
def mock_tail(self, topic="", passive=False, **kw):
yield ("name", "endpoint", "topic", dict(topic="topic"))
config = {}
with mock.patch("fedmsg.__local", self.local):
with mock.patch("fedmsg.config.__cache", config):
with mock.patch("fedmsg.core.FedMsgContext.tail_messages",
mock_tail):
command = fedmsg.commands.tail.TailCommand()
command.execute()
output = stdout.getvalue()
expected = "{'topic': 'topic'}\n"
assert(output.endswith(expected))
def test_relay(self, argv):
actual_options = []
def mock_main(options, consumers, framework):
actual_options.append(options)
config = {}
with mock.patch("fedmsg.__local", self.local):
with mock.patch("fedmsg.config.__cache", config):
with mock.patch("moksha.hub.main", mock_main):
command = fedmsg.commands.relay.RelayCommand()
command.execute()
actual_options = actual_options[0]
assert (fedmsg.consumers.relay.RelayConsumer.config_key
in actual_options)
assert (
actual_options[fedmsg.consumers.relay.RelayConsumer.config_key])
def test_relay(self, argv):
actual_options = []
def mock_main(options, consumers, framework):
actual_options.append(options)
config = {}
with mock.patch("fedmsg.__local", self.local):
with mock.patch("fedmsg.config.__cache", config):
with mock.patch("moksha.hub.main", mock_main):
command = fedmsg.commands.relay.RelayCommand()
command.execute()
actual_options = actual_options[0]
assert(
fedmsg.consumers.relay.RelayConsumer.config_key in actual_options
)
assert(
actual_options[fedmsg.consumers.relay.RelayConsumer.config_key]
)
def test_logger_basic(self, stdout, argv):
test_input = "a message for you"
if six.PY3:
stdin = lambda: six.StringIO(test_input)
else:
stdin = lambda: six.StringIO(test_input.encode('utf-8'))
msgs = []
def mock_publish(context, topic=None, msg=None, modname=None):
msgs.append(msg)
config = {}
with mock.patch("fedmsg.__local", self.local):
with mock.patch("fedmsg.config.__cache", config):
with mock.patch("fedmsg.core.FedMsgContext.publish", mock_publish):
with mock.patch("sys.stdin", new_callable=stdin):
command = LoggerCommand()
command.execute()
eq_(msgs, [{'log': test_input}])
def test_logger_json(self, stdout, argv):
test_input_dict = {"hello": "world"}
test_input = json.dumps(test_input_dict)
if six.PY3:
stdin = lambda: six.StringIO(test_input)
else:
stdin = lambda: six.StringIO(test_input.encode('utf-8'))
msgs = []
def mock_publish(context, topic=None, msg=None, modname=None):
msgs.append(msg)
config = {}
with mock.patch("fedmsg.__local", self.local):
with mock.patch("fedmsg.config.__cache", config):
with mock.patch("fedmsg.core.FedMsgContext.publish", mock_publish):
with mock.patch("sys.stdin", new_callable=stdin):
command = LoggerCommand()
command.execute()
eq_(msgs, [test_input_dict])
def test_config_query_broken(self, stderr, stdout, argv):
try:
with mock.patch('fedmsg.config.__cache', {}):
config_command()
except SystemExit as exc:
eq_(exc.code, 1)
else:
output = "output: %r, error: %r" % (
stdout.getvalue(), stderr.getvalue())
assert False, output
output = stdout.getvalue()
error = stderr.getvalue()
eq_(output.strip(), "")
eq_(error.strip(), "Key `endpoints.broken` does not exist in config")
