def test_mpdclient_connect_no_password(self): mpdconf={"host":"host","port":6600} mpdclient = MpdClient(mpdconf) #Mock the mpd client mpdmock=Mock() mpdclient.client=mpdmock # Call mpdclient.connect() # Test mpdmock.connect.assert_called_with(mpdconf["host"], mpdconf["port"])
def test_mpdclient_connect_bad_host_raises_mpdclientexception(self): mpdconf={"host":"badhost","port":6600} mpdclient = MpdClient(mpdconf) #Mock the mpd client mpdmock=Mock() mpdclient.client=mpdmock #Mock the connect method connect=Mock(side_effect=ConnectionError()) mpdmock.connect=connect # Call with self.assertRaises(MpdClientException): mpdclient.connect()
def test_mpdclient_connectexception_raises_mpdclientexception(self): mpdconf={"host":"badhost","port":6600} mpdclient = MpdClient(mpdconf) #Mock the mpd client mpdmock=Mock() mpdclient.client=mpdmock #Mock the connect method and reaise MPDError connect=Mock(side_effect=MPDError()) mpdmock.connect=connect # Call with self.assertRaises(MpdClientException): mpdclient.connect() # Methods call assertions mpdmock.connect.assert_called_with(mpdconf["host"], mpdconf["port"])
def test_mpdclient_connect_bad_password_raises_mpdclientexception(self): mpdconf={"host":"badhost","port":6600, "password":"******"} mpdclient = MpdClient(mpdconf) #Mock the mpd client mpdmock=Mock() mpdclient.client=mpdmock #Mock the password method password=Mock(side_effect=CommandError()) mpdmock.password=password # Call with self.assertRaises(MpdClientException): mpdclient.connect() # Methods call assertions mpdmock.connect.assert_called_with(mpdconf["host"], mpdconf["port"]) mpdmock.password.assert_called_with(mpdconf["password"])
class MpdRmqAgent(RabbitMqAgent): def __init__(self, config={}, mpdconf={}, rmqconf={}): RabbitMqAgent.__init__(self, config, rmqconf) self.mpdclient = MpdClient(mpdconf) def before_execute(self): LOGGER.debug("before_execute") self.mpdclient.connect() RabbitMqAgent.before_execute(self) def after_execute(self): LOGGER.debug("after_execute") RabbitMqAgent.after_execute(self) def ensure_after_execute(self): LOGGER.debug("ensure_after_execute") try: RabbitMqAgent.ensure_after_execute(self) except Exception, ex: LOGGER.error("Error on error calling parent ensure_after_execute") raise MpdRmqException, MpdRmqException(ex), sys.exc_info()[2] # keep stacktrace finally: