test_upload.py

from cbutil.upload import Uploader, DBConfig, load_config_from_env
import os


def setup_env():

    os.environ["INFLUXDB_HOST"] = "INFLUXDB_HOST"
    os.environ["INFLUXDB_PORT"] = "1234"
    os.environ["INFLUXDB_USER_NAME"] = "INFLUXDB_USER_NAME"
    os.environ["INFLUXDB_DATABASE"] = "INFLUXDB_DATABASE"
    os.environ["INFLUXDB_WRITE_USER_PASSWORD"] = "INFLUXDB_WRITE_USER_PASSWORD"


def test_init_with_conf():
    conf = DBConfig("host", 1234, "user_name", "database", "write_user_pw")
    Uploader(conf)


def test_load_conf():
    setup_env()

    conf = load_config_from_env()
    assert conf.host == os.environ["INFLUXDB_HOST"]
    assert conf.port == os.environ["INFLUXDB_PORT"]
    assert conf.user_name == os.environ["INFLUXDB_USER_NAME"]
    assert conf.database == os.environ["INFLUXDB_DATABASE"]
    assert conf.write_user_pw == os.environ["INFLUXDB_WRITE_USER_PASSWORD"]

    Uploader(conf)


def test_init_from_env():
    setup_env()
    up = Uploader()
    assert up.config.host == os.environ["INFLUXDB_HOST"]
    assert up.config.port == os.environ["INFLUXDB_PORT"]
    assert up.config.user_name == os.environ["INFLUXDB_USER_NAME"]
    assert up.config.database == os.environ["INFLUXDB_DATABASE"]
    assert up.config.write_user_pw == os.environ["INFLUXDB_WRITE_USER_PASSWORD"]