Skip to content

Commit 554b75a

Browse files
committed
Add functional tests for updating tags & metadata for runs in the created state
1 parent 1a9a1d9 commit 554b75a

File tree

2 files changed

+57
-0
lines changed

2 files changed

+57
-0
lines changed
Lines changed: 28 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,28 @@
1+
import unittest
2+
import uuid
3+
from simvue import Run, Client
4+
5+
import common
6+
7+
class TestRunMetadataUpdatedCreated(unittest.TestCase):
8+
def test_run_metadata_update_created(self):
9+
"""
10+
Check metadata can be updated & retrieved
11+
"""
12+
name = 'test-%s' % str(uuid.uuid4())
13+
metadata = {'a': 'string', 'b': 1, 'c': 2.5}
14+
run = Run()
15+
run.init(name, metadata=metadata, folder=common.FOLDER, running=False)
16+
run.update_metadata({'b': 2})
17+
18+
metadata['b'] = 2
19+
20+
client = Client()
21+
data = client.get_run(name, metadata=True)
22+
self.assertEqual(data['metadata'], metadata)
23+
24+
runs = client.delete_runs(common.FOLDER)
25+
self.assertEqual(len(runs), 1)
26+
27+
if __name__ == '__main__':
28+
unittest.main()
Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,29 @@
1+
import unittest
2+
import uuid
3+
from simvue import Run, Client
4+
5+
import common
6+
7+
class TestRunUpdateTagsCreated(unittest.TestCase):
8+
def test_run_tags_update_created(self):
9+
"""
10+
Check tags can be updated & retrieved for a run in the
11+
created state
12+
"""
13+
name = 'test-%s' % str(uuid.uuid4())
14+
tags = ['a1']
15+
run = Run()
16+
run.init(name, tags=tags, folder=common.FOLDER, running=False)
17+
run.update_tags(['b2'])
18+
19+
tags.append('b2')
20+
21+
client = Client()
22+
data = client.get_run(name, tags=True)
23+
self.assertEqual(tags, data['tags'])
24+
25+
runs = client.delete_runs(common.FOLDER)
26+
self.assertEqual(len(runs), 1)
27+
28+
if __name__ == '__main__':
29+
unittest.main()

0 commit comments

Comments
 (0)