def test_load_single(self, mock_sparql_delete, mock_sparql_insert):
with Store(self.data_path) as store:
store.add("0000-0003-1527-0030")
(orcid_id, active, last_update, person_uri, person_id, person_class, confirmed) = \
store["0000-0003-1527-0030"]
self.assertIsNone(last_update)
graph1, add_graph1, delete_graph1 = load_single(
"0000-0003-1527-0030", None, None, None, self.data_path,
"http://vivo.mydomain.edu/sparql", "*****@*****.**", "password")
self.assertEqual(319, len(add_graph1))
self.assertEqual(0, len(delete_graph1))
self.assertEqual(to_isomorphic(graph1), to_isomorphic(add_graph1))
with Store(self.data_path) as store:
# Last update now set
(orcid_id, active, last_update, person_uri, person_id, person_class, confirmed) = \
store["0000-0003-1527-0030"]
self.assertIsNotNone(last_update)
# Make sure turtle file created
self.assertTrue(
os.path.exists(
os.path.join(self.data_path, "0000-0003-1527-0030.ttl")))
# Now change a fact and run again. Changed fact is provided by vcr recording.
# Changed year of Amherst degree.
# Had to rig the Accept-Encoding to create the vcr recording with:
# r = requests.get('https://pub.orcid.org/v2.0/%s' % orcid,
# headers={"Accept": "application/json", "Accept-Encoding": "identity"})
graph2, add_graph2, delete_graph2 = load_single(
"0000-0003-1527-0030", None, None, None, self.data_path,
"http://vivo.mydomain.edu/sparql", "*****@*****.**", "password")
self.assertEqual(319, len(graph2))
self.assertEqual(17, len(add_graph2))
self.assertEqual(17, len(delete_graph2))
mock_sparql_insert.assert_has_calls([
call(add_graph1, "http://vivo.mydomain.edu/sparql",
"*****@*****.**", "password"),
call(add_graph2, "http://vivo.mydomain.edu/sparql",
"*****@*****.**", "password")
])
mock_sparql_delete.assert_has_calls([
call(delete_graph1, "http://vivo.mydomain.edu/sparql",
"*****@*****.**", "password"),
call(delete_graph2, "http://vivo.mydomain.edu/sparql",
"*****@*****.**", "password")
])
def test_persist(self):
self.assertFalse(os.path.exists(self.db_filepath))
with Store(self.data_path) as store:
self.assertEqual(self.db_filepath, store.db_filepath)
# Created
self.assertTrue(os.path.exists(self.db_filepath))
# Add
store.add("0000-0003-1527-0030")
# Still exists after close
self.assertTrue(os.path.exists(self.db_filepath))
with Store(self.data_path) as store:
self.assertTrue("0000-0003-1527-0030" in store)
def test_add_item(self):
with Store(self.data_path) as store:
# Insert
store.add("0000-0003-1527-0030")
(orcid_id, active, last_update, person_uri, person_id, person_class, confirmed) = \
store["0000-0003-1527-0030"]
self.assertTrue(active)
self.assertIsNone(person_uri)
self.assertIsNone(person_id)
self.assertIsNone(person_class)
self.assertIsNone(last_update)
self.assertFalse(confirmed)
# Update
store.add("0000-0003-1527-0030",
person_uri="http://me",
person_id="me",
person_class="Librarian",
confirmed=True)
(orcid_id, active, last_update, person_uri, person_id, person_class, confirmed) = \
store["0000-0003-1527-0030"]
self.assertTrue(active)
self.assertEqual("http://me", person_uri)
self.assertEqual("me", person_id)
self.assertEqual("Librarian", person_class)
self.assertIsNone(last_update)
self.assertTrue(confirmed)
def test_get_least_recent(self):
with Store(self.data_path) as store:
store.add("0000-0003-1527-0030")
store.add("0000-0003-1527-0031")
store.add("0000-0003-1527-0032")
# Deactivate one to make sure not returned.
del store["0000-0003-1527-0032"]
# Touch first
time.sleep(1)
t = datetime.datetime.utcnow()
time.sleep(1)
store.touch("0000-0003-1527-0030")
results = list(store.get_least_recent())
self.assertEqual(2, len(results))
self.assertEqual("0000-0003-1527-0031", results[0][0])
self.assertEqual("0000-0003-1527-0030", results[1][0])
# With limit
results = list(store.get_least_recent(limit=1))
self.assertEqual(1, len(results))
self.assertEqual("0000-0003-1527-0031", results[0][0])
# Before
results = list(store.get_least_recent(before_datetime=t))
self.assertEqual(1, len(results))
self.assertEqual("0000-0003-1527-0031", results[0][0])
def test_iter(self):
with Store(self.data_path) as store:
store.add("0000-0003-1527-0030")
store.add("0000-0003-1527-0031")
for orcid_id, active, new_last_update, person_uri, person_id, person_class, confirmed in store:
self.assertTrue(orcid_id in ("0000-0003-1527-0030",
"0000-0003-1527-0031"))
self.assertEqual(2, len(list(store)))
def test_contains(self):
with Store(self.data_path) as store:
# Add
store.add("0000-0003-1527-0030")
self.assertTrue("0000-0003-1527-0030" in store)
self.assertFalse("X000-0003-1527-0030" in store)
self.assertTrue(store.contains("0000-0003-1527-0030"))
self.assertTrue(store.contains("0000-0003-1527-0030", True))
self.assertFalse(store.contains("0000-0003-1527-0030", False))
def test_delete_all(self):
with Store(self.data_path) as store:
store.add("0000-0003-1527-0030")
store.add("0000-0003-1527-0031")
self.assertTrue("0000-0003-1527-0030" in store)
self.assertTrue("0000-0003-1527-0031" in store)
store.delete_all()
self.assertFalse("0000-0003-1527-0030" in store)
self.assertFalse("0000-0003-1527-0031" in store)
def test_touch(self):
with Store(self.data_path) as store:
store.add("0000-0003-1527-0030")
(orcid_id, active, last_update, person_uri, person_id, person_class, confirmed) = \
store["0000-0003-1527-0030"]
time.sleep(1)
store.touch("0000-0003-1527-0030")
(orcid_id, active, new_last_update, person_uri, person_id, person_class, confirmed) = \
store["0000-0003-1527-0030"]
self.assertNotEqual(last_update, new_last_update)
def test_add_deleted_item(self):
with Store(self.data_path) as store:
# Insert
store.add("0000-0003-1527-0030")
(orcid_id, active, last_update, person_uri, person_id, person_class, confirmed) = \
store["0000-0003-1527-0030"]
self.assertTrue(active)
self.assertIsNone(person_uri)
self.assertIsNone(person_id)
self.assertIsNone(person_class)
self.assertIsNone(last_update)
self.assertFalse(confirmed)
# Delete
del store["0000-0003-1527-0030"]
# Add again
store.add("0000-0003-1527-0030")
(orcid_id, active, last_update, person_uri, person_id, person_class, confirmed) = \
store["0000-0003-1527-0030"]
self.assertTrue(active)
self.assertIsNone(person_uri)
self.assertIsNone(person_id)
self.assertIsNone(person_class)
self.assertIsNone(last_update)
def test_del(self):
with Store(self.data_path) as store:
store.add("0000-0003-1527-0030")
self.assertTrue("0000-0003-1527-0030" in store)
del store["0000-0003-1527-0030"]
self.assertFalse("0000-0003-1527-0030" in store)