U kO @sddlZddlZddlmZdejkr.setupcs$tddlm}|jddS)Nrr r)rrrrZdestroy_test_db)clientrZold_namerrteardown!s z/pytest_funcarg__django_client..teardownsession)rZ DATABASE_NAME cached_setuprequestrrrrrpytest_funcarg__django_clients rcs"fdd}dd}||dS)zCreates a test environment using the 'django_client' funcarg above, but also ensures the database is flushed after running each test.cs dS)NZ django_client)Zgetfuncargvaluerrrrr*sz%pytest_funcarg__client..setupcSstddddgt_dS)NflushrF) verbosity interactive)rr Zoutbox)rrrrr,sz(pytest_funcarg__client..teardownfunction)rrrrrpytest_funcarg__client's r#cKs<tjj|||d}|D]\}}t||||q|S)zCreates a user.)usernamepasswordemail)r objectsZ create_user iteritemssetattrsave)namer&extrauserattrvaluerrr user_creator1s  r0cCs|ttdddddS)z*Create a user with no special permissions.r-zuser@example.comcSs|SNdeleter-rrrAz&pytest_funcarg__user..rrrr0rrrrpytest_funcarg__user<sr8c Cs |ttddddddddS)z*Create an admin user with all permissions.Zadminzadmin@example.comT)Z is_superuserZis_staffcSs|Sr1r2r4rrrr5Kr6z'pytest_funcarg__admin..rr7rrrrpytest_funcarg__adminDsr9)ossys functoolsrenvironpathappendZ django.confrZdjango.test.clientrZdjango.test.utilsrrZdjango.core.managementrZ django.corer Zdjango.contrib.auth.modelsr rr#r0r8r9rrrrs