def storage():
ds = DataStorage()
ds.create_item("John", "Smith", "+1111111111", "05.11.1955")
ds.create_item("John", "Doe", "+122222222", "14.11.1970")
ds.create_item("Maria", "Hanasaku", "+1333333333", "11.11.1911")
ds.create_item("David", "Iroha", "+1444444444", "08.11.1855")
return ds
def saved_storage():
ds = DataStorage()
ds.create_item("John", "Smith", "+1111111111", "05.11.1955")
ds.create_item("John", "Doe", "+122222222", "14.11.1970")
ds.create_item("Maria", "Hanasaku", "+1333333333", "11.11.1911")
ds.create_item("David", "Iroha", "+1444444444", "08.11.1855")
ds.save("tmp.db")
yield ds
os.remove("tmp.db")
def test_add_few_items():
ds = DataStorage()
ds.create_item("John", "First", "+1111111111", "05.11.2016")
ds.create_item("John", "Second", "+1111111111", "05.11.2016")
ds.create_item("John", "Third", "+1111111111", "05.11.2016")
items = ds.items
assert len(items) == 3
assert items[0].last_name == "First" and items[
1].last_name == "Second" and items[2].last_name == "Third"
def test_add_one_item():
ds = DataStorage()
ds.create_item("John", "Smith", "+1111111111", "05.11.2016")
items = ds.items
assert len(items) == 1
assert items[0].first_name == "John" and items[0].last_name == "Smith"