class BigMLAPIMixInTest(unittest.TestCase):
def setUp(self):
self.bigml = BigMLAPIMixIn()
@mock.patch.object(BigML, 'list_projects')
def test_get_bigml_api_check_bigml_auth(self, mock_bigml_lp):
# Uncomplete credentials
with self.assertRaises(NotConfigured):
self.bigml.get_bigml_api(username='******')
with self.assertRaises(NotConfigured):
self.bigml.get_bigml_api(api_key='api_key_only')
# Wrong credentials
mock_bigml_lp.return_value = {'code': 402}
with self.assertRaises(NotConfigured):
self.bigml.get_bigml_api(username='******', api_key='bad_key')
# Correct credentials
mock_bigml_lp.return_value = {'code': 200}
self.bigml.get_bigml_api(username='******', api_key='good_key')
def setUp(self):
self.bigml = BigMLAPIMixIn()
