def test_model_to_json():
    HashOnlyModelFactory.create_table()
    obj = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)

    json_data = obj.to_json()
    assert '"is_enabled": null' in json_data
    assert ('"updated": "%s"' % obj.updated.isoformat()) in json_data
示例#2
0
def test_model_to_json():
    HashOnlyModelFactory.create_table()
    obj = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)

    json_data = obj.to_json()
    assert '"is_enabled": null' in json_data
    assert ('"updated": "%s"' % obj.updated.isoformat()) in json_data
def test_query_primary_index_model_hash_key_should_return():
    HashOnlyModelFactory.create_table()
    HashOnlyModelFactory(agency_subdomain='metzler')
    HashOnlyModelFactory(agency_subdomain='other')

    query = list(HashOnlyModel.query(agency_subdomain='metzler'))
    assert query[0].agency_subdomain == 'metzler'
def test_local_change_then_reload():
    HashOnlyModelFactory.create_table()
    obj = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)
    obj.external_id = 124

    reloaded = obj.reload()
    assert reloaded.external_id == 123
示例#5
0
def test_query_primary_index_model_hash_key_should_return():
    HashOnlyModelFactory.create_table()
    HashOnlyModelFactory(agency_subdomain='metzler')
    HashOnlyModelFactory(agency_subdomain='other')

    query = list(HashOnlyModel.query(agency_subdomain='metzler'))
    assert query[0].agency_subdomain == 'metzler'
def test_query_secondary_index_model_hash_key_should_return():
    HashOnlyModelFactory.create_table()
    HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)
    HashOnlyModelFactory(agency_subdomain='other', external_id=123)

    query = list(HashOnlyModel.query(external_id=123,
                                     query_index='HashOnlyExternalId'))
    assert set([o.agency_subdomain for o in query]) == {'metzler', 'other'}
def test_return_different_fields_except_should_ignore_and_return_true():
    HashOnlyModelFactory.create_table()
    obj1 = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)
    obj2 = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)

    assert obj1.created != obj2.created
    assert obj1.updated != obj2.updated
    assert not return_different_fields_except(obj1.item, obj2.item, ['created', 'updated'])
示例#8
0
def test_query_secondary_index_model_hash_key_should_return():
    HashOnlyModelFactory.create_table()
    HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)
    HashOnlyModelFactory(agency_subdomain='other', external_id=123)

    query = list(
        HashOnlyModel.query(external_id=123, query_index='HashOnlyExternalId'))
    assert set([o.agency_subdomain for o in query]) == {'metzler', 'other'}
def test_return_different_fields_except_should_ignore_and_return_true():
    HashOnlyModelFactory.create_table()
    obj1 = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)
    obj2 = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)

    assert obj1.created != obj2.created
    assert obj1.updated != obj2.updated
    assert not return_different_fields_except(obj1.item, obj2.item,
                                              ['created', 'updated'])
def test_negative_timestamp():
    long_ago = datetime.datetime.utcnow()
    long_ago = long_ago.replace(year=1899)
    HashOnlyModelFactory.create_table()
    HashOnlyModelFactory(agency_subdomain='metzler', external_id=123,
                         created=long_ago)

    obj = HashOnlyModel.all().next()
    assert obj.created.year == long_ago.year
    assert obj.item['created'] < 0
def test_return_different_fields_except_should_return_diff():
    HashOnlyModelFactory.create_table()
    obj1 = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123,
                                created=datetime.datetime.utcnow() - datetime.timedelta(minutes=10))
    obj2 = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)

    assert obj1.created != obj2.created
    assert return_different_fields_except(obj1.item, obj2.item, ['updated']) == dict(
            old=dict(created=obj2.item['created']),
            new=dict(created=obj1.item['created'])
    )
示例#12
0
def test_negative_timestamp():
    long_ago = datetime.datetime.utcnow()
    long_ago = long_ago.replace(year=1899)
    HashOnlyModelFactory.create_table()
    HashOnlyModelFactory(agency_subdomain='metzler',
                         external_id=123,
                         created=long_ago)

    obj = HashOnlyModel.all().next()
    assert obj.created.year == long_ago.year
    assert obj.item['created'] < 0
示例#13
0
def test_all_paginate():
    HashOnlyModelFactory.create_table()
    HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)
    HashOnlyModelFactory(agency_subdomain='metzler2', external_id=124)

    obj, exclusive_start_key = HashOnlyModel.all(limit=1, paginate=True).next()
    obj2, _ = HashOnlyModel.all(
        limit=1, paginate=True,
        exclusive_start_key=exclusive_start_key).next()

    assert obj.external_id != obj2.external_id
    assert {obj.external_id, obj2.external_id} == {123, 124}
def test_return_different_fields_except_should_return_diff():
    HashOnlyModelFactory.create_table()
    obj1 = HashOnlyModelFactory(agency_subdomain='metzler',
                                external_id=123,
                                created=datetime.datetime.utcnow() -
                                datetime.timedelta(minutes=10))
    obj2 = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)

    assert obj1.created != obj2.created
    assert return_different_fields_except(
        obj1.item, obj2.item,
        ['updated']) == dict(old=dict(created=obj2.item['created']),
                             new=dict(created=obj1.item['created']))
示例#15
0
def test_has_changed_primary_key_save_logs(log_data_mock):
    HashOnlyModelFactory.create_table()
    obj = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)
    obj.agency_subdomain = 'other'
    assert obj.has_changed_primary_key()

    obj.save()
    assert log_data_mock.called
    called_with = log_data_mock.call_args_list[0]
    assert called_with[0][0] == 'save overwrite=True table=dev_hash_only'
示例#16
0
def test_non_field_set_on_item_then_get():
    HashOnlyModelFactory.create_table()
    obj = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)
    obj.non_field = 123
    obj.non_field_class_attr = "Changed string"  # This makes more sense when the attr is a property
    obj.save()

    reloaded = HashOnlyModel.get(agency_subdomain='metzler')
    assert not reloaded.item.get('non_field')
    assert not reloaded.item.get('non_field_class_attr')
def test_has_changed_primary_key_save_logs(log_data_mock):
    HashOnlyModelFactory.create_table()
    obj = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)
    obj.agency_subdomain = 'other'
    assert obj.has_changed_primary_key()

    obj.save()
    assert log_data_mock.called
    called_with = log_data_mock.call_args_list[0]
    assert called_with[0][0] == 'save overwrite=True table=dev_hash_only'
def test_non_field_set_on_item_then_get():
    HashOnlyModelFactory.create_table()
    obj = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)
    obj.non_field = 123
    obj.non_field_class_attr = "Changed string"  # This makes more sense when the attr is a property
    obj.save()

    reloaded = HashOnlyModel.get(agency_subdomain='metzler')
    assert not reloaded.item.get('non_field')
    assert not reloaded.item.get('non_field_class_attr')
示例#19
0
def test_local_change_then_reload():
    HashOnlyModelFactory.create_table()
    obj = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)
    obj.external_id = 124

    reloaded = obj.reload()
    assert reloaded.external_id == 123
示例#20
0
def test_update_item_then_get_consistent():
    HashOnlyModelFactory.create_table()
    obj = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)
    assert obj.external_id == 123
    obj.external_id = 124
    obj.save()
    assert obj.external_id == 124

    reloaded = HashOnlyModel.get(agency_subdomain='metzler',
                                 consistent_read=True)
    assert reloaded.external_id == 124
def test_update_item_then_get_consistent():
    HashOnlyModelFactory.create_table()
    obj = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)
    assert obj.external_id == 123
    obj.external_id = 124
    obj.save()
    assert obj.external_id == 124

    reloaded = HashOnlyModel.get(agency_subdomain='metzler', consistent_read=True)
    assert reloaded.external_id == 124
def test_primary_key_should_return():
    HashOnlyModelFactory.create_table()
    obj = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)

    assert obj.get_primary_key() == dict(agency_subdomain='metzler')
示例#23
0
def test_primary_key_should_return():
    HashOnlyModelFactory.create_table()
    obj = HashOnlyModelFactory(agency_subdomain='metzler', external_id=123)

    assert obj.get_primary_key() == dict(agency_subdomain='metzler')