def setUp(self): _db = getDBConnection('file://test.json') with open("test-file.txt", "w") as f: f.write("This is a test file.") self.artifact = Artifact.registerArtifact( name=f'test-artifact', typ='text', path=f'test-file.txt', cwd='./', command='echo "This is a test file" > test-file.txt', inputs=[], documentation=f"This artifact is made for testing.")
return self.db[self.hashes[key]] def upload(self, key, path): pass def downloadFile(self, key, path): pass # Add the MockDB as a scheme artifact._artifactdb._db_schemes["mockdb"] = MockDB # This needs to be a global variable so # that this getDBConnection is the first # call to create a DB connection _db = getDBConnection("mockdb://") class TestGit(unittest.TestCase): def test_keys(self): git = artifact.artifact.getGit(Path(".")) self.assertSetEqual( set(git.keys()), set(["origin", "hash", "name"]), "git keys wrong" ) def test_origin(self): git = artifact.artifact.getGit(Path(".")) self.assertTrue( git["origin"].endswith("gem5art"), "Origin should end with gem5art" )
return self.db[self.hashes[key]] def upload(self, key, path): pass def downloadFile(self, key, path): pass # Add the MockDB as a scheme artifact._artifactdb._db_schemes['mockdb'] = MockDB # This needs to be a global variable so # that this getDBConnection is the first # call to create a DB connection _db = getDBConnection('mockdb://') class TestGit(unittest.TestCase): def test_keys(self): git = artifact.artifact.getGit(Path('.')) self.assertSetEqual(set(git.keys()), set(['origin', 'hash', 'name']), "git keys wrong") def test_origin(self): git = artifact.artifact.getGit(Path('.')) self.assertTrue(git['origin'].endswith('gem5art'), "Origin should end with gem5art") class TestArtifact(unittest.TestCase):
return self.db[key] else: # This is a hash return self.db[self.hashes[key]] def upload(self, key, path): pass def downloadFile(self, key, path): pass # This needs to be a global variable so # that this getDBConnection is the first # call to create a DB connection _db = getDBConnection(typ = MockDB) class TestGit(unittest.TestCase): def test_keys(self): git = artifact.artifact.getGit(Path('.')) self.assertSetEqual(set(git.keys()), set(['origin', 'hash', 'name']), "git keys wrong") def test_origin(self): git = artifact.artifact.getGit(Path('.')) self.assertTrue(git['origin'].endswith('gem5art'), "Origin should end with gem5art") class TestArtifact(unittest.TestCase):