from os import getenv
from unittest import TestCase

import openai
from osbot_utils.utils.Dev import pprint

from test_bot.api.Open_API import Open_API, OPEN_API_KEY


class test_Open_API(TestCase):

    def setUp(self) -> None:
        self.open_api = Open_API().setup()

    def test___init__(self):
        assert type(self.open_api) == Open_API

    def test_api_key(self):
        api_key = self.open_api.api_key()
        assert api_key is not None
        assert api_key == getenv(OPEN_API_KEY)

    def test_create(self):
        response = self.open_api.create()
        pprint(list(response))

    def test_setup(self):
        assert openai.api_key == self.open_api.api_key()