def test_select():
db = Database()
addr1 = 'localhost'
addr2 = '127.0.0.1'
addr3 = 'x011?!-\n/\\nw#!/.,x;{test}[lol]'
host1 = Host(addr1, db)
db.insert(Host(addr1, db))
assert db.select()[0].get_addr() == addr1
assert db.select(addr1).get_addr() == addr1
assert db.select(addr2) == None
assert db.select(addr3) == None
assert db.select([addr1, addr2, addr3]) == None
assert db.select((addr1, addr2, addr3)) == None
def test_insert():
db = Database()
addr1 = 'localhost'
addr2 = '127.0.0.1'
addr3 = 'x011?!-\n/\\nw#!/.,x;{test}[lol]'
host1 = Host(addr1, db)
db.insert(Host(addr1, db))
db.insert(Host(addr2, db))
assert db._Database__hosts[0].get_addr() == addr1
assert db._Database__hosts[1].get_addr() == addr2
with pytest.raises(ValueError):
db.insert(addr1)
with pytest.raises(ValueError):
db.insert([addr1])
with pytest.raises(ValueError):
db.insert((addr1))