From f381a451da452d7f84150e43dd75110ea558077f Mon Sep 17 00:00:00 2001
From: Stefano Borini <sborini@enthought.com>
Date: Thu, 20 Jul 2017 09:50:08 +0100
Subject: [PATCH] Added workflow reader tests
---
force_bdss/io/tests/test_workflow_reader.py | 47 +++++++++++++++++++++
1 file changed, 47 insertions(+)
diff --git a/force_bdss/io/tests/test_workflow_reader.py b/force_bdss/io/tests/test_workflow_reader.py
index e69de29..3b6a7f8 100644
--- a/force_bdss/io/tests/test_workflow_reader.py
+++ b/force_bdss/io/tests/test_workflow_reader.py
@@ -0,0 +1,47 @@
+import json
+import unittest
+from six import StringIO
+
+from force_bdss.bundle_registry_plugin import BundleRegistryPlugin
+from force_bdss.io.workflow_reader import (
+ WorkflowReader,
+ InvalidVersionException, InvalidFileException)
+
+try:
+ import mock
+except ImportError:
+ from unittest import mock
+
+
+class TestWorkflowReader(unittest.TestCase):
+ def setUp(self):
+ self.mock_bundle_registry = mock.Mock(spec=BundleRegistryPlugin)
+ self.wfreader = WorkflowReader(self.mock_bundle_registry)
+
+ def test_initialization(self):
+ self.assertEqual(self.wfreader.bundle_registry,
+ self.mock_bundle_registry)
+
+ def test_invalid_version(self):
+ data = {
+ "version": "2"
+ }
+
+ with self.assertRaises(InvalidVersionException):
+ self.wfreader.read(self._as_json_stringio(data))
+
+ def test_absent_version(self):
+ data = {
+ }
+
+ with self.assertRaises(InvalidFileException):
+ self.wfreader.read(self._as_json_stringio(data))
+
+ def _as_json_stringio(self, data):
+ fp = StringIO()
+ json.dump(data, fp)
+ fp.seek(0)
+
+ return fp
+
+ def
--
GitLab