def test_SqlStatusFlagScheduler():
class Scheduler(HashAndProcessImplement, SqlStatusFlagScheduler):
uri = "postgres://*****:*****@baasu.db.elephantsql.com:5432/dhdshknw"
duplicate_flag = 50
update_interval = 24 * 3600
with Scheduler(table="sql_status_flag_scheduler_%s" % py_ver) as s:
validate_schduler_implement(s, test_multiprocess=False, processes=2)
def test_MongoDBScheduler():
class Scheduler(HashAndProcessImplement, MongoDBScheduler):
pass
s = Scheduler(collection=mongomock.MongoClient().db.test_MongoDBScheduler)
validate_schduler_implement(s)
for doc in s._col.find():
input_data = int(doc["_id"])
output_data = pickle.loads(doc[s.output_key])
assert input_data * 1000 == output_data
def test_StatusFlagScheduler():
class Scheduler(HashAndProcessImplement, MongoDBStatusFlagScheduler):
duplicate_flag = 50
update_interval = 24 * 3600
s = Scheduler(
collection=mongomock.MongoClient().db.test_StatusFlagScheduler)
validate_schduler_implement(s)
for doc in s._col.find():
input_data = int(doc["_id"])
output_data = pickle.loads(doc["_out"])
assert input_data * 1000 == output_data
assert doc[s.status_key] == 50
assert isinstance(doc[s.edit_at_key], datetime)
def test():
class Scheduler(HashAndProcessImplement, SqlScheduler):
uri = "postgres://*****:*****@baasu.db.elephantsql.com:5432/ujvdiudg"
with Scheduler(table="sql_scheduler_%s" % py_ver) as s:
validate_schduler_implement(s, test_multiprocess=False, processes=2)
def test():
class Scheduler(HashAndProcessImplement, SqliteDictScheduler):
user_db_path = ":memory:"
s = Scheduler()
validate_schduler_implement(s)
def test_wrong_implement():
s = BaseScheduler()
with raises(Exception):
validate_schduler_implement(s)
def test_basic_implement():
s = MyScheduler()
validate_schduler_implement(s)