augustjd / flask-warehouse Goto Github PK

@mock_s3
def test_s3_service_metadata_preservation(app):
    """Sample pytest test function with the pytest fixture as an argument.
    """
    app.config['WAREHOUSE_DEFAULT_SERVICE'] = 's3'
    app.config['WAREHOUSE_DEFAULT_LOCATION'] = 'us-west-1'

    warehouse = Warehouse(app)

    assert warehouse.service is not None
    assert warehouse.service.id == 's3'

    with app.app_context():
        cubby = warehouse('s3:///something/beautiful')
        cubby.store(bytes=b'12345')

        # setting meta data works
        assert cubby.metadata() == {}
        metadata = {"tag": "value"}
        assert cubby.set_metadata(metadata)
        assert cubby.metadata() == metadata

        # setting mimetype works
        assert cubby.mimetype() == None
        cubby.set_mimetype("application/octet-stream")
        assert cubby.mimetype() == 'application/octet-stream'

        # this fails, setting mimetype actually overwrites metadata
        assert cubby.metadata() == metadata

>           assert cubby.metadata() == metadata
E           AssertionError: assert {} == {'tag': 'value'}
E             Right contains more items:
E             {'tag': 'value'}
E             Full diff:
E             - {}
E             + {'tag': 'value'}