def test_durable_table(durable_vpc_table):
"""Tests for the Durable PynamoDB model."""
from historical.vpc.models import DurableVPCModel
# we are explicit about our eventTimes because as RANGE_KEY it will need to be unique.
vpc = VPC.copy()
vpc.pop("eventSource")
vpc['eventTime'] = datetime(2017, 5, 11, 23, 30)
DurableVPCModel(**vpc).save()
items = list(DurableVPCModel.query('arn:aws:ec2:us-east-1:123456789012:vpc/vpc-123343'))
assert len(items) == 1
assert not getattr(items[0], 'ttl', None)
vpc['eventTime'] = datetime(2017, 5, 12, 23, 30)
DurableVPCModel(**vpc).save()
items = list(DurableVPCModel.query('arn:aws:ec2:us-east-1:123456789012:vpc/vpc-123343'))
assert len(items) == 2
def test_durable_table(durable_vpc_table):
from historical.vpc.models import DurableVPCModel
# we are explicit about our eventTimes because as RANGE_KEY it will need to be unique.
VPC['eventTime'] = datetime(2017, 5, 11, 23, 30)
DurableVPCModel(**VPC).save()
items = list(
DurableVPCModel.query(
'arn:aws:ec2:us-east-1:123456789012:vpc/vpc-123343'))
assert len(items) == 1
assert not getattr(items[0], 'ttl', None)
VPC['eventTime'] = datetime(2017, 5, 12, 23, 30)
DurableVPCModel(**VPC).save()
items = list(
DurableVPCModel.query(
'arn:aws:ec2:us-east-1:123456789012:vpc/vpc-123343'))
assert len(items) == 2