def test_nested_select(self): nested = Select('service_set_descr', columns=['name'], cond=Eq('name', 'registration ru')) c, p = nested.glue() self.assertEqual(c, 'SELECT "name" FROM "service_set_descr" WHERE "name" = %s') self.assertEqual(p, ['registration ru']) scoped_cond = Scoped(nested) c, p = scoped_cond.glue() self.assertEqual(c, '(SELECT "name" FROM "service_set_descr" WHERE "name" = %s)') self.assertEqual(p, ['registration ru']) cond_eq = Eq('service_set_descr_id', Scoped(nested)) c, p = cond_eq.glue() self.assertEqual(c, '"service_set_descr_id" = (SELECT "name" FROM "service_set_descr" WHERE "name" = %s)') self.assertEqual(p, ['registration ru']) nested = Select('service_set', columns=['service_type_id'], cond=cond_eq) c, p = nested.glue() self.assertEqual(c, 'SELECT "service_type_id" FROM "service_set" WHERE "service_set_descr_id" = (SELECT "name" FROM "service_set_descr" WHERE "name" = %s)') self.assertEqual(p, ['registration ru']) in_cond = In('id', nested) c, p = in_cond.glue() self.assertEqual(c, '''"id" IN (SELECT "service_type_id" FROM "service_set" WHERE "service_set_descr_id" = (SELECT "name" FROM "service_set_descr" WHERE "name" = %s))''') self.assertEqual(p, ['registration ru']) any_cond = Any('id', nested) c, p = any_cond.glue() self.assertEqual(c, '''%s = ANY (SELECT "service_type_id" FROM "service_set" WHERE "service_set_descr_id" = (SELECT "name" FROM "service_set_descr" WHERE "name" = %s))''') self.assertEqual(p, ['id', 'registration ru'])
def test_scoped_cond(self): cond_lh = BinaryOperator('billing.id', '=', 'a') cond_rh = BinaryOperator('billing.cd', '!=', 'b') cond_and = And(cond_lh, cond_rh) cond_end = Scoped(cond_and) cond, params = cond_end.glue() self.assertEqual( '("billing"."id" = %s AND "billing"."cd" != %s)', cond ) self.assertEqual(['a', 'b'], params) cond, params = Scoped(NullLeaf()).glue() self.assertEqual('', cond) self.assertEqual([], params)