From bbbb2a53d5decf3b613a92c4ff77c84bfc5d4903 Mon Sep 17 00:00:00 2001 From: Paul Eggleton Date: Tue, 13 Dec 2016 20:07:11 +1300 Subject: data_smart: support serialisation The COW object used within VariableHistory can't be serialised itself, but we can convert it to a dict when serializing and then back when deserialising. This finally allows DataSmart objects to be serialized. NOTE: "serialisation" here means pickling, not over XMLRPC or any other transport. Signed-off-by: Paul Eggleton Signed-off-by: Richard Purdie --- lib/bb/tests/data.py | 23 +++++++++++++++++++++++ 1 file changed, 23 insertions(+) (limited to 'lib/bb/tests') diff --git a/lib/bb/tests/data.py b/lib/bb/tests/data.py index a17245f90..ba30d1280 100644 --- a/lib/bb/tests/data.py +++ b/lib/bb/tests/data.py @@ -446,6 +446,29 @@ class Contains(unittest.TestCase): self.assertFalse(bb.utils.contains_any("SOMEFLAG", "x y z", True, False, self.d)) +class Serialize(unittest.TestCase): + + def test_serialize(self): + import tempfile + import pickle + d = bb.data.init() + d.enableTracking() + d.setVar('HELLO', 'world') + d.setVarFlag('HELLO', 'other', 'planet') + with tempfile.NamedTemporaryFile(delete=False) as tmpfile: + tmpfilename = tmpfile.name + pickle.dump(d, tmpfile) + + with open(tmpfilename, 'rb') as f: + newd = pickle.load(f) + + os.remove(tmpfilename) + + self.assertEqual(d, newd) + self.assertEqual(newd.getVar('HELLO', True), 'world') + self.assertEqual(newd.getVarFlag('HELLO', 'other'), 'planet') + + class Remote(unittest.TestCase): def test_remote(self): class TestConnector: -- cgit 1.2.3-korg