def test_number_set_attribute(self):
"""
NumberSetAttribute.default
"""
attr = NumberSetAttribute()
assert attr is not None
attr = NumberSetAttribute(default={1, 2})
assert attr.default == {1, 2}
def test_number_set_attribute(self):
"""
NumberSetAttribute.default
"""
attr = NumberSetAttribute()
self.assertIsNotNone(attr)
attr = NumberSetAttribute(default=set([1, 2]))
self.assertEqual(attr.default, set([1, 2]))
def test_number_set_serialize(self):
"""
NumberSetAttribute.serialize
"""
attr = NumberSetAttribute()
self.assertEqual(attr.serialize(set([1, 2])), [json.dumps(val) for val in sorted(set([1, 2]))])
self.assertEqual(attr.serialize(None), None)
def test_contains_number_set(self):
condition = NumberSetAttribute(attr_name='foo').contains(1)
placeholder_names, expression_attribute_values = {}, {}
expression = condition.serialize(placeholder_names, expression_attribute_values)
assert expression == "contains (#0, :0)"
assert placeholder_names == {'foo': '#0'}
assert expression_attribute_values == {':0': {'N' : '1'}}
def test_number_set_deserialize(self):
"""
NumberSetAttribute.deserialize
"""
attr = NumberSetAttribute()
self.assertEqual(attr.attr_type, NUMBER_SET)
self.assertEqual(attr.deserialize([json.dumps(val) for val in sorted(set([1, 2]))]), set([1, 2]))
def test_delete_action_serialized(self):
action = NumberSetAttribute(attr_name='foo').delete({'NS': ['0']})
placeholder_names, expression_attribute_values = {}, {}
expression = action.serialize(placeholder_names, expression_attribute_values)
assert expression == "#0 :0"
assert placeholder_names == {'foo': '#0'}
assert expression_attribute_values == {':0': {'NS': ['0']}}
def test_number_set_serialize(self):
"""
NumberSetAttribute.serialize
"""
attr = NumberSetAttribute()
assert attr.serialize(set([1, 2])) == [json.dumps(val) for val in sorted(set([1, 2]))]
assert attr.serialize(None) is None
def test_number_set_deserialize(self):
"""
NumberSetAttribute.deserialize
"""
attr = NumberSetAttribute()
assert attr.attr_type == NUMBER_SET
assert attr.deserialize([json.dumps(val) for val in sorted(set([1, 2]))]) == set([1, 2])
class Office(Model):
class Meta:
table_name = "OfficeModel"
host = "http://localhost:{}".format(environ.get("DOCKER_PORT", 8000))
office_id = NumberAttribute(hash_key=True)
address = Location()
employees = ListAttribute(of=OfficeEmployeeMap)
departments = UnicodeSetAttribute()
numbers = NumberSetAttribute()
class InfectionModel(Model):
class Meta:
table_name = 'infections'
if 'aws' in config:
aws_access_key_id = config['aws']['access_key_id']
aws_secret_access_key = config['aws']['secret_access_key']
else:
host = "http://localhost:8000"
regional_cell = NumberAttribute(hash_key=True)
local_cell = NumberAttribute(range_key=True)
timestamps = NumberSetAttribute()
class Reporter(Model):
class Meta:
table_name = 'test_graphene_pynamodb_reporters'
host = DB_HOST
region = DB_REGION
id = NumberAttribute(hash_key=True)
first_name = UnicodeAttribute()
last_name = UnicodeAttribute()
email = UnicodeAttribute(null=True)
pets = NumberSetAttribute(null=True)
articles = OneToMany(Article, null=True)
favorite_article = OneToOne(Article, null=True)
class AttributeTestModel(Model):
class Meta:
host = 'http://localhost:8000'
table_name = 'test'
binary_attr = BinaryAttribute()
binary_set_attr = BinarySetAttribute()
number_attr = NumberAttribute()
number_set_attr = NumberSetAttribute()
unicode_attr = UnicodeAttribute()
unicode_set_attr = UnicodeSetAttribute()
datetime_attr = UTCDateTimeAttribute()
bool_attr = BooleanAttribute()
json_attr = JSONAttribute()
class Office(Model):
class Meta:
table_name = "OfficeModel"
host = "http://localhost:{}".format(environ.get("DOCKER_PORT", 8000))
aws_access_key_id = "my_access_key_id"
aws_secret_access_key = "my_secret_access_key"
office_id = UUIDAttribute(hash_key=True)
address = Location()
employees = ListAttribute(of=OfficeEmployeeMap)
departments = UnicodeSetAttribute()
numbers = NumberSetAttribute()
security_number = UnicodeAttribute(null=True)
office_times = ListAttribute()
cls = DiscriminatorAttribute()
class TestModel(Model):
"""
A model for testing
"""
class Meta:
region = 'us-east-1'
table_name = 'pynamodb-ci'
host = ddb_url
forum = UnicodeAttribute(hash_key=True)
thread = UnicodeAttribute(range_key=True)
view = NumberAttribute(default=0)
view_index = LSIndex()
epoch_index = GSIndex()
epoch = UTCDateTimeAttribute(default=datetime.now)
content = BinaryAttribute(null=True)
scores = NumberSetAttribute()
def test_should_number_set_convert_list():
assert_attribute_conversion(NumberSetAttribute(), graphene.List)
def setUp(self):
self.attribute = UnicodeAttribute(attr_name='foo')
self.set_attribute = NumberSetAttribute(attr_name='foo_set')
self.list_attribute = ListAttribute(attr_name='foo_list')
