• Search by Module
• Search by Words
• Search Projects

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

• PynamoDB-master
  • .coveragerc
  • examples
    • attributes.py
    • office_model.py
    • model.py
    • indexes.py
    • connection.py
    • table_connection.py
    • optimistic_locking.py
    • url_shortener
      • shortener.py
      • templates
        • index.html
      • README.rst
  • LICENSE
  • pynamodb
    • util.py
    • exceptions.py
    • py.typed
    • attributes.py
    • models.py
    • connection
      • table.py
      • __init__.py
      • base.py
    • expressions
      • condition.py
      • util.py
      • operand.py
      • update.py
      • __init__.py
      • projection.py
    • constants.py
    • indexes.py
    • _compat.py
    • __init__.py
    • transactions.py
    • settings.py
    • signals.py
    • types.py
    • pagination.py
  • README.rst
  • mypy.ini
  • setup.py
  • requirements-dev.txt
  • setup.cfg
  • .travis.yml
  • tests
    • data.py
    • test_base_connection.py
    • test_exceptions.py
    • test_attributes.py
    • mypy_helpers.py
    • integration
      • binary_update_test.py
      • test_transaction_integration.py
      • model_integration_test.py
      • __init__.py
      • table_integration_test.py
      • base_integration_test.py
      • conftest.py
    • deep_eq.py
    • test_expressions.py
    • test_mypy.py
    • test_signals.py
    • test_transaction.py
    • response.py
    • test_table_connection.py
    • test_model.py
    • test_settings.py
    • __init__.py
    • test_pagination.py
    • conftest.py
  • pytest.ini
  • .gitignore
  • docs
    • release_notes.rst
    • optimistic_locking.rst
    • logging.rst
    • Makefile
    • transaction.rst
    • signals.rst
    • backup_restore.rst
    • low_level.rst
    • batch.rst
    • versioning.rst
    • api.rst
    • tutorial.rst
    • updates.rst
    • examples.rst
    • awsaccess.rst
    • rate_limited_operations.rst
    • indexes.rst
    • attributes.rst
    • conditional.rst
    • quickstart.rst
    • settings.rst
    • requirements.txt
    • index.rst
    • conf.py
    • local.rst
    • contributing.rst
  • MANIFEST.in
  • tox.ini

import pytest

from pynamodb.attributes import UnicodeAttribute, BinaryAttribute, BinarySetAttribute
from pynamodb.models import Model


@pytest.mark.ddblocal
def test_binary_attribute_update(ddb_url):
    class DataModel(Model):
        class Meta:
            table_name = 'binary_attr_update'
            host = ddb_url
        pkey = UnicodeAttribute(hash_key=True)
        data = BinaryAttribute()

    DataModel.create_table(read_capacity_units=1, write_capacity_units=1, wait=True)
    data = b'\x00hey\xfb'
    pkey = 'pkey'
    DataModel(pkey, data=data).save()
    m = DataModel.get(pkey)
    assert m.data == data

    new_data = b'\xff'
    m.update(actions=[DataModel.data.set(new_data)])
    assert new_data == m.data

@pytest.mark.ddblocal
def test_binary_set_attribute_update(ddb_url):
    class DataModel(Model):
        class Meta:
            table_name = 'binary_set_attr_update'
            host = ddb_url
        pkey = UnicodeAttribute(hash_key=True)
        data = BinarySetAttribute()

    DataModel.create_table(read_capacity_units=1, write_capacity_units=1, wait=True)
    data = {b'\x00hey\xfb', b'\x00beautiful\xfb'}
    pkey = 'pkey'
    DataModel(pkey, data=data).save()
    m = DataModel.get(pkey)
    assert m.data == data

    new_data = {b'\xff'}
    m.update(actions=[DataModel.data.set(new_data)])
    assert new_data == m.data