Test your AI Agents with the all-new Agent to Agent Testing Platform.Learn More
How to use _prepare method in Molotov
Best Python code snippet using molotov_python
cassandra_client.py
Source:cassandra_client.py 
...73 self.session = cluster.connect(keyspace)74 self.session.row_factory = datetime_hack_dict_factory75 self.session.default_consistency_level = cassandra.ConsistencyLevel.LOCAL_QUORUM76 self.timer = DummyTimer()77 def _prepare(self, query):78 if query in self.prepared:79 return self.prepared[query]80 prep = self.session.prepare(query)81 self.prepared[query] = prep82 return prep83 def _get_statement(self, table, columns=None, idcolumns=('id',)):84 if columns is None:85 columns = self.default_columns[table]86 where = ' AND '.join(('{} = ?'.format(col) for col in idcolumns))87 stmt = 'SELECT {} FROM {} WHERE {}'.format(','.join(columns), table, where)88 return self._prepare(stmt)89 def _get_compound_pk(self, table, idvs, columns, idcolumns):90 prep = self._get_statement(table, columns, idcolumns)91 res = self.session.execute(prep.bind(idvs))92 try:93 return next(iter(res))94 except StopIteration:95 raise KeyError('{} entry not found'.format(table))96 def _get(self, table, idv, columns=None, idcolumn='id'):97 return self._get_compound_pk(table, [idv], columns, [idcolumn])98 def _get_concurrent(self, statement, ids):99 output = {}100 if ids and not isinstance(ids[0], tuple):101 qids = [(u,) for u in ids]102 else:103 qids = ids104 results = cassandra.concurrent.execute_concurrent_with_args(self.session, statement, qids)105 for id_, (success, result) in zip(ids, results):106 if not success:107 raise result108 if result:109 output[id_] = result[0]110 return output111 @staticmethod112 def val_to_store(client, colname, jsoncols):113 ret = client[colname]114 if colname in jsoncols:115 ret = json.dumps(ret)116 return ret117 def insert_generic(self, data, tablename):118 table_columns = self.default_columns[tablename]119 colnames = ','.join(table_columns)120 placeholders = ('?,'*len(table_columns))[:-1] # '?,?,..,?'121 stmt = 'INSERT INTO {} ({}) VALUES ({})'.format(tablename, colnames, placeholders)122 prep = self._prepare(stmt)123 jsoncols = set(self.json_columns[tablename])124 bindvals = [self.val_to_store(data, colname, jsoncols) for colname in table_columns]125 self.session.execute(prep.bind(bindvals))126 def insert_client(self, client):127 self.insert_generic(client, 'clients')128 def get_client_by_id(self, clientid):129 return self._get('clients', clientid)130 def get_clients_by_id(self, clientids):131 statement = self._get_statement('clients')132 return self._get_concurrent(statement, clientids)133 def get_generic(self, table, selectors, values, maxrows):134 if len(selectors) != len(values):135 raise KeyError('Selectors and values not same length')136 cols = ','.join(self.default_columns[table])137 if not selectors:138 stmt = 'SELECT {} from {} LIMIT {}'.format(cols, table, maxrows)139 else:140 tmpl = 'SELECT {} from {} WHERE {} LIMIT {} ALLOW FILTERING'141 stmt = tmpl.format(cols, table, ' and '.join(selectors), maxrows)142 with self.timer.time('cassandra.get_generic.{}'.format(table)):143 prep = self._prepare(stmt)144 res = self.session.execute(prep.bind(values))145 return res146 def get_clients(self, selectors, values, maxrows):147 return self.get_generic('clients', selectors, values, maxrows)148 def get_clients_by_owner(self, owner):149 prep = self._prepare('SELECT * from clients WHERE owner = ?')150 with self.timer.time('cassandra.get_clients_by_owner'):151 res = self.session.execute(prep.bind([owner]))152 return res153 def get_clients_by_scope(self, scope):154 prep = self._prepare('SELECT * from clients WHERE scopes CONTAINS ?')155 res = self.session.execute(prep.bind([scope]))156 return res157 def get_clients_by_scope_requested(self, scope):158 prep = self._prepare('SELECT * from clients WHERE scopes_requested CONTAINS ?')159 res = self.session.execute(prep.bind([scope]))160 return res161 def add_client_counters(self, clients):162 ids = [c['id'] for c in clients]163 print(ids)164 statement = self._get_statement('clients_counters')165 counters = self._get_concurrent(statement, ids)166 for client in clients:167 client.update(counters.get(client['id'], {'count_users': 0, 'count_tokens': 0}))168 def delete_client(self, clientid):169 prep = self._prepare('DELETE FROM clients WHERE id = ?')170 with self.timer.time('cassandra.delete_client'):171 self.session.execute(prep.bind([clientid]))172 authzq = self._prepare('SELECT userid FROM oauth_authorizations WHERE clientid = ?')173 userids = self.session.execute(authzq.bind([clientid]))174 for row in userids:175 userid = row['userid']176 self.delete_authorization(userid, clientid)177 def insert_orgauthorization(self, clientid, realm, scopes):178 prep = self._prepare('UPDATE clients SET orgauthorization[?] = ? WHERE id = ?')179 self.session.execute(prep.bind([realm, scopes, clientid]))180 def delete_orgauthorization(self, clientid, realm):181 prep = self._prepare('DELETE orgauthorization[?] FROM clients WHERE id = ?')182 self.session.execute(prep.bind([realm, clientid]))183 def get_token(self, tokenid):184 return self._get('oauth_tokens', tokenid, ['*'], 'access_token')185 def get_tokens_by_scope(self, scope):186 prep = self._prepare('SELECT * FROM oauth_tokens WHERE scope contains ?')187 return self.session.execute(prep.bind([scope]))188 def update_token_scopes(self, access_token, scopes):189 prep = self._prepare('UPDATE oauth_tokens SET scope = ? WHERE access_token = ?')190 self.session.execute(prep.bind([scopes, access_token]))191 def get_user_by_id(self, userid):192 return self._get('users', userid, None, 'userid')193 def get_users(self, userids):194 statement = self._get_statement('users', None, ['userid'])195 return self._get_concurrent(statement, userids)196 def get_user_profilephoto(self, userid):197 userinfo = self._get('users', userid,198 ['selectedsource', 'profilephoto', 'updated'], 'userid')199 selectedsource = userinfo['selectedsource']200 profilephoto = userinfo['profilephoto']201 updated = userinfo['updated']202 if profilephoto is None:203 return None, updated204 return profilephoto.get(selectedsource, None), updated205 def insert_user(self, userid, email, name, profilephoto,206 profilephotohash, selectedsource, userid_sec):207 tstamp = now()208 sec_prep = self._prepare('INSERT INTO userid_sec (userid_sec, userid) VALUES (?, ?)')209 for sec in userid_sec:210 self.session.execute(sec_prep.bind([sec, userid]))211 userid_sec_seen = {sec: tstamp for sec in userid_sec}212 prep = self._prepare(213 'INSERT INTO users (userid, created, email, name, profilephoto, profilephotohash, ' +214 'selectedsource, updated, userid_sec, userid_sec_seen) VALUES (?,?,?,?,?,?,?,?,?,?)')215 self.session.execute(prep.bind([216 userid,217 tstamp,218 email,219 name,220 profilephoto,221 profilephotohash,222 selectedsource,223 tstamp,224 userid_sec,225 userid_sec_seen226 ]))227 def reset_user(self, userid):228 prep = self._prepare(229 'INSERT INTO users (userid, usageterms,email, name, profilephoto, profilephotohash, ' +230 'selectedsource, aboveagelimit) VALUES (?,?,?,?,?,?,?,?)')231 self.session.execute(prep.bind([232 userid, False, {}, {}, {}, {}, None, None233 ]))234 def get_userid_by_userid_sec(self, sec):235 return self._get('userid_sec', sec, ['userid'], 'userid_sec')['userid']236 def get_apigk(self, gkid):237 return parse_apigk(self._get('apigk', gkid))238 def get_apigks(self, selectors, values, maxrows):239 return [parse_apigk(gk) for gk in self.get_generic('apigk', selectors, values, maxrows)]240 def delete_apigk(self, gkid):241 prep = self._prepare('DELETE FROM apigk WHERE id = ?')242 self.session.execute(prep.bind([gkid]))243 def insert_apigk(self, apigk):244 self.insert_generic(apigk, 'apigk')245 def _get_logo(self, table, idvalue):246 res = self._get(table, idvalue, ['logo', 'updated'])247 return res['logo'], res['updated']248 def get_client_logo(self, clientid):249 return self._get_logo('clients', clientid)250 def get_apigk_logo(self, gkid):251 return self._get_logo('apigk', gkid)252 def save_logo(self, table, itemid, data, updated):253 prep = self._prepare('INSERT INTO {} (id, logo, updated) VALUES (?, ?, ?)'.format(table))254 self.session.execute(prep.bind([itemid, data, updated]))255 def get_authorizations(self, userid):256 prep = self._prepare('SELECT * FROM oauth_authorizations WHERE userid = ?')257 return self.session.execute(prep.bind([userid]))258 def delete_token(self, token):259 prep = self._prepare('DELETE FROM oauth_tokens WHERE access_token = ?')260 prep.consistency_level = cassandra.ConsistencyLevel.ALL261 self.session.execute(prep.bind([token]))262 def delete_authorization(self, userid, clientid):263 stmt = 'DELETE FROM oauth_authorizations WHERE userid = ? AND clientid = ?'264 prep_del_auth = self._prepare(stmt)265 prep_del_auth.consistency_level = cassandra.ConsistencyLevel.ALL266 self.session.execute(prep_del_auth.bind([userid, clientid]))267 prep_find_tokens = self._prepare(268 'SELECT access_token FROM oauth_tokens WHERE userid = ? AND clientid = ? ' +269 'ALLOW FILTERING')270 for token in self.session.execute(prep_find_tokens.bind([userid, clientid])):271 tokenid = token['access_token']272 self.log.debug('deleting token', token=tokenid)273 self.delete_token(tokenid)274 def delete_all_authorizations(self, clientid):275 prep = self._prepare('SELECT userid FROM oauth_tokens WHERE clientid = ?')276 prep.consistency_level = cassandra.ConsistencyLevel.ALL277 res = self.session.execute(prep.bind([clientid]))278 for userid in {row['userid'] for row in res}:279 self.delete_authorization(userid, clientid)280 def get_oauth_authz_by_scope(self, scope):281 prep = self._prepare('SELECT * FROM oauth_authorizations WHERE scopes CONTAINS ?')282 return self.session.execute(prep.bind([scope]))283 def update_oauth_auth_scopes(self, auth):284 prep = self._prepare(285 'INSERT INTO oauth_authorizations (userid, clientid, scopes) VALUES (?, ?, ?)')286 self.session.execute(prep.bind([auth['userid'], auth['clientid'], auth['scopes']]))287 def get_group(self, groupid):288 return self._get('groups', groupid)289 def delete_group(self, groupid):290 prep = self._prepare('DELETE FROM groups WHERE id = ?')291 self.session.execute(prep.bind([groupid]))292 def insert_group(self, group):293 prep = self._prepare(294 'INSERT INTO groups (id, created, descr, name, owner, updated, public, ' +295 'invitation_token) VALUES (?, ?, ?, ?, ?, ?, ?, ?)')296 self.session.execute(prep.bind([297 group['id'],298 group['created'],299 group['descr'],300 group['name'],301 group['owner'],302 group['updated'],303 group['public'],304 group['invitation_token'],305 ]))306 def get_group_logo(self, groupid):307 return self._get_logo('groups', groupid)308 def get_groups(self, selectors, values, maxrows):309 return self.get_generic('groups', selectors, values, maxrows)310 def get_group_members(self, groupid):311 prep = self._prepare('SELECT * FROM group_members WHERE groupid=?')312 return self.session.execute(prep.bind([groupid]))313 def add_group_member(self, groupid, userid, mtype, status, added_by):314 prep = self._prepare(315 'INSERT INTO group_members (groupid, userid, type, status, added_by) ' +316 'values (?,?,?,?,?)')317 self.session.execute(prep.bind([groupid, userid, mtype, status, added_by]))318 def set_group_member_status(self, groupid, userid, status):319 prep = self._prepare('INSERT INTO group_members (groupid, userid, status) values (?,?,?)')320 self.session.execute(prep.bind([groupid, userid, status]))321 def set_group_member_type(self, groupid, userid, mtype):322 prep = self._prepare('INSERT INTO group_members (groupid, userid, type) values (?,?,?)')323 self.session.execute(prep.bind([groupid, userid, mtype]))324 def del_group_member(self, groupid, userid):325 prep = self._prepare('DELETE FROM group_members WHERE groupid = ? AND userid = ?')326 self.session.execute(prep.bind([groupid, userid]))327 def get_membership_data(self, groupid, userid):328 return self._get_compound_pk('group_members', [groupid, userid],329 ['*'], ['groupid', 'userid'])330 def get_group_memberships(self, userid, mtype, status, maxrows):331 selectors = ['userid = ?']332 values = [userid]333 if mtype is not None:334 selectors.append('type = ?')335 values.append(mtype)336 if status is not None:337 selectors.append('status = ?')338 values.append(status)339 return self.get_generic('group_members', selectors, values, maxrows)340 def insert_grep_code(self, grep):341 prep = self._prepare(342 'INSERT INTO grep_codes (id, code, title, type, last_changed) values (?,?,?,?,?)')343 self.session.execute(prep.bind([grep['id'], grep['code'], grep['title'],344 grep['type'], grep['last_changed']]))345 def get_grep_code(self, grepid):346 return self._get('grep_codes', grepid, ['*'])347 def get_grep_code_by_code(self, code, greptype):348 prep = self._prepare('SELECT * from grep_codes WHERE code = ? and type = ? ALLOW FILTERING')349 data = list(self.session.execute(prep.bind([code, greptype])))350 if not data:351 raise KeyError('No such grep code')352 return data[0]353 def get_org(self, orgid):354 data = self._get('organizations', orgid)355 if 'name' in data and data['name'] is not None:356 data['name'] = translatable(data['name'])357 return data358 def get_orgs(self, orgids):359 statement = self._get_statement('organizations')360 result = self._get_concurrent(statement, orgids)361 for org in result.values():362 if 'name' in org and org['name'] is not None:363 org['name'] = translatable(org['name'])364 return result365 def get_org_by_realm(self, realm):366 data = self._get('organizations', realm, idcolumn='realm')367 if 'name' in data and data['name'] is not None:368 data['name'] = translatable(data['name'])369 return data370 def list_orgs(self):371 tbl = 'organizations'372 stmt = 'SELECT {} from {}'.format(','.join(self.default_columns[tbl]), tbl)373 prep = self._prepare(stmt)374 data = self.session.execute(prep)375 for item in data:376 if 'name' in item and item['name'] is not None:377 item['name'] = translatable(item['name'])378 yield item379 def insert_org(self, org):380 self.insert_generic(org, 'organizations')381 def delete_org(self, orgid):382 prep = self._prepare('DELETE FROM organizations WHERE id = ?')383 self.session.execute(prep.bind([orgid]))384 def get_org_logo(self, orgid):385 res = self._get('organizations', orgid, ['logo', 'logo_updated'])386 return res['logo'], res['logo_updated']387 def save_org_logo(self, table, itemid, data, updated):388 stmt = 'INSERT INTO {} (id, logo, logo_updated) VALUES (?, ?, ?)'.format(table)389 prep = self._prepare(stmt)390 self.session.execute(prep.bind([itemid, data, updated]))391 def org_use_fs_groups(self, realm):392 prep = self._prepare('SELECT id FROM organizations WHERE realm = ?'393 ' and services contains \'fsgroups\' ALLOW FILTERING')394 res = list(self.session.execute(prep.bind([realm])))395 if not res:396 return False397 return True398 def is_org_admin(self, identity, orgid):399 prep = self._prepare('SELECT role from orgroles where identity = ? AND orgid = ?')400 res = list(self.session.execute(prep.bind([identity, orgid])))401 if not res:402 return False403 return 'admin' in res[0]['role']404 def get_mandatory_clients(self, realm):405 prep = self._prepare('SELECT clientid from mandatory_clients where realm = ?')406 res = self.session.execute(prep.bind([realm]))407 return [x['clientid'] for x in res]408 def add_mandatory_client(self, realm, clientid):409 prep = self._prepare('INSERT INTO mandatory_clients (realm, clientid) values (?, ?)')410 self.session.execute(prep.bind([realm, clientid]))411 def del_mandatory_client(self, realm, clientid):412 prep = self._prepare('DELETE FROM mandatory_clients WHERE realm = ? AND clientid = ?')413 return self.session.execute(prep.bind([realm, clientid]))414 def add_services(self, org, services):415 stmt = 'UPDATE organizations set services = services + ? WHERE id = ?'416 prep = self._prepare(stmt)417 self.session.execute(prep.bind([services, org]))418 def del_services(self, org, services):419 stmt = 'UPDATE organizations set services = services - ? WHERE id = ?'420 prep = self._prepare(stmt)421 self.session.execute(prep.bind([services, org]))422 def get_roles(self, selectors, values, maxrows):423 return self.get_generic('orgroles', selectors, values, maxrows)424 def insert_role(self, role):425 self.insert_generic(role, 'orgroles')426 def del_role(self, orgid, identity):427 prep = self._prepare('DELETE FROM orgroles WHERE orgid = ? AND identity = ?')428 self.session.execute(prep.bind([orgid, identity]))429 def apigk_allowed_dn(self, dn):430 prep = self._prepare('SELECT dn from remote_apigatekeepers WHERE dn = ?')431 res = list(self.session.execute(prep.bind([dn])))432 return len(res) > 0433 def get_logins_stats(self, clientid, dates, authsource, maxrows):434 cols = 'date, authsource, timeslot, login_count'435 tmpl = 'SELECT {} FROM logins_stats WHERE clientid = ? AND date in ({})'436 dateslots = (','.join(['?']*len(dates))) # e.g. (?,?,?)437 stmt = tmpl.format(cols, dateslots)438 bindvals = [clientid]439 bindvals += (str(date) for date in dates)440 if maxrows:441 limit = ' LIMIT {}'.format(maxrows)442 else:443 limit = ''444 if authsource:445 stmt += ' AND authsource = ?'446 bindvals.append(authsource)447 limit += ' ALLOW FILTERING'448 stmt += limit449 prep = self._prepare(stmt)450 return self.session.execute(prep.bind(bindvals))451 def get_statistics(self, date, metric):452 statement = 'SELECT * from statistics WHERE date=?'453 if metric:454 statement += " AND metric=?"455 prep = self._prepare(statement)456 params = [date]457 if metric:458 params.append(metric)459 return self.session.execute(prep.bind(params))460 def get_clients_counters(self, maxrows):...
unittest_rqlannotation.py
Source:unittest_rqlannotation.py
...27 self.__class__.repo = repo28 super(SQLGenAnnotatorTC, self).setUp()29 def test_0_1(self):30 with self.admin_access.cnx() as cnx:31 rqlst = self._prepare(cnx, 'Any SEN,RN,OEN WHERE X from_entity SE, '32 'SE eid 44, X relation_type R, R eid 139, '33 'X to_entity OE, OE eid 42, R name RN, SE name SEN, '34 'OE name OEN')35 self.assertEqual(rqlst.defined_vars['SE']._q_invariant, False)36 self.assertEqual(rqlst.defined_vars['OE']._q_invariant, False)37 self.assertEqual(rqlst.defined_vars['R']._q_invariant, False)38 self.assertEqual(rqlst.defined_vars['SE'].stinfo['attrvar'], None)39 self.assertEqual(rqlst.defined_vars['OE'].stinfo['attrvar'], None)40 self.assertEqual(rqlst.defined_vars['R'].stinfo['attrvar'], None)41 def test_0_2(self):42 with self.admin_access.cnx() as cnx:43 rqlst = self._prepare(cnx, 'Any O WHERE NOT S ecrit_par O, S eid 1, '44 'S inline1 P, O inline2 P')45 self.assertEqual(rqlst.defined_vars['P']._q_invariant, True)46 self.assertEqual(rqlst.defined_vars['O'].stinfo['attrvar'], None)47 def test_0_4(self):48 with self.admin_access.cnx() as cnx:49 rqlst = self._prepare(cnx, 'Any A,B,C WHERE A eid 12,A comment B, '50 'A ?wf_info_for C')51 self.assertEqual(rqlst.defined_vars['A']._q_invariant, False)52 self.assertTrue(rqlst.defined_vars['B'].stinfo['attrvar'])53 self.assertEqual(rqlst.defined_vars['C']._q_invariant, False)54 self.assertEqual(rqlst.solutions, [{'A': 'TrInfo', 'B': 'String', 'C': 'Affaire'},55 {'A': 'TrInfo', 'B': 'String', 'C': 'CWUser'},56 {'A': 'TrInfo', 'B': 'String', 'C': 'Note'}])57 def test_0_5(self):58 with self.admin_access.cnx() as cnx:59 rqlst = self._prepare(cnx, 'Any P WHERE N ecrit_par P, N eid 0')60 self.assertEqual(rqlst.defined_vars['N']._q_invariant, False)61 self.assertEqual(rqlst.defined_vars['P']._q_invariant, True)62 def test_0_6(self):63 with self.admin_access.cnx() as cnx:64 rqlst = self._prepare(cnx, 'Any P WHERE NOT N ecrit_par P, N eid 512')65 self.assertEqual(rqlst.defined_vars['P']._q_invariant, False)66 def test_0_7(self):67 with self.admin_access.cnx() as cnx:68 rqlst = self._prepare(cnx, 'Personne X,Y where X nom NX, '69 'Y nom NX, X eid XE, not Y eid XE')70 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)71 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)72 self.assertTrue(rqlst.defined_vars['XE'].stinfo['attrvar'])73 def test_0_8(self):74 with self.admin_access.cnx() as cnx:75 rqlst = self._prepare(cnx, 'Any P WHERE X eid 0, NOT X connait P')76 self.assertEqual(rqlst.defined_vars['P']._q_invariant, False)77 self.assertEqual(len(rqlst.solutions), 1, rqlst.solutions)78 def test_0_10(self):79 with self.admin_access.cnx() as cnx:80 rqlst = self._prepare(cnx, 'Any X WHERE X concerne Y, Y is Note')81 self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)82 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)83 def test_0_11(self):84 with self.admin_access.cnx() as cnx:85 rqlst = self._prepare(cnx, 'Any X WHERE X todo_by Y, X is Affaire')86 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)87 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)88 def test_0_12(self):89 with self.admin_access.cnx() as cnx:90 rqlst = self._prepare(cnx, 'Personne P WHERE P concerne A, '91 'A concerne S, S nom "Logilab"')92 self.assertEqual(rqlst.defined_vars['P']._q_invariant, True)93 self.assertEqual(rqlst.defined_vars['A']._q_invariant, True)94 self.assertEqual(rqlst.defined_vars['S']._q_invariant, False)95 def test_1_0(self):96 with self.admin_access.cnx() as cnx:97 rqlst = self._prepare(cnx, 'Any X,Y WHERE X created_by Y, '98 'X eid 5, NOT Y eid 6')99 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)100 def test_1_1(self):101 with self.admin_access.cnx() as cnx:102 rqlst = self._prepare(cnx, 'Any X,Y WHERE X created_by Y, X eid 5, '103 'NOT Y eid IN (6,7)')104 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)105 def test_2(self):106 with self.admin_access.cnx() as cnx:107 rqlst = self._prepare(cnx, 'Any X WHERE X identity Y, Y eid 1')108 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)109 def test_7(self):110 with self.admin_access.cnx() as cnx:111 rqlst = self._prepare(cnx, 'Personne X,Y where X nom NX, Y nom NX, '112 'X eid XE, not Y eid XE')113 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)114 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)115 def test_8(self):116 with self.admin_access.cnx() as cnx:117 # DISTINCT Any P WHERE P require_group %(g)s,118 # NOT %(u)s has_group_permission P, P is CWPermission119 rqlst = self._prepare(cnx, 'DISTINCT Any X WHERE A concerne X, '120 'NOT N migrated_from X, '121 'X is Note, N eid 1')122 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)123 def test_diff_scope_identity_deamb(self):124 with self.admin_access.cnx() as cnx:125 rqlst = self._prepare(cnx, 'Any X WHERE X concerne Y, Y is Note, '126 'EXISTS(Y identity Z, Z migrated_from N)')127 self.assertEqual(rqlst.defined_vars['Z']._q_invariant, True)128 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)129 def test_optional_inlined(self):130 with self.admin_access.cnx() as cnx:131 rqlst = self._prepare(cnx, 'Any X,S where X from_state S?')132 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)133 self.assertEqual(rqlst.defined_vars['S']._q_invariant, True)134 def test_optional_inlined_2(self):135 with self.admin_access.cnx() as cnx:136 rqlst = self._prepare(cnx, 'Any N,A WHERE N? inline1 A')137 self.assertEqual(rqlst.defined_vars['N']._q_invariant, False)138 self.assertEqual(rqlst.defined_vars['A']._q_invariant, False)139 def test_optional_1(self):140 with self.admin_access.cnx() as cnx:141 rqlst = self._prepare(cnx, 'Any X,S WHERE X travaille S?')142 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)143 self.assertEqual(rqlst.defined_vars['S']._q_invariant, True)144 def test_greater_eid(self):145 with self.admin_access.cnx() as cnx:146 rqlst = self._prepare(cnx, 'Any X WHERE X eid > 5')147 self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)148 def test_greater_eid_typed(self):149 with self.admin_access.cnx() as cnx:150 rqlst = self._prepare(cnx, 'Any X WHERE X eid > 5, X is Note')151 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)152 def test_max_eid(self):153 with self.admin_access.cnx() as cnx:154 rqlst = self._prepare(cnx, 'Any MAX(X)')155 self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)156 def test_max_eid_typed(self):157 with self.admin_access.cnx() as cnx:158 rqlst = self._prepare(cnx, 'Any MAX(X) WHERE X is Note')159 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)160 def test_all_entities(self):161 with self.admin_access.cnx() as cnx:162 rqlst = self._prepare(cnx, 'Any X')163 self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)164 def test_all_typed_entity(self):165 with self.admin_access.cnx() as cnx:166 rqlst = self._prepare(cnx, 'Any X WHERE X is Note')167 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)168 def test_has_text_1(self):169 with self.admin_access.cnx() as cnx:170 rqlst = self._prepare(cnx, 'Any X WHERE X has_text "toto tata"')171 self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)172 self.assertEqual(rqlst.defined_vars['X'].stinfo['principal'].r_type,173 'has_text')174 def test_has_text_2(self):175 with self.admin_access.cnx() as cnx:176 rqlst = self._prepare(cnx, 'Any X WHERE X is Personne, '177 'X has_text "coucou"')178 self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)179 self.assertEqual(rqlst.defined_vars['X'].stinfo['principal'].r_type,180 'has_text')181 def test_not_relation_1(self):182 with self.admin_access.cnx() as cnx:183 # P can't be invariant since deambiguification caused by "NOT X require_permission P"184 # is not considered by generated sql (NOT EXISTS(...))185 rqlst = self._prepare(cnx, 'Any P,G WHERE P require_group G, '186 'NOT X require_permission P')187 self.assertEqual(rqlst.defined_vars['P']._q_invariant, False)188 self.assertEqual(rqlst.defined_vars['G']._q_invariant, True)189 self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)190 def test_not_relation_2(self):191 with self.admin_access.cnx() as cnx:192 rqlst = self._prepare(cnx, 'TrInfo X WHERE X eid 2, '193 'NOT X from_state Y, Y is State')194 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)195 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)196 def test_not_relation_3(self):197 with self.admin_access.cnx() as cnx:198 rqlst = self._prepare(cnx, 'Any X, Y WHERE X eid 1, Y eid in (2, 3)')199 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)200 def test_not_relation_4_1(self):201 with self.admin_access.cnx() as cnx:202 rqlst = self._prepare(cnx, 'Note X WHERE NOT Y evaluee X')203 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)204 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)205 def test_not_relation_4_2(self):206 with self.admin_access.cnx() as cnx:207 rqlst = self._prepare(cnx, 'Any X WHERE NOT Y evaluee X')208 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)209 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)210 def test_not_relation_4_3(self):211 with self.admin_access.cnx() as cnx:212 rqlst = self._prepare(cnx, 'Any Y WHERE NOT Y evaluee X')213 self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)214 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)215 def test_not_relation_4_4(self):216 with self.admin_access.cnx() as cnx:217 rqlst = self._prepare(cnx, 'Any X WHERE NOT Y evaluee X, Y is CWUser')218 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)219 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)220 def test_not_relation_4_5(self):221 with self.admin_access.cnx() as cnx:222 rqlst = self._prepare(cnx, 'Any X WHERE NOT Y evaluee X, '223 'Y eid %s, X is Note' % self.ueid)224 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)225 self.assertEqual(rqlst.solutions, [{'X': 'Note'}])226 def test_not_relation_5_1(self):227 with self.admin_access.cnx() as cnx:228 rqlst = self._prepare(cnx, 'Any X,Y WHERE X name "CWGroup", '229 'Y eid IN(1, 2, 3), NOT X read_permission Y')230 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)231 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)232 def test_not_relation_5_2(self):233 with self.admin_access.cnx() as cnx:234 rqlst = self._prepare(cnx, 'DISTINCT Any X,Y WHERE X name "CWGroup", '235 'Y eid IN(1, 2, 3), NOT X read_permission Y')236 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)237 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)238 def test_not_relation_6(self):239 with self.admin_access.cnx() as cnx:240 rqlst = self._prepare(cnx, 'Personne P where NOT P concerne A')241 self.assertEqual(rqlst.defined_vars['P']._q_invariant, False)242 self.assertEqual(rqlst.defined_vars['A']._q_invariant, True)243 def test_not_relation_7(self):244 with self.admin_access.cnx() as cnx:245 rqlst = self._prepare(cnx, 'Any K,V WHERE P is CWProperty, '246 'P pkey K, P value V, NOT P for_user U')247 self.assertEqual(rqlst.defined_vars['P']._q_invariant, False)248 self.assertEqual(rqlst.defined_vars['U']._q_invariant, True)249 def test_exists_1(self):250 with self.admin_access.cnx() as cnx:251 rqlst = self._prepare(cnx, 'Any U WHERE U eid IN (1,2), EXISTS(X owned_by U)')252 self.assertEqual(rqlst.defined_vars['U']._q_invariant, False)253 self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)254 def test_exists_2(self):255 with self.admin_access.cnx() as cnx:256 rqlst = self._prepare(cnx, 'Any U WHERE EXISTS(U eid IN (1,2), X owned_by U)')257 self.assertEqual(rqlst.defined_vars['U']._q_invariant, False)258 self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)259 def test_exists_3(self):260 with self.admin_access.cnx() as cnx:261 rqlst = self._prepare(cnx, 'Any U WHERE EXISTS(X owned_by U, X bookmarked_by U)')262 self.assertEqual(rqlst.defined_vars['U']._q_invariant, False)263 self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)264 def test_exists_4(self):265 with self.admin_access.cnx() as cnx:266 rqlst = self._prepare(cnx, 'Any X,Y WHERE X name "CWGroup", '267 'Y eid IN(1, 2, 3), EXISTS(X read_permission Y)')268 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)269 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)270 def test_exists_5(self):271 with self.admin_access.cnx() as cnx:272 rqlst = self._prepare(cnx, 'DISTINCT Any X,Y WHERE X name "CWGroup", '273 'Y eid IN(1, 2, 3), EXISTS(X read_permission Y)')274 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)275 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)276 def test_not_exists_1(self):277 with self.admin_access.cnx() as cnx:278 rqlst = self._prepare(cnx, 'Any U WHERE NOT EXISTS(X owned_by U, '279 'X bookmarked_by U)')280 self.assertEqual(rqlst.defined_vars['U']._q_invariant, False)281 self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)282 def test_not_exists_2(self):283 with self.admin_access.cnx() as cnx:284 rqlst = self._prepare(cnx, 'Any X,Y WHERE X name "CWGroup", '285 'Y eid IN(1, 2, 3), NOT EXISTS(X read_permission Y)')286 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)287 def test_not_exists_distinct_1(self):288 with self.admin_access.cnx() as cnx:289 rqlst = self._prepare(cnx, 'DISTINCT Any X,Y WHERE X name "CWGroup", '290 'Y eid IN(1, 2, 3), NOT EXISTS(X read_permission Y)')291 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)292 def test_or_1(self):293 with self.admin_access.cnx() as cnx:294 rqlst = self._prepare(cnx, 'Any X WHERE X concerne B OR '295 'C concerne X, B eid 12, C eid 13')296 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)297 def test_or_2(self):298 with self.admin_access.cnx() as cnx:299 rqlst = self._prepare(cnx, 'Any X WHERE X created_by U, X concerne B OR '300 'C concerne X, B eid 12, C eid 13')301 self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)302 self.assertEqual(rqlst.defined_vars['U']._q_invariant, True)303 self.assertEqual(rqlst.defined_vars['X'].stinfo['principal'].r_type, 'created_by')304 def test_or_3(self):305 with self.admin_access.cnx() as cnx:306 rqlst = self._prepare(cnx, 'Any N WHERE A evaluee N or EXISTS(N todo_by U)')307 self.assertEqual(rqlst.defined_vars['N']._q_invariant, False)308 self.assertEqual(rqlst.defined_vars['A']._q_invariant, True)309 self.assertEqual(rqlst.defined_vars['U']._q_invariant, True)310 def test_or_exists_1(self):311 with self.admin_access.cnx() as cnx:312 # query generated by security rewriting313 rqlst = self._prepare(cnx, 'DISTINCT Any A,S WHERE A is Affaire, S nom "chouette", '314 'S is IN(Division, Societe, SubDivision),'315 '(EXISTS(A owned_by D)) '316 'OR ((((EXISTS(E concerne C?, C owned_by D, A identity E, '317 ' C is Note, E is Affaire)) '318 'OR (EXISTS(I concerne H?, H owned_by D, H is Societe, '319 ' A identity I, I is Affaire))) '320 'OR (EXISTS(J concerne G?, G owned_by D, G is SubDivision, '321 ' A identity J, J is Affaire))) '322 'OR (EXISTS(K concerne F?, F owned_by D, F is Division, '323 ' A identity K, K is Affaire)))')324 self.assertEqual(rqlst.defined_vars['A']._q_invariant, False)325 self.assertEqual(rqlst.defined_vars['S']._q_invariant, False)326 def test_or_exists_2(self):327 with self.admin_access.cnx() as cnx:328 rqlst = self._prepare(cnx, 'Any U WHERE EXISTS(U in_group G, G name "managers") OR '329 'EXISTS(X owned_by U, X bookmarked_by U)')330 self.assertEqual(rqlst.defined_vars['U']._q_invariant, False)331 self.assertEqual(rqlst.defined_vars['G']._q_invariant, False)332 self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)333 def test_or_exists_3(self):334 with self.admin_access.cnx() as cnx:335 rqlst = self._prepare(cnx, 'Any COUNT(S),CS GROUPBY CS ORDERBY 1 DESC LIMIT 10 '336 'WHERE C is Societe, S concerne C, C nom CS, '337 '(EXISTS(S owned_by D)) '338 'OR (EXISTS(S documented_by N, N title "published"))')339 self.assertEqual(rqlst.defined_vars['S']._q_invariant, True)340 rqlst = self._prepare(cnx, 'Any COUNT(S),CS GROUPBY CS ORDERBY 1 DESC LIMIT 10 '341 'WHERE S is Affaire, C is Societe, S concerne C, C nom CS, '342 '(EXISTS(S owned_by D)) '343 'OR (EXISTS(S documented_by N, N title "published"))')344 self.assertEqual(rqlst.defined_vars['S']._q_invariant, True)345 def test_nonregr_ambiguity(self):346 with self.admin_access.cnx() as cnx:347 rqlst = self._prepare(cnx, 'Note N WHERE N attachment F')348 # N may be an image as well, not invariant349 self.assertEqual(rqlst.defined_vars['N']._q_invariant, False)350 self.assertEqual(rqlst.defined_vars['F']._q_invariant, True)351 def test_nonregr_ambiguity_2(self):352 with self.admin_access.cnx() as cnx:353 rqlst = self._prepare(cnx, 'Any S,SN WHERE X has_text "tot", '354 'X in_state S, S name SN, X is CWUser')355 # X use has_text but should not be invariant as ambiguous, and has_text356 # may not be its principal357 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)358 self.assertEqual(rqlst.defined_vars['S']._q_invariant, False)359 def test_remove_from_deleted_source_1(self):360 with self.admin_access.cnx() as cnx:361 rqlst = self._prepare(cnx, 'Note X WHERE X eid 999998, NOT X cw_source Y')362 self.assertNotIn('X', rqlst.defined_vars) # simplified363 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)364 def test_remove_from_deleted_source_2(self):365 with self.admin_access.cnx() as cnx:366 rqlst = self._prepare(cnx, 'Note X WHERE X eid IN (999998, 999999), NOT X cw_source Y')367 self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)368 self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)369 def test_has_text_security_cache_bug(self):370 with self.admin_access.cnx() as cnx:371 rqlst = self._prepare(cnx, 'Any X WHERE X has_text "toto" WITH X BEING '372 '(Any C WHERE C is Societe, C nom CS)')373 self.assertTrue(rqlst.parent.has_text_query)374if __name__ == '__main__':375 import unittest...
client.py
Source:client.py
...21 }22 23 # Loads the specified API resource24 if (api_version == 'v2'):25 self.items = self._prepare(v2.Items)26 self.recipes = self._prepare(v2.Recipes)27 self.skins = self._prepare(v2.Skins)28 self.materials = self._prepare(v2.Materials)29 self.search = self._prepare(v2.Search)30 self.maps = self._prepare(v2.Maps)31 self.continents = self._prepare(v2.Continents)32 self.continents.floors = self._prepare(v2.Continents.Floors)33 self.continents.floors = self._prepare(v2.Continents.Floors)34 self.continents.floors.regions = self._prepare(v2.Continents.Floors.Regions)35 self.continents.floors.regions.maps = self._prepare(v2.Continents.Floors.Regions.Maps)36 self.continents.floors.regions.maps.pois = self._prepare(v2.Continents.Floors.Regions.Maps.Subresource, "pois")37 self.continents.floors.regions.maps.sectors = self._prepare(v2.Continents.Floors.Regions.Maps.Subresource, "sectors")38 self.continents.floors.regions.maps.tasks = self._prepare(v2.Continents.Floors.Regions.Maps.Subresource, "tasks")39 self.build = self._prepare(v2.Build)40 self.colors = self._prepare(v2.Colors)41 self.files = self._prepare(v2.Files)42 self.specializations = self._prepare(v2.Specializations)43 elif (api_version =='v1'):44 self.items = self._prepare(v1.Items)45 self.recipes = self._prepare(v1.Recipes)46 self.recipe_details = self._prepare(v1.RecipeDetails)47 self.item_details = self._prepare(v1.ItemDetails)48 self.skins = self._prepare(v1.Skins)49 self.skin_details = self._prepare(v1.Skin_Details)50 self.matches = self._prepare(v1.Matches)51 self.match_details = self._prepare(v1.MatchDetails)52 self.objective_names = self._prepare(v1.ObjectiveNames)53 self.event_names = self._prepare(v1.EventNames)54 self.map_names = self._prepare(v1.MapNames)55 self.world_names = self._prepare(v1.WorldNames)56 self.guild_details = self._prepare(v1.GuildDetails)57 self.continents = self._prepare(v1.Continents)58 self.maps = self._prepare(v1.Maps)59 self.map_floor = self._prepare(v1.Map_Floor)60 self.build = self._prepare(v1.Build)61 self.colors = self._prepare(v1.Colors)62 self.files = self._prepare(v1.Files)63 self.event_details = self._prepare(v1.EventDetails)64 else:65 raise ValueError("Only v1 or v2 accepted")66 67 def _prepare(self, resource, subresource_name=None):68 """69 Intializes the resource using the settings supplied70 """71 if subresource_name is not None:72 return resource(self.options, self.session, subresource_name)...
Command.js
Source:Command.js
...6test('arguments:required', t => {7 const command = new Command()8 command.arguments = [new InputArgument('arg', InputArgument.VALUE_REQUIRED)]9 try {10 command._prepare(new Input([]))11 t.fail('Should have thrown an error')12 } catch (err) {13 t.pass()14 }15 command._prepare(new Input(['command', 'value']))16 t.true(command.containsArgument('arg'))17 t.is(command.argument('arg'), 'value')18})19test('arguments:optional', t => {20 const command = new Command()21 command.arguments = [new InputArgument('arg', InputArgument.VALUE_OPTIONAL)]22 command._prepare(new Input(['command']))23 t.false(command.containsArgument('arg'))24 command._prepare(new Input(['command', 'value']))25 t.true(command.containsArgument('arg'))26 t.is(command.argument('arg'), 'value')27})28test('arguments:default', t => {29 const command = new Command()30 command.arguments = [new InputArgument('arg', InputArgument.VALUE_OPTIONAL, null, 'def')]31 command._prepare(new Input(['command']))32 t.is(command.argument('arg'), 'def')33 command._prepare(new Input(['command', 'value']))34 t.is(command.argument('arg'), 'value')35})36test('arguments:fallback', t => {37 const command = new Command()38 command.arguments = [new InputArgument('arg', InputArgument.VALUE_OPTIONAL)]39 command._prepare(new Input(['command']))40 t.is(command.argument('arg'), null)41 t.is(command.argument('arg', 'fall'), 'fall')42 command._prepare(new Input(['command', 'value']))43 t.is(command.argument('arg'), 'value')44 t.is(command.argument('arg', 'fall'), 'value')45})46test('options:optional', t => {47 const command = new Command()48 command.options = [new InputOption('opt', InputOption.VALUE_OPTIONAL)]49 command._prepare(new Input(['command']))50 t.false(command.containsOption('opt'))51 command._prepare(new Input(['command', '--opt=ion']))52 t.true(command.containsOption('opt'))53})54test('options:default', t => {55 const command = new Command()56 command.options = [new InputOption('opt', InputOption.VALUE_OPTIONAL, null, 5)]57 command._prepare(new Input(['command']))58 t.is(command.option('opt'), 5)59 command._prepare(new Input(['command', '--opt=10']))60 t.is(command.option('opt'), '10')61})62test('options:fallback', t => {63 const command = new Command()64 command.options = [new InputOption('opt', InputOption.VALUE_OPTIONAL)]65 command._prepare(new Input(['command']))66 t.is(command.option('opt'), null)67 t.is(command.option('opt', 5), 5)68 command._prepare(new Input(['command', '--opt=10']))69 t.is(command.option('opt'), '10')70 t.is(command.option('opt', 5), '10')71})72test('options:flag', t => {73 const command = new Command()74 command.options = [new InputOption('opt', InputOption.VALUE_NONE)]75 command._prepare(new Input(['command']))76 t.is(command.containsOption('opt'), false)77 t.is(command.option('opt'), null)78 command._prepare(new Input(['command', '--opt']))79 t.is(command.containsOption('opt'), true)80 t.is(command.option('opt'), true)...
Automation Testing Tutorials
Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.
LambdaTest Learning Hubs:
- JUnit Tutorial
- TestNG Tutorial
- Webdriver Tutorial
- WebDriverIO Tutorial
- Protractor Tutorial
- Selenium 4 Tutorial
- Jenkins Tutorial
- NUnit Tutorial
- Jest Tutorial
- Playwright Tutorial
- Cypress Tutorial
- PyTest Tutorial
YouTube
You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.
Run Molotov automation tests on LambdaTest cloud grid
Perform automation testing on 3000+ real desktop and mobile devices online.
Try LambdaTest Now !!
Get 100 minutes of automation test minutes FREE!!
Was this article helpful?
