fiableDB/test/test_delete.py

from fiable_db import start, add, delete, get_database


def test_delete_simple():
    """Test delete in the default table."""
    start()
    add(
        [
            {"name": "Noelia", "age": 34, "height": 165},
            {"name": "Juan", "age": 41, "height": 187},
            {"name": "Valentina", "age": 12, "height": 142},
        ]
    )
    delete(1)
    assert get_database() == [
        {
            "id": 1,
            "rev": 1,
            "table": "default",
            "data": {"name": "Noelia", "age": 34, "height": 165},
        },
        {
            "id": 2,
            "rev": 1,
            "table": "default",
            "data": {"name": "Juan", "age": 41, "height": 187},
        },
        {
            "id": 3,
            "rev": 1,
            "table": "default",
            "data": {"name": "Valentina", "age": 12, "height": 142},
        },
        {"id": 1, "rev": 2, "table": "default", "data": {}},
    ]


def test_delete_multiple():
    """Test delete two rows"""
    start()
    add(
        [
            {"name": "Noelia", "age": 34, "height": 165},
            {"name": "Juan", "age": 41, "height": 187},
            {"name": "Valentina", "age": 12, "height": 142},
        ]
    )
    delete(2)
    delete(3)
    assert get_database() == [
        {
            "id": 1,
            "rev": 1,
            "table": "default",
            "data": {"name": "Noelia", "age": 34, "height": 165},
        },
        {
            "id": 2,
            "rev": 1,
            "table": "default",
            "data": {"name": "Juan", "age": 41, "height": 187},
        },
        {
            "id": 3,
            "rev": 1,
            "table": "default",
            "data": {"name": "Valentina", "age": 12, "height": 142},
        },
        {"id": 2, "rev": 2, "table": "default", "data": {}},
        {"id": 3, "rev": 2, "table": "default", "data": {}},
    ]


def test_delete_with_table():
    """Test delete two rows"""
    start()
    add(
        [
            {"name": "Noelia", "age": 34, "height": 165},
            {"name": "Juan", "age": 41, "height": 187},
            {"name": "Valentina", "age": 12, "height": 142},
        ],
        table="people",
    )
    add(
        [
            {"name": "Mortadelo", "age": 22, "height": 184},
            {"name": "Filemon", "age": 25, "height": 185},
        ],
    )
    delete(2, table="people")
    delete(1)
    assert get_database() == [
        {
            "id": 1,
            "rev": 1,
            "table": "people",
            "data": {"name": "Noelia", "age": 34, "height": 165},
        },
        {
            "id": 2,
            "rev": 1,
            "table": "people",
            "data": {"name": "Juan", "age": 41, "height": 187},
        },
        {
            "id": 3,
            "rev": 1,
            "table": "people",
            "data": {"name": "Valentina", "age": 12, "height": 142},
        },
        {
            "id": 1,
            "rev": 1,
            "table": "default",
            "data": {"name": "Mortadelo", "age": 22, "height": 184},
        },
        {
            "id": 2,
            "rev": 1,
            "table": "default",
            "data": {"name": "Filemon", "age": 25, "height": 185},
        },
        {"id": 2, "rev": 2, "table": "people", "data": {}},
        {"id": 1, "rev": 2, "table": "default", "data": {}},
    ]