def test_get_by_user(self): self.mox.StubOutWithMock(db, 'key_pair_get_all_by_user') self.mox.StubOutWithMock(db, 'key_pair_count_by_user') db.key_pair_get_all_by_user(self.context, 'fake-user').AndReturn( [fake_keypair]) db.key_pair_count_by_user(self.context, 'fake-user').AndReturn(1) self.mox.ReplayAll() keypairs = keypair.KeyPairList.get_by_user(self.context, 'fake-user') self.assertEqual(1, len(keypairs)) self.compare_obj(keypairs[0], fake_keypair) self.assertEqual(1, keypair.KeyPairList.get_count_by_user(self.context, 'fake-user'))
def test_get_by_user(self): self.mox.StubOutWithMock(db, 'key_pair_get_all_by_user') self.mox.StubOutWithMock(db, 'key_pair_count_by_user') db.key_pair_get_all_by_user(self.context, 'fake-user').AndReturn( [fake_keypair]) db.key_pair_count_by_user(self.context, 'fake-user').AndReturn(1) self.mox.ReplayAll() keypairs = keypair.KeyPairList.get_by_user(self.context, 'fake-user') self.assertEqual(1, len(keypairs)) self._compare(keypairs[0], fake_keypair) self.assertEqual(1, keypair.KeyPairList.get_count_by_user(self.context, 'fake-user'))
def allowed_key_pairs(context, requested_key_pairs): """Check quota and return min(requested, allowed) key pairs.""" user_id = context.user_id project_id = context.project_id context = context.elevated() used_key_pairs = db.key_pair_count_by_user(context, user_id) quota = get_project_quotas(context, project_id) allowed_key_pairs = _get_request_allotment(requested_key_pairs, used_key_pairs, quota['key_pairs']) return min(requested_key_pairs, allowed_key_pairs)
def allowed_key_pairs(context, requested_key_pairs): """Check quota and return min(requested, allowed) key pairs.""" user_id = context.user_id project_id = context.project_id context = context.elevated() used_key_pairs = db.key_pair_count_by_user(context, user_id) quota = get_project_quotas(context, project_id) allowed_key_pairs = _get_request_allotment(requested_key_pairs, used_key_pairs, quota['key_pairs']) return min(requested_key_pairs, allowed_key_pairs)
def get_count_by_user(cls, context, user_id): return db.key_pair_count_by_user(context, user_id)
def get_count_by_user(cls, context, user_id): return (cls._get_count_from_db(context, user_id) + db.key_pair_count_by_user(context, user_id))
def get_count_by_user(cls, context, user_id): return db.key_pair_count_by_user(context, user_id)
def get_count_by_user(cls, context, user_id): return (cls._get_count_from_db(context, user_id) + db.key_pair_count_by_user(context, user_id))