def test_query_engine_params(self):
from luiti.daemon.query_engine.params import Params
default_query = {"date_value": "2014-09-01", "language": "Chinese", "gender": "Male"}
selected_query = {"date_value": "2015-07-02", "city": ["Beijing", "London"]}
params_array = Params.build_params_array(default_query, selected_query)
base_expected_opt = {
"date_value": ArrowParameter.get("2015-07-02"),
"city": ["Beijing", "London"],
"language": "Chinese",
"gender": "Male",
}
expected_opts = [
dict(base_expected_opt.items() + {"city": "Beijing"}.items()),
dict(base_expected_opt.items() + {"city": "London"}.items()),
]
self.assertEqual(params_array, expected_opts)
def test_utils(self):
from luiti.daemon.utils import TaskStorageSet, TaskStorageDict
current_time = ArrowParameter.now()
t1 = FoobarDay(date_value=current_time, foobar="1")
t2 = FoobarDay(date_value=current_time, foobar="1")
s1 = TaskStorageSet()
s1.add(t1)
self.assertEqual(len(s1), 1)
s1.add(t2)
self.assertEqual(len(s1), 1)
self.assertTrue(t1 in s1)
self.assertTrue(t2 in s1)
d1 = TaskStorageDict()
d1[t1].add(t1)
d1[t2].add(t2)
self.assertEqual(len(d1), 1)
self.assertEqual(len(d1[t1]), 1)
# -*- coding: utf-8 -*-
import os
import sys
root_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.insert(0, root_dir)
import unittest
from luiti.task_templates import TaskDay
from luiti import luigi
from luiti.luigi_extensions import ArrowParameter
current_time = ArrowParameter.now()
from luiti.tests import SetupLuitiPackages, date_begin
config = SetupLuitiPackages.config
class FoobarDay(TaskDay):
root_dir = "/tmp"
foobar = luigi.Parameter()
class TestDaemon(unittest.TestCase):
def test_Server(self):
from luiti.daemon import Server
s1 = Server("0.0.0.0", 8082)
self.assertEqual(s1.url, "http://localhost:8082")
self.assertTrue(s1.app)
