buku/tests/test_import_firefox_json.py

import json
from buku import import_firefox_json


def test_load_from_empty():
    """test method."""
    # Arrange
    data = json.loads("{}")

    # Act
    items = import_firefox_json(data)

    # Assert
    count = sum(1 for _ in items)
    assert 0 == count

def test_load_full_entry():
    """test method."""

    # Arrange
    data = json.loads("""
        {
            "title" : "title",
            "children": [
                {
                  "dateAdded": 1269200039653000,
                  "guid": "xxxydfalkj",
                  "id": 113,
                  "index": 0,
                  "lastModified": 1305978154986000,
                  "title": "title",
                  "type": "text/x-moz-place",
                  "typeCode": 1,
                  "tags" : "x,y",
                  "uri": "http://uri.com/abc?234&536",
                  "annos" : [{
                         "name": "bookmarkProperties/description",
                         "value": "desc"
                   }]
                }]
        }""")

    # Act
    items = import_firefox_json(data)

    # Assert
    result = []
    for item in items:
        result.append(item)

    assert 1 == len(result)
    assert 'http://uri.com/abc?234&536' == result[0][0]
    assert 'title' == result[0][1]
    assert ',x,y,' == result[0][2]
    assert 'desc' == result[0][3]


def test_load_no_typecode():
    """test method."""
    # Arrange
    data = json.loads("""
        {
            "title" : "title",
            "children": [
                {
                    "title" : "title1",
                    "uri" : "http://uri1",
                    "annos" : [{
                         "name": "bookmarkProperties/description",
                         "value": "desc"
                     }]
                }]
        }""")

    # Act
    items = import_firefox_json(data)

    # Assert
    result = []
    for item in items:
        result.append(item)

    assert 0 == len(result)


def test_load_invalid_typecode():
    """test method."""
    # Arrange
    data = json.loads("""
        {
            "title" : "title",
            "children": [
                {
                    "title" : "title1",
                    "typeCode" : 99,
                    "uri" : "http://uri1",
                    "annos" : [{
                         "name": "bookmarkProperties/description",
                         "value": "desc"
                     }]
                }]
        }""")
    # Act
    items = import_firefox_json(data)

    # Assert
    result = []
    for item in items:
        result.append(item)

    assert 0 == len(result)


def test_load_one_child():
    """test method."""

    # Arrange
    data = json.loads("""
        {
            "title" : "title",
            "typeCode" : 2,
            "children": [
                {
                    "title" : "title1",
                    "typeCode" : 1,
                    "uri" : "http://uri1",
                    "annos" : [{
                         "name": "bookmarkProperties/description",
                         "value": "desc"
                      }]
                 }
            ]
        } """)

    # Act
    items = import_firefox_json(data)

    # Assert
    result = []
    for item in items:
        result.append(item)

    assert 1 == len(result)
    assert 'http://uri1' == result[0][0]
    assert 'title1' == result[0][1]
    assert ',' == result[0][2]
    assert 'desc' == result[0][3]

def test_load_one_container_child():
    """test method."""

    # Arrange
    data = json.loads("""
        {
            "title" : "title",
            "typeCode" : 2,
            "children": [
                {
                    "title":"bookmark folder",
                    "typeCode":2
                } ]
         }""")

    # Act
    items = import_firefox_json(data)

    # Assert
    result = []
    for item in items:
        result.append(item)

    assert 0 == len(result)

def test_load_many_children():
    """test method."""

    # Arrange
    data = json.loads("""
            {
                "title":"Weitere Lesezeichen",
                "typeCode":2,
                "children": [
                    {"title":"title1","typeCode":1,"uri":"http://uri1.com/#more-74"},
                    {"title":"title2","typeCode":1,"uri":"http://uri2.com/xyz"},
                    {"title":"title3","typeCode":1,"uri":"http://uri3.com"}
                ]
            } """)

    # Act
    items = import_firefox_json(data)

    # Assert
    result = []
    for item in items:
        result.append(item)

    assert 3 == len(result)

def test_load_hierarchical_container():
    """test method."""

    # Arrange
    data = json.loads("""
        {
            "title" : "title",
            "typeCode" : 2,
            "children": [
                    {
                        "title" : "title",
                        "typeCode" : 2,
                        "children": [
                            {"title":"title1","typeCode":1,"uri":"http://uri1.com/#more-74"},
                            {"title":"title2","typeCode":1,"uri":"http://uri2.com/xyz"},
                            {"title":"title3","typeCode":1,"uri":"http://uri3.com"}
                        ]
                    },
                    {"title":"title4","typeCode":1,"uri":"http://uri4.com/#more-74"},
                    {"title":"title5","typeCode":1,"uri":"http://uri5.com/xyz"},
                    {"title":"title6","typeCode":1,"uri":"http://uri6.com"}
            ]
        }
                      """)

    # Act
    items = import_firefox_json(data)

    # Assert
    result = []
    for item in items:
        result.append(item)

    assert 6 == len(result)
    assert 'http://uri1.com/#more-74' == result[0][0]
    assert 'http://uri2.com/xyz' == result[1][0]
    assert 'http://uri3.com' == result[2][0]
    assert 'http://uri4.com/#more-74' == result[3][0]
    assert 'http://uri5.com/xyz' == result[4][0]
    assert 'http://uri6.com' == result[5][0]

def test_load_separator():
    """test method."""

    # Arrange
    data = json.loads("""
        {
            "title" : "title",
            "typeCode" : 2,
            "children": [
                {
                  "title": "",
                  "type": "text/x-moz-place-separator",
                  "typeCode": 3
                } ]
         }""")

    # Act
    items = import_firefox_json(data)

    # Assert
    result = []
    for item in items:
        result.append(item)

    assert 0 == len(result)

def test_load_multiple_tags():
    """test method."""
    # Arrange
    data = json.loads("""
        {
            "title" : "title",
            "children": [
                {
                    "title" : "title1",
                    "uri" : "http://uri1",
                    "tags" : "tag1, tag2",
                    "typeCode": 1,
                    "annos" : [{
                         "name": "bookmarkProperties/description",
                         "value": "desc"
                     }]
                }]
        }""")

    # Act
    items = import_firefox_json(data)

    # Assert
    result = []
    for item in items:
        result.append(item)

    assert 1 == len(result)
    assert ",tag1,tag2," == result[0][2]