test.py

from petstore.user.user import User
from petstore.pet.pet import Pet, SoldPets, list_sold_pets
import logging

LOGGER = logging.getLogger(__name__)

def test_create_user():
    LOGGER.info(">>>>>>>> Create User >>>>>>>>")
    user = User(username="test_user", password="password", email="test@test.com",
                firstName="Test", lastName="User", phone="1234567890", userStatus=0)
    response = user.create_user()
    LOGGER.info(f"Create user response: {response.json()}")
    assert response.status_code == 200
    LOGGER.info(">>>>>>>> Get created user info >>>>>>>>")
    response = user.get_user()
    assert response.status_code == 200
    assert response.json()["username"] == "test_user"
    assert response.json()["firstName"] == "Test"
    assert response.json()["lastName"] == "User"

    LOGGER.info(f"User info: {response.json()}")



def test_create_sold_pets():
    LOGGER.info(">>>>>>>> Create sold pets >>>>>>>>")
    pet_a = Pet(id=0, category={"id": 0, "name": "TestJordi"}, name="TestJordi", photoUrls=["string"], tags=[{"id": 0, "name": "TestJordi"}], status="sold")
    pet_b = Pet(id=0, category={"id": 0, "name": "TestJordi"}, name="TestJordi", photoUrls=["string"], tags=[{"id": 0, "name": "TestJordi"}], status="sold")
    response = pet_a.create_pet()
    assert response.status_code == 200
    LOGGER.info(f"Create pet response: {response.json()}")
    response = pet_b.create_pet()
    assert response.status_code == 200
    LOGGER.info(f"Create pet response: {response.json()}")


def test_list_sold_pets():
    LOGGER.info(">>>>>>>> List sold pets >>>>>>>>")
    sold_pets = list_sold_pets()

    for pet in sold_pets.items():
        LOGGER.info(f"Sold pet: {pet}")

    LOGGER.info(">>>>>>>> Count pet repeated names >>>>>>>>")

    sold_pets_object = SoldPets(sold_pets)
    pet_repeated_names = sold_pets_object.count_repeated_pet_names()

    LOGGER.info(f"Count of pet names: {pet_repeated_names}")