def test_get_provider_failure(self): """ Test if get_provider method when passed an invalid key raises an exception. """ nb = NuleculeBase(params = [], basepath = '', namespace = '') # purposefully give the wrong provider key provider_key = u'mesos' nb.config = {u'general': {u'provider': provider_key}} with self.assertRaises(NuleculeException): nb.get_provider()
def test_get_provider_success(self): """ Test if get_provider method when passed a particular valid key returns the corresponding class. """ nb = NuleculeBase(params = [], basepath = '', namespace = '') provider_key = u'openshift' # method `get_provider` will read from this config, we give it here # since we have neither provided it before nor it is auto-generated nb.config = {u'general': {u'provider': provider_key}} return_provider = mock.Mock() # mocking return value of method plugin.getProvider,because it returns # provider class and that class gets called with values nb.plugin.getProvider = mock.Mock(return_value=return_provider) ret_provider_key, ret_provider = nb.get_provider() self.assertEqual(provider_key, ret_provider_key) return_provider.assert_called_with({u'provider': provider_key}, '', False)