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)