def test_credential_get(self): """Tests get of credentials by tenant ID and credential ID.""" cred11 = self._cred_test_obj(1, 1) cred11_db = self._add_credential(cred11) cred21 = self._cred_test_obj(2, 1) cred21_db = self._add_credential(cred21) cred22 = self._cred_test_obj(2, 2) cred22_db = self._add_credential(cred22) cred = self._get_credential(cred11_db) self._assert_cred_equal(cred, cred11) cred = self._get_credential(cred21_db) self._assert_cred_equal(cred, cred21) cred = self._get_credential(cred22_db) self._assert_cred_equal(cred, cred22) with testtools.ExpectedException(exceptions.CredentialNotFound): network_db_v2.get_credential("dummyTenantId", "dummyCredentialId") cred_all_t1 = network_db_v2.get_all_credentials(cred11.tenant_id) self.assertEqual(len(cred_all_t1), 1) cred_all_t2 = network_db_v2.get_all_credentials(cred21.tenant_id) self.assertEqual(len(cred_all_t2), 2)
def _get_credential(self, db_cred): """Lists credentials that match a credential's tenant and cred IDs.""" return network_db_v2.get_credential(db_cred.tenant_id, db_cred.credential_id)