35 lines
2.6 KiB
Python
35 lines
2.6 KiB
Python
import json
|
|
import tempfile
|
|
import unittest
|
|
from pathlib import Path
|
|
from spatial_lab.project import Project,write_json
|
|
from spatial_lab.examples import small_project
|
|
|
|
class Editing(unittest.TestCase):
|
|
def setUp(self):
|
|
self.tmp=tempfile.TemporaryDirectory();self.path=Path(self.tmp.name)/'study.json';write_json(self.path,small_project());self.project=Project(self.path)
|
|
def tearDown(self):self.tmp.cleanup()
|
|
def test_edit_undo_redo_exactly(self):
|
|
original=self.project.build();modified=self.project.apply([{'action':'set_parameter','name':'height','value':8}])
|
|
self.assertNotEqual(original['bundle_hash'],modified['bundle_hash']);self.assertEqual([o['id'] for o in original['objects']],[o['id'] for o in modified['objects']])
|
|
self.assertEqual(Project(self.path).travel('undo'),original);self.assertEqual(Project(self.path).travel('redo'),modified)
|
|
def test_failed_transaction_changes_nothing(self):
|
|
before=self.path.read_bytes()
|
|
with self.assertRaises(ValueError):self.project.apply([{'action':'set_parameter','name':'height','value':9},{'action':'patch_node','id':'path','patch':{'params':{'xyz':['1/0',0,0]}}}])
|
|
self.assertEqual(self.path.read_bytes(),before);self.assertFalse(self.project.history_path.exists())
|
|
def test_dependency_aware_removal(self):
|
|
with self.assertRaisesRegex(ValueError,'used by'):self.project.apply([{'action':'remove_node','id':'path'}])
|
|
b=self.project.apply([{'action':'remove_node','id':'path','cascade':True}]);self.assertEqual(b['stats']['nodes'],0)
|
|
self.assertEqual(self.project.travel('undo')['stats']['nodes'],3)
|
|
def test_add_and_patch_nodes(self):
|
|
b=self.project.apply([{'action':'upsert_node','node':{'id':'echo','op':'transform','inputs':{'source':'gallery'},'params':{'translate':[0,0,10]}}},{'action':'patch_node','id':'gallery','patch':{'visible':False}}])
|
|
self.assertEqual(b['objects'][0]['id'],'echo/main');self.assertGreater(b['bounds']['min'][2],9)
|
|
def test_new_edit_clears_redo(self):
|
|
self.project.apply([{'action':'set_parameter','name':'height','value':8}]);self.project.travel('undo');self.project.apply([{'action':'set_parameter','name':'height','value':6}])
|
|
with self.assertRaisesRegex(ValueError,'Nothing to redo'):self.project.travel('redo')
|
|
def test_external_edit_does_not_restore_stale_history(self):
|
|
self.project.apply([{'action':'set_parameter','name':'height','value':8}]);p=self.project.read();p['parameters']['height']=10;write_json(self.path,p)
|
|
with self.assertRaisesRegex(ValueError,'Nothing to undo'):self.project.travel('undo')
|
|
|
|
if __name__=='__main__':unittest.main()
|