def test_keywords_mixed_add_get(storage_socket):
opts1 = ptl.models.KeywordSet(values={"o": 5})
id1 = storage_socket.add_keywords([opts1])["data"][0]
opts2 = ptl.models.KeywordSet(values={"o": 6})
opts = storage_socket.get_add_keywords_mixed([opts1, opts2, id1, bad_id1, bad_id2])["data"]
assert opts[0].id == id1
assert opts[1].values["o"] == 6
assert opts[2].id == id1
assert opts[3] is None
assert opts[4] is None
assert 1 == storage_socket.del_keywords(id=id1)
assert 1 == storage_socket.del_keywords(id=opts[1].id)
def kw_fixtures(storage_socket):
kw1 = ptl.models.KeywordSet(**{"values": {"something": "kwfixture"}})
ret = storage_socket.add_keywords([kw1])
yield list(ret['data'])
r = storage_socket.del_keywords(ret['data'][0])
assert r == 1
def test_keywords_add(storage_socket):
kw = ptl.models.KeywordSet(**{"values": {"o": 5}, "hash_index": "something_unique"})
ret = storage_socket.add_keywords([kw, kw.copy()])
assert len(ret["data"]) == 2
assert ret["meta"]["n_inserted"] == 1
assert ret["data"][0] == ret["data"][1]
ret = storage_socket.add_keywords([kw])
assert ret["meta"]["n_inserted"] == 0
ret = storage_socket.get_keywords(hash_index="something_unique")
ret_kw = ret["data"][0]
assert ret["meta"]["n_found"] == 1
assert ret_kw.values == kw.values
assert 1 == storage_socket.del_keywords(id=ret_kw.id)
