tests: cleanup location/tz tests
This commit is contained in:
parent
82e2f96192
commit
94ace823e0
2 changed files with 20 additions and 22 deletions
|
@ -20,6 +20,15 @@ def test() -> None:
|
||||||
|
|
||||||
@pytest.fixture(autouse=True)
|
@pytest.fixture(autouse=True)
|
||||||
def prepare(tmp_path: Path):
|
def prepare(tmp_path: Path):
|
||||||
|
user_config = _prepare_google_config(tmp_path)
|
||||||
|
|
||||||
|
import my.core.cfg as C
|
||||||
|
with C.tmp_config() as config:
|
||||||
|
config.google = user_config # type: ignore
|
||||||
|
yield
|
||||||
|
|
||||||
|
|
||||||
|
def _prepare_google_config(tmp_path: Path):
|
||||||
testdata = Path(__file__).absolute().parent.parent / 'testdata'
|
testdata = Path(__file__).absolute().parent.parent / 'testdata'
|
||||||
assert testdata.exists(), testdata
|
assert testdata.exists(), testdata
|
||||||
|
|
||||||
|
@ -31,7 +40,6 @@ def prepare(tmp_path: Path):
|
||||||
zf.writestr('Takeout/Location History/Location History.json', track.read_bytes())
|
zf.writestr('Takeout/Location History/Location History.json', track.read_bytes())
|
||||||
|
|
||||||
from my.cfg import config
|
from my.cfg import config
|
||||||
class user_config:
|
class google_config:
|
||||||
takeout_path = tmp_path
|
takeout_path = tmp_path
|
||||||
config.google = user_config # type: ignore
|
return google_config
|
||||||
yield
|
|
||||||
|
|
28
tests/tz.py
28
tests/tz.py
|
@ -6,6 +6,7 @@ import pytest # type: ignore
|
||||||
import pytz # type: ignore
|
import pytz # type: ignore
|
||||||
|
|
||||||
from my.core.error import notnone
|
from my.core.error import notnone
|
||||||
|
|
||||||
import my.time.tz.main as TZ
|
import my.time.tz.main as TZ
|
||||||
import my.time.tz.via_location as LTZ
|
import my.time.tz.via_location as LTZ
|
||||||
|
|
||||||
|
@ -78,22 +79,8 @@ def D(dstr: str) -> datetime:
|
||||||
def prepare(tmp_path: Path):
|
def prepare(tmp_path: Path):
|
||||||
LTZ._FASTER = True
|
LTZ._FASTER = True
|
||||||
|
|
||||||
from more_itertools import one
|
from .location import _prepare_google_config
|
||||||
testdata = Path(__file__).absolute().parent.parent / 'testdata'
|
google = _prepare_google_config(tmp_path)
|
||||||
assert testdata.exists(), testdata
|
|
||||||
|
|
||||||
track = one(testdata.rglob('italy-slovenia-2017-07-29.json'))
|
|
||||||
|
|
||||||
# todo ugh. unnecessary zipping, but at the moment takeout provider doesn't support plain dirs
|
|
||||||
import zipfile
|
|
||||||
with zipfile.ZipFile(tmp_path / 'takeout.zip', 'w') as zf:
|
|
||||||
zf.writestr('Takeout/Location History/Location History.json', track.read_bytes())
|
|
||||||
|
|
||||||
# FIXME ugh. early import/inheritance of user_confg in my.google.takeout.paths messes things up..
|
|
||||||
from my.cfg import config
|
|
||||||
class google:
|
|
||||||
takeout_path = tmp_path
|
|
||||||
config.google = google # type: ignore
|
|
||||||
|
|
||||||
class location:
|
class location:
|
||||||
home = (
|
home = (
|
||||||
|
@ -105,11 +92,14 @@ def prepare(tmp_path: Path):
|
||||||
(datetime.fromtimestamp(1600000000, tz=timezone.utc), (55.7558 , 37.6173 )), # Moscow, Russia
|
(datetime.fromtimestamp(1600000000, tz=timezone.utc), (55.7558 , 37.6173 )), # Moscow, Russia
|
||||||
)
|
)
|
||||||
# note: order doesn't matter, will be sorted in the data provider
|
# note: order doesn't matter, will be sorted in the data provider
|
||||||
config.location = location # type: ignore
|
|
||||||
|
|
||||||
class time:
|
class time:
|
||||||
class tz:
|
class tz:
|
||||||
pass # just rely on the default..
|
pass # just rely on the default..
|
||||||
config.time = time # type: ignore
|
|
||||||
|
|
||||||
yield
|
import my.core.cfg as C
|
||||||
|
with C.tmp_config() as config:
|
||||||
|
config.google = google
|
||||||
|
config.time = time
|
||||||
|
config.location = location
|
||||||
|
yield
|
||||||
|
|
Loading…
Add table
Reference in a new issue