fixing error with build.

toumorokoshi · Jan 22, 2016 · a5ebc9f · a5ebc9f
1 parent 868de4f
commit a5ebc9f
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/tests/test_history.py b/tests/test_history.py
@@ -40,3 +40,15 @@ def test_not_able_to_serialize_subattribute(history):
     history["unserializable"] = {"foo": re.compile("foo")}
     with pytest.raises(HistoryException):
         history.save()
+
+
+def test_history_with_build(tmpdir, build):
+
+    @build.task
+    def main(build):
+        was_set = build.history.get("set", False)
+        build.history["set"] = True
+        return was_set
+
+    assert build.run_task("main") is False
+    assert build.run_task("main") is True
diff --git a/uranium/build.py b/uranium/build.py
@@ -178,7 +178,9 @@ def _finalize(self):
         activate_content = ""
         activate_content += self.envvars.generate_activate_content()
         write_activate_this(self._root, additional_content=activate_content)
-        self.history[HISTORY_KEY]["python_version"] = "{0}.{1}".format(*sys.version_info[:2])
+        self.history[HISTORY_KEY] = {
+            "python_version": "{0}.{1}".format(*sys.version_info[:2])
+        }
         self.history.save()