Как проверить, если еще функция по модульному тесту? - PullRequest
Новая
       2

Как проверить, если еще функция по модульному тесту?

0 голосов
/ 13 февраля 2020

У меня есть файл конфигурации YAML, используемый для проверки соединения с базой данных. Мне нужно написать несколько тестов для моего конфигурационного файла, включающего в себя несколько сценариев ios:

  1. Если поле базы данных пусто, тестовый сценарий не пройден.
  2. Если база данных не Sqlite или Postgre, тестовый пример не пройден.

Я создал тестовый класс, но не знаю, как его реализовать. Кто-нибудь может мне предложить?

data_source.yml: 

database: 

dbopt:
   host: bvcbvcbvcb.com
   port: 5432
   dbname: db1
   user: username
   password: 1234
   client_encoding: utf-8
   connect_timeout: 60
   sslmode: none

query:
   select * from manufacturing_product

data_source.py: 

import yaml

class InvalidConfigError(Exception):
    pass

class DB():
    def __init__(self, dbconf):
        self._dbconf = dict(dbconf)
        # checking for database type
        dbtype = self.get_db_type()
        if dbtype != 'sqlite' and dbtype != 'postgres':
            raise InvalidConfigError(
                'E01001', 'Invalid database type, should be sqlite or postgres.')
        else:
            self.dbtype = dbtype

    def get_db_type(self):
        return self._dbconf['database']

with open('data_source.yml') as f:
    data = yaml.full_load(f)

    for item, doc in data.items():
        print(item, ":", doc)

    database = DB(data)

Мой тестовый класс: 

import yaml
import unittest
import data_source

class TestDBtype(unittest.TestCase):
    def test_missing_db(self):
        # if database is None then failed
        # What should I do?
        pass
    def test_db_type(self):
        # if database is not Sqlite or Postgres then failed
        # What should I do?
        pass
if __name__ == '__main__':
    unittest.main()

1 Ответ

0 голосов
/ 13 февраля 2020

Вы можете использовать как ниже. Идея состоит в том, чтобы инициализировать БД один раз и использовать экземпляр соединения в тестовых случаях. Я не проверял, но это идея. 

import data_source


class TestDBtype(unittest.TestCase):

    #setUpClass gets calls only once where as setUp gets called before every test
    @classmethod
    def setUpClass(cls):
        cls.init_db()

    @classmethod
    def init_db(cls):
        self.db_instance = data_source.DB( config_file )



    def test_missing_db(self):
        self.assertEqual( db_instance, None, "Connection returned None.")
...