• Search by Module
• Search by Words
• Search Projects

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

• elasticsearch-dsl-py-master
  • AUTHORS
  • elasticsearch_dsl
    • index.py
    • connections.py
    • exceptions.py
    • serializer.py
    • search.py
    • analysis.py
    • mapping.py
    • wrappers.py
    • response
      • __init__.py
      • hit.py
      • aggs.py
    • faceted_search.py
    • update_by_query.py
    • document.py
    • query.py
    • function.py
    • __init__.py
    • utils.py
    • field.py
    • aggs.py
  • examples
    • completion.py
    • parent_child.py
    • search_as_you_type.py
    • README.rst
    • alias_migration.py
    • composite_agg.py
    • percolate.py
  • LICENSE
  • README
  • README.rst
  • setup.py
  • test_elasticsearch_dsl
    • test_integration
      • test_mapping.py
      • test_examples
        • test_percolate.py
        • completion.py
        • parent_child.py
        • test_alias_migration.py
        • test_completion.py
        • test_parent_child.py
        • test_composite_aggs.py
        • alias_migration.py
        • composite_agg.py
        • __init__.py
        • percolate.py
      • test_count.py
      • test_data.py
      • test_index.py
      • test_document.py
      • test_faceted_search.py
      • test_search.py
      • test_update_by_query.py
      • __init__.py
      • test_analysis.py
    • test_mapping.py
    • test_utils.py
    • test_wrappers.py
    • test_query.py
    • test_aggs.py
    • test_index.py
    • test_document.py
    • run_tests.py
    • test_faceted_search.py
    • test_field.py
    • test_search.py
    • test_update_by_query.py
    • __init__.py
    • test_connections.py
    • test_analysis.py
    • test_validation.py
    • test_result.py
    • conftest.py
  • Changelog.rst
  • setup.cfg
  • .travis.yml
  • CONTRIBUTING.rst
  • .gitignore
  • docs
    • update_by_query.rst
    • Makefile
    • persistence.rst
    • api.rst
    • search_dsl.rst
    • Changelog.rst
    • CONTRIBUTING.rst
    • index.rst
    • conf.py
    • configuration.rst
    • faceted_search.rst
  • MANIFEST.in

import string
from random import choice

from pytest import raises

from elasticsearch_dsl import Date, Document, Index, IndexTemplate, Text, analyzer


class Post(Document):
    title = Text()
    published_from = Date()


def test_multiple_doc_types_will_combine_mappings():
    class User(Document):
        username = Text()

    i = Index('i')
    i.document(Post)
    i.document(User)
    assert {
        'mappings': {
            'properties': {
                'title': {'type': 'text'},
                'username': {'type': 'text'},
                'published_from': {'type': 'date'}
            }
        }
    } == i.to_dict()


def test_search_is_limited_to_index_name():
    i = Index('my-index')
    s = i.search()

    assert s._index == ['my-index']


def test_cloned_index_has_copied_settings_and_using():
    client = object()
    i = Index('my-index', using=client)
    i.settings(number_of_shards=1)

    i2 = i.clone('my-other-index')

    assert 'my-other-index' == i2._name
    assert client is i2._using
    assert i._settings == i2._settings
    assert i._settings is not i2._settings


def test_cloned_index_has_analysis_attribute():
    """
    Regression test for Issue #582 in which `Index.clone()` was not copying
    over the `_analysis` attribute.
    """
    client = object()
    i = Index('my-index', using=client)

    random_analyzer_name = ''.join((choice(string.ascii_letters) for _ in range(100)))
    random_analyzer = analyzer(random_analyzer_name, tokenizer="standard", filter="standard")

    i.analyzer(random_analyzer)

    i2 = i.clone('my-clone-index')

    assert i.to_dict()['settings']['analysis'] == i2.to_dict()['settings']['analysis']


def test_settings_are_saved():
    i = Index('i')
    i.settings(number_of_replicas=0)
    i.settings(number_of_shards=1)

    assert {
        'settings': {
            'number_of_shards': 1,
            'number_of_replicas': 0,
        }
    } == i.to_dict()


def test_registered_doc_type_included_in_to_dict():
    i = Index('i', using='alias')
    i.document(Post)

    assert {
        'mappings': {
            'properties': {
                'title': {'type': 'text'},
                'published_from': {'type': 'date'},
            }
        }
    } == i.to_dict()


def test_registered_doc_type_included_in_search():
    i = Index('i', using='alias')
    i.document(Post)

    s = i.search()

    assert s._doc_type == [Post]


def test_aliases_add_to_object():
    random_alias = ''.join((choice(string.ascii_letters) for _ in range(100)))
    alias_dict = {random_alias: {}}

    index = Index('i', using='alias')
    index.aliases(**alias_dict)

    assert index._aliases == alias_dict


def test_aliases_returned_from_to_dict():
    random_alias = ''.join((choice(string.ascii_letters) for _ in range(100)))
    alias_dict = {random_alias: {}}

    index = Index('i', using='alias')
    index.aliases(**alias_dict)

    assert index._aliases == index.to_dict()['aliases'] == alias_dict


def test_analyzers_added_to_object():
    random_analyzer_name = ''.join((choice(string.ascii_letters) for _ in range(100)))
    random_analyzer = analyzer(random_analyzer_name, tokenizer="standard", filter="standard")

    index = Index('i', using='alias')
    index.analyzer(random_analyzer)

    assert index._analysis["analyzer"][random_analyzer_name] == {"filter": ["standard"], "type": "custom", "tokenizer": "standard"}


def test_analyzers_returned_from_to_dict():
    random_analyzer_name = ''.join((choice(string.ascii_letters) for _ in range(100)))
    random_analyzer = analyzer(random_analyzer_name, tokenizer="standard", filter="standard")
    index = Index('i', using='alias')
    index.analyzer(random_analyzer)

    assert index.to_dict()["settings"]["analysis"]["analyzer"][random_analyzer_name] == {"filter": ["standard"], "type": "custom", "tokenizer": "standard"}


def test_conflicting_analyzer_raises_error():
    i = Index('i')
    i.analyzer('my_analyzer', tokenizer='whitespace', filter=['lowercase', 'stop'])

    with raises(ValueError):
        i.analyzer('my_analyzer', tokenizer='keyword', filter=['lowercase', 'stop'])


def test_index_template_can_have_order():
    i = Index('i-*')
    it = i.as_template('i', order=2)

    assert {
        "index_patterns": ["i-*"],
        "order": 2
    } == it.to_dict()


def test_index_template_save_result(mock_client):
    it = IndexTemplate('test-template', 'test-*')

    assert it.save(using='mock') == mock_client.indices.put_template()