Skip to content
Snippets Groups Projects
Commit b778eb6a authored by Christoph Alt's avatar Christoph Alt
Browse files

fixed uploader and added test

parent f86fb6e7
Branches
No related merge requests found
......@@ -39,7 +39,10 @@ class DBConfig:
class Uploader:
def __init__(self, config: DBConfig = None):
self.config = config if config else load_config_from_env()
if config is None:
config = load_config_from_env()
self.config = config
self.client = InfluxDBClient(
host=config.host,
......
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()
Uploader()
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment