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 count == 0

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

    # Arrange
    data = json.loads("""
        {
            "title" : "main",
            "typeCode": 2,
            "children": [
                {
                "title" : "title",
                "typeCode": 2,
                "children": [
                    {
                      "dateAdded": 1269200039653000,
                      "guid": "xxxydfalkj",
                      "id": 113,
                      "index": 0,
                      "lastModified": 1305978154986000,
                      "title": "entry 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 len(result) == 1
    assert result[0][0] == 'http://uri.com/abc?234&536'
    assert result[0][1] == 'entry title'
    assert result[0][2] == ',x,y,'
    assert result[0][3] == 'desc'


def test_load_no_typecode():
    """test method."""
    # Arrange
    data = json.loads("""
        {
            "title" : "main",
            "typeCode": 2,
            "children": [
                {
                "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 len(result) == 0


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 len(result) == 0

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

    # Arrange
    data = json.loads("""
        {
            "title" : "title",
            "typeCode" : 2
        } """)

    # Act
    items = import_firefox_json(data)

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

    assert len(result) == 0

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

    # Arrange
    data = json.loads("""
        {
            "title" : "main",
            "typeCode" : 2,
            "children": [
                {
                "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 len(result) == 1
    assert result[0][0] == 'http://uri1'
    assert result[0][1] == 'title1'
    assert result[0][2] == ','
    assert result[0][3] == 'desc'

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

    # Arrange
    data = json.loads("""
        {
            "title" : "main",
            "typeCode": 2,
            "children": [
                {
                "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 len(result) == 0

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

    # Arrange
    data = json.loads("""
        {
            "title" : "main",
            "typeCode" : 2,
            "children": [
                {
                "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 len(result) == 3

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

    # Arrange
    data = json.loads("""
        {
            "title" : "main",
            "typeCode" : 2,
            "children": [
                {
                "typeCode" : 2,
                "children": [
                        {"title":"title1","typeCode":1,"uri":"http://uri.com"}
                ]}
            ]
        }
    """)

    # Act
    items = import_firefox_json(data, add_bookmark_folder_as_tag=True)

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

    assert len(result) == 1
    assert result[0][0] == 'http://uri.com'
    assert result[0][2] == ',<no title>,'

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

    # Arrange
    data = json.loads("""
        {
            "title" : "main",
            "typeCode" : 2,
            "children": [
                {
                "title" : "title",
                "typeCode" : 2,
                "children": [
                     {"title":"title1","typeCode":1,"uri":"http://uri.com"}
                ]
            }]
       }
    """)

    # Act
    items = import_firefox_json(data, add_bookmark_folder_as_tag=True)

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

    assert len(result) == 1
    assert result[0][0] == 'http://uri.com'
    assert result[0][2] == ',title,'

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

    # Arrange
    data = json.loads("""
        {
            "title" : "main",
            "typeCode" : 2,
            "children": [
                {
                "title" : "title",
                "typeCode" : 2,
                "root": "bookmarksMenuFolder",
                "children": [
                        {
                            "title" : "title2",
                            "typeCode" : 2,
                            "children": [
                                {"title":"title1","typeCode":1,"uri":"http://uri1.com/#more-74"}
                            ]
                        },
                        {"title":"title4","typeCode":1,"uri":"http://uri4.com/#more-74"}
                ]
            }]
       }
    """)

    # Act
    items = import_firefox_json(data, add_bookmark_folder_as_tag=True)

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

    assert len(result) == 2
    assert result[0][0] == 'http://uri1.com/#more-74'
    assert result[1][0] == 'http://uri4.com/#more-74'

    assert result[0][2] == ',title2,'
    assert result[1][2] == ','

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

    # Arrange
    data = json.loads("""
        {
            "title" : "main",
            "typeCode" : 2,
            "children": [
                {
                "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 len(result) == 0

def test_load_multiple_tags():
    """test method."""
    # Arrange
    data = json.loads("""
        {
            "title" : "main",
            "typeCode": 2,
            "children": [
                {
                "title" : "title",
                "typeCode": 2,
                "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 len(result) == 1
    assert result[0][2] == ",tag1,tag2,"