Result of dojo 2018-10-06

rafaelhenrique · rafaelhenrique · commit c81c276b317d · 2018-10-06T12:21:32.000-03:00
diff --git a/2018-10-06/commands.py b/2018-10-06/commands.py
@@ -0,0 +1,52 @@
+
+DB = {}
+
+
+def _init_db():
+    DB.clear()
+
+
+def command_add(_id, first_name, last_name, birthday, phone_number):
+    if _id in DB:
+        return "ID " + str(_id) + " ja cadastrado."
+
+    DB[_id] = {
+        'fn': first_name,
+        'ln': last_name,
+        'bd': birthday,
+        'pn': phone_number
+    }
+    return ""
+
+
+def command_del(_id):
+    try:
+        DB.pop(_id)
+        return ''
+    except KeyError:
+        return f'ID {_id} nao existente.'
+
+
+def command_info(_id):
+    try:
+        data = DB[_id]
+        return ' '.join(data.values())
+    except KeyError:
+        return f'ID {_id} nao existente.'
+
+
+def command_query(query):
+    # fn:Roberto
+    result = []
+
+    tag, data_value = query.split(':')
+    for key, value in DB.items():
+        if value[tag] == data_value:
+            result.append(key)
+
+    # result2 = []
+    # for number in result:
+    #     result2.append(str(number))
+    # return ' '.join(result2)
+
+    return ' '.join(str(number) for number in sorted(result))
diff --git a/2018-10-06/commands_test.py b/2018-10-06/commands_test.py
@@ -0,0 +1,98 @@
+import pytest
+from commands import command_add, command_del, command_info, command_query, DB, _init_db
+
+
+@pytest.fixture
+def roberto():
+    return [
+        123,
+        'Roberto',
+        'Nascimento',
+        '01/01/1960',
+        '+55-21-0190-0190'
+    ]
+
+
+@pytest.fixture
+def rosangela():
+    return [
+        456,
+        'Rosangela',
+        'Nascimento',
+        '01/01/1960',
+        '+55-21-0190-0190'
+    ]
+
+
+def test_command_add(roberto):
+    _init_db()
+    assert command_add(*roberto) == ''
+
+
+def test_command_add_values(roberto):
+    _init_db()
+    roberto[0] = 124
+    command_add(*roberto)
+    assert len(DB) == 1
+
+
+def test_command_add_id(roberto):
+    _init_db()
+    roberto[0] = 125
+    command_add(*roberto)
+    assert 125 in DB
+
+
+def test_command_add_same(roberto):
+    _init_db()
+    assert command_add(*roberto) == ''
+    assert command_add(*roberto) == 'ID 123 ja cadastrado.'
+
+
+def test_command_del(roberto):
+    _init_db()
+    command_add(*roberto)
+    assert command_del(123) == ''
+
+
+def test_command_del_not_found_id():
+    _init_db()
+    assert command_del(123) == 'ID 123 nao existente.'
+
+
+def test_command_del_2(roberto):
+    _init_db()
+    command_add(*roberto)
+    assert command_del(123) == ''
+    assert command_del(123) == 'ID 123 nao existente.'
+
+
+def test_command_info(roberto):
+    _init_db()
+    command_add(*roberto)
+    assert command_info(roberto[0]) == 'Roberto Nascimento 01/01/1960 +55-21-0190-0190'
+
+
+def test_command_info_not_found():
+    _init_db()
+    assert command_info(1) == 'ID 1 nao existente.'
+
+
+def test_command_query(roberto):
+    _init_db()
+    command_add(*roberto)
+    assert command_query('fn:Roberto') == '123'
+
+
+def test_command_query_multiple(roberto, rosangela):
+    _init_db()
+    command_add(*roberto)
+    command_add(*rosangela)
+    assert command_query('ln:Nascimento') == '123 456'
+
+
+def test_command_query_multiple_unordered(roberto, rosangela):
+    _init_db()
+    command_add(*rosangela)
+    command_add(*roberto)
+    assert command_query('ln:Nascimento') == '123 456'
