D6itdZddlmZddlZddlmZddlmZm Z dZ dZ dZ d Z d Zd Zd Zd ZdZdZy)a The functions in this module can be used for testing that the constraints of your models. Each assert function runs SQL UPDATEs that check for the existence of given constraint. Consider the following model:: class User(Base): __tablename__ = 'user' id = sa.Column(sa.Integer, primary_key=True) name = sa.Column(sa.String(200), nullable=True) email = sa.Column(sa.String(255), nullable=False) user = User(name='John Doe', email='john@example.com') session.add(user) session.commit() We can easily test the constraints by assert_* functions:: from sqlalchemy_utils import ( assert_nullable, assert_non_nullable, assert_max_length ) assert_nullable(user, 'name') assert_non_nullable(user, 'email') assert_max_length(user, 'name', 200) # raises AssertionError because the max length of email is 255 assert_max_length(user, 'email', 300) )DecimalN)ARRAY) DataErrorIntegrityErrorcHtjj|}tj|jj |}|j jjdi|j|i}|j||jy)N) saormobject_sessioninspect __class__columnstableupdatevalueskeyexecuteflush)objfieldvaluesessioncolumnquerys [/home/azureuser/techstart-app/venv/lib/python3.12/site-packages/sqlalchemy_utils/asserts.py _update_fieldr+ssff##C(G ZZ & . .u 5F (FLL   ! ( ( ?FJJ+> ?E OOE MMOc t|||y#|$rD}tjj|}|j Jt |d}~wwxYwN)rr r r rollbackstr)rrr reraise_excers r_expect_successful_updater$3sQc5%( &&'',c!fus A?AAc t|||td|z#|$rYnwxYw tjj |}|j y#tjj |}|j wxYw)NzExpected update to raise %s)rAssertionErrorr r r r )rrr expected_excrs r_expect_failing_updater(<sc5%(:\IJJ     &&'',&&'',s! A%A%A1B c6t|trt|jtjrdgSt|jtj rdgSt|jtj r tdgStdy)Nra0zUnknown array item type) isinstancer item_typer IntegerStringNumericr TypeErrortype_s r_repeated_valuer4Hsk% eoorzz 23J  35L  4CL> !56 6rc:t|trtStSr)r,rrrr2s r_expected_exceptionr6Vs%rc(t||dty)z Assert that given column is nullable. This is checked by running an SQL update that assigns given column as None. :param obj: SQLAlchemy declarative model object :param column: Name of the column N)r$rrrs rassert_nullabler9]sc64@rc(t||dty)z Assert that given column is not nullable. This is checked by running an SQL update that assigns given column as None. :param obj: SQLAlchemy declarative model object :param column: Name of the column N)r(rr8s rassert_non_nullabler;hs3n=rctj|jj|j}t ||t ||zt|t||t ||dzzt|y)a Assert that the given column is of given max length. This function supports string typed columns as well as PostgreSQL array typed columns. In the following example we add a check constraint that user can have a maximum of 5 favorite colors and then test this.:: class User(Base): __tablename__ = 'user' id = sa.Column(sa.Integer, primary_key=True) favorite_colors = sa.Column(ARRAY(sa.String), nullable=False) __table_args__ = ( sa.CheckConstraint( sa.func.array_length(favorite_colors, 1) <= 5 ) ) user = User(name='John Doe', favorite_colors=['red', 'blue']) session.add(user) session.commit() assert_max_length(user, 'favorite_colors', 5) :param obj: SQLAlchemy declarative model object :param column: Name of the column :param max_length: Maximum length of given column N) r r r rtyper$r4r6r()rr max_lengthr3s rassert_max_lengthr@ssr@ JJs}} % - -f 5 : :E V_U+j8:Me:T *q.1E" rcRt|||tt|||dz ty)z Assert that the given column must have a minimum value of `min_value`. :param obj: SQLAlchemy declarative model object :param column: Name of the column :param min_value: The minimum allowed value for given column r=Nr$rr(rr min_values rassert_min_valuerE$c69nE3 A ~FrcRt|||tt|||dzty)z Assert that the given column must have a minimum value of `max_value`. :param obj: SQLAlchemy declarative model object :param column: Name of the column :param max_value: The maximum allowed value for given column r=NrBrCs rassert_max_valuerHrFr)__doc__decimalr sqlalchemyr sqlalchemy.dialects.postgresqlrsqlalchemy.excrrrr$r(r4r6r9r;r@rErHrrrrNsO!F04  A>)X G Gr