005-user.py 887 B

123456789101112131415161718192021222324252627282930313233343536373839
  1. import sys
  2. import pathlib
  3. import asyncio
  4. import tortoise
  5. test_file = pathlib.Path(__file__)
  6. project = test_file.parent.parent
  7. sys.path.append(str(project))
  8. import server_source as source
  9. from test import test
  10. async def main():
  11. modules = {
  12. source.model.Meta.app: [ "server_source" ]
  13. }
  14. await tortoise.Tortoise.init(
  15. db_url = "sqlite://:memory:",
  16. modules = modules
  17. )
  18. await tortoise.Tortoise.generate_schemas()
  19. first = await source.user_proxy.create("test", "password")
  20. test(await first.compare_password("password"), True)
  21. await first.result().save()
  22. found = await source.user.filter(nick = "test").first()
  23. found_proxy = source.user_proxy(found)
  24. test(found.nick, "test")
  25. test(await found_proxy.compare_password("password"), True)
  26. await tortoise.Tortoise.close_connections()
  27. asyncio.run(main())