def test_config_store():
store = sign.MemorySignatureStore()
c = Config()
c.NotebookNotary.store_factory = lambda: store
notary = sign.NotebookNotary(config=c)
assert notary.store is store
def setUp(self):
self.data_dir = tempfile.mkdtemp()
self.notary = sign.NotebookNotary(
db_file=':memory:',
secret=b'secret',
data_dir=self.data_dir,
)
with self.fopen(u'test3.ipynb', u'r') as f:
self.nb = read(f, as_version=4)
with self.fopen(u'test3.ipynb', u'r') as f:
self.nb3 = read(f, as_version=3)
def setUp(self):
self.data_dir = tempfile.mkdtemp()
self.notary = sign.NotebookNotary(
db_file=":memory:",
secret=b"secret",
data_dir=self.data_dir,
)
with self.fopen("test3.ipynb", "r") as f:
self.nb = read(f, as_version=4)
with self.fopen("test3.ipynb", "r") as f:
self.nb3 = read(f, as_version=3)
def test_invalid_db_file(self):
invalid_sql_file = os.path.join(self.data_dir, 'invalid_db_file.db')
with open(invalid_sql_file, 'w') as tempfile:
tempfile.write(u'[invalid data]')
invalid_notary = sign.NotebookNotary(
db_file=invalid_sql_file,
secret=b'secret',
)
invalid_notary.sign(self.nb)
testpath.assert_isfile(os.path.join(self.data_dir, invalid_sql_file))
testpath.assert_isfile(
os.path.join(self.data_dir, invalid_sql_file + '.bak'))
def test_invalid_db_file(self):
invalid_sql_file = os.path.join(self.data_dir, "invalid_db_file.db")
with open(invalid_sql_file, "w") as tempfile:
tempfile.write("[invalid data]")
invalid_notary = sign.NotebookNotary(
db_file=invalid_sql_file,
secret=b"secret",
)
invalid_notary.sign(self.nb)
invalid_notary.store.close()
testpath.assert_isfile(os.path.join(self.data_dir, invalid_sql_file))
testpath.assert_isfile(
os.path.join(self.data_dir, invalid_sql_file + ".bak"))
def _notary_default(self):
return sign.NotebookNotary(parent=self)
