• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• bloop-master
  • Makefile
  • examples
    • replication.py
    • tweet.py
    • documents.py
    • mixins.py
  • bloop
    • actions.py
    • engine.py
    • util.py
    • exceptions.py
    • stream
      • shard.py
      • buffer.py
      • __init__.py
      • coordinator.py
      • stream.py
    • search.py
    • conditions.py
    • models.py
    • ext
      • pendulum.py
      • delorean.py
      • arrow.py
      • __init__.py
    • __init__.py
    • transactions.py
    • signals.py
    • session.py
    • types.py
  • LICENSE
  • .readthedocs.yml
  • README.rst
  • setup.py
  • CHANGELOG.rst
  • .travis.yml
  • tests
    • helpers
      • models.py
      • __init__.py
      • utils.py
    • integ
      • test_basic.py
      • models.py
      • test_queries.py
      • test_inheritance.py
      • __init__.py
      • conftest.py
    • __init__.py
    • unit
      • test_actions.py
      • test_ext
        • test_pendulum.py
        • test_delorean.py
        • test_arrow.py
        • __init__.py
      • test_types.py
      • test_stream
        • test_buffer.py
        • test_stream.py
        • test_coordinator.py
        • __init__.py
        • test_shard.py
        • conftest.py
      • test_conditions.py
      • test_util.py
      • test_models.py
      • test_search.py
      • test_session.py
      • test_engine.py
      • __init__.py
      • test_transactions.py
      • conftest.py
  • scripts
    • graph-dependencies
    • single-test
  • requirements.txt
  • .gitignore
  • docs
    • Makefile
    • make.bat
    • docutils.conf
    • user
      • install.rst
      • types.rst
      • signals.rst
      • engine.rst
      • patterns.rst
      • extensions.rst
      • conditions.rst
      • streams.rst
      • quickstart.rst
      • models.rst
      • transactions.rst
    • api
      • public.rst
      • internal.rst
    • index.rst
    • conf.py
    • meta
      • about.rst
      • changelog.rst
    • _templates
      • layout.html
    • _static
      • favicon-cog.ico
      • bloop.css
  • MANIFEST.in
  • tox.ini

from datetime import datetime

import delorean
import pytest
import pytz

from bloop.ext.delorean import DateTime, Timestamp
from bloop.types import FIXED_ISO8601_FORMAT


now = datetime.now(pytz.utc).replace(microsecond=0)
now_eastern = datetime.now(pytz.timezone("US/Eastern"))
now_iso8601 = now.strftime(FIXED_ISO8601_FORMAT)
now_timestamp = str(int(now.timestamp()))


@pytest.mark.parametrize("timezone", ["utc", "US/Eastern"])
def test_datetime(timezone):
    delorean_now = delorean.Delorean(now)
    typedef = DateTime(timezone)

    assert typedef.dynamo_dump(delorean_now, context={}) == now_iso8601
    assert typedef.dynamo_load(now_iso8601, context={}).shift("utc").datetime == now


@pytest.mark.parametrize("timezone", ["utc", "US/Eastern"])
def test_timestamp(timezone):
    delorean_now = delorean.Delorean(now)
    typedef = Timestamp(timezone)

    assert typedef.dynamo_dump(delorean_now, context={}) == now_timestamp
    assert typedef.dynamo_load(now_timestamp, context={}).shift("utc").datetime == now


@pytest.mark.parametrize("typedef_cls", (DateTime, Timestamp))
def test_none(typedef_cls):
    typedef = typedef_cls()
    assert typedef.dynamo_dump(None, context={}) is None
    assert typedef.dynamo_load(None, context={}) is None