Skip to content
Snippets Groups Projects
Commit f381a451 authored by Stefano Borini's avatar Stefano Borini
Browse files

Added workflow reader tests

parent 719835bb
No related branches found
No related tags found
1 merge request!29Extract io layer to writer/reader class
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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment