test_db.py
# Copyright (C) 2015-2017 The Software Heritage developers
# See the AUTHORS file at the top-level directory of this distribution
# License: GNU General Public License version 3, or any later version
# See top-level LICENSE file for more information
from swh.model.hashutil import hash_to_bytes
def test_add_content(swh_storage):
with swh_storage.db() as db:
cur = db.cursor()
sha1 = hash_to_bytes("34973274ccef6ab4dfaaf86599792fa9c3fe4689")
db.mktemp("content", cur)
db.copy_to(
[
{
"sha1": sha1,
"sha1_git": hash_to_bytes(
"d81cc0710eb6cf9efd5b920a8453e1e07157b6cd"
),
"sha256": hash_to_bytes(
"673650f936cb3b0a2f93ce09d81be107"
"48b1b203c19e8176b4eefc1964a0cf3a"
),
"blake2s256": hash_to_bytes(
"69217a3079908094e11121d042354a7c"
"1f55b6482ca1a51e1b250dfd1ed0eef9"
),
"length": 3,
}
],
"tmp_content",
["sha1", "sha1_git", "sha256", "blake2s256", "length"],
cur,
)
db.content_add_from_temp(cur)
cur.execute("SELECT sha1 FROM content WHERE sha1 = %s", (sha1,))
assert cur.fetchone()[0] == sha1