api_testing.py

from API import Get, Post
import requests


class Login(Post):
    def __init__(self, phone_number: str):
        super().__init__()
        self.url = "https://somedomain.com/login"
        self.headers = {
            "Content-Type": "application/json"
        }
        self.json = {
            "phone_number": phone_number
        }
        self.execute()


class VerifyOtp(Post):
    def __init__(self, otp: str):
        super().__init__()
        self.url = "https://somedomain.com/verify_otp"
        self.headers = {
            "Content-Type": "application/json"
        }
        self.json = {
            "otp": otp
        }
        self.execute()


if __name__ == "__main__":
    login = Login("8123456789")
    assert login.response.ok

    verify_otp = VerifyOtp(login.json_object.otp)
    assert verify_otp.response.ok
    assert verify_otp.json_object.token is not None