class TestJSHandler(unittest.TestCase): def setUp(self): with open("test.js") as test_file: test_code = test_file.read() self.js_handler_a = JavascriptHandler(test_code, "a") self.js_handler_b = JavascriptHandler(test_code, "b") def test_js_handler_a(self): before_code = self.js_handler_a.js_code assert isinstance(before_code, str) is True self.js_handler_a.extract_javascript_a() after_result = self.js_handler_a.js_code assert isinstance(after_result, list) is True def test_js_handler_b(self): before_code = self.js_handler_b.js_code assert isinstance(before_code, str) is True self.js_handler_b.extract_javascript_b() after_result = self.js_handler_b.js_code assert isinstance(after_result, list) is True @patch('dot_formatter.DotFormatter.convert_to_dot_a') def test_dot_converter_a(self, test_method): self.js_handler_a.create_puml() self.assertTrue(test_method.called) @patch('dot_formatter.DotFormatter.convert_to_dot_b') def test_dot_converter_b(self, test_method): self.js_handler_b.create_puml() self.assertTrue(test_method.called)
def handle_javascript(self, js: str, current_cmd: str): "Creates a javascript handler for given set of javascript file(s)" my_javascript = JavascriptHandler(js, current_cmd) my_serializer = Serializer() if current_cmd == "a": my_javascript.extract_javascript_a() my_serializer.serialize_a(my_javascript.js_code) else: my_javascript.extract_javascript_b() my_serializer.serialize_b(my_javascript.js_code) my_javascript.create_puml()
class JavaScriptHandlerTests(unittest.TestCase): def setUp(self): global test_js with open('test.js') as js_file: js = js_file.read() test_js += js + ' \n' def test_extract_javascript_a(self): self.js_handler = JavascriptHandler(test_js, "a") current_value = self.js_handler.js_code self.js_handler.extract_javascript_a() new_value = self.js_handler.js_code # Expect handler.js_code to have changed after extract_javascript method called self.assertNotEqual(current_value, new_value, "extract javascript a test failed") def test_extract_javascript_b(self): self.js_handler = JavascriptHandler(test_js, "b") current_value = self.js_handler.js_code self.js_handler.extract_javascript_b() new_value = self.js_handler.js_code # Expect js_handler.js_code to have changed after extract_javascript method called self.assertNotEqual(current_value, new_value, "extract javascript a test failed") @patch('dot_formatter.DotFormatter.convert_to_dot_a') def test_create_puml_a(self, MockMethod): # Create mock for my_dot_formatter.convert_to_dot_a and expect it to have been called self.js_handler = JavascriptHandler(test_js, "a") self.js_handler.create_puml() self.assertTrue(MockMethod.called) @patch('dot_formatter.DotFormatter.convert_to_dot_b') def test_create_puml_b(self, MockMethod): # Create mock for my_dot_formatter.convert_to_dot_b and expect it to have been called self.js_handler = JavascriptHandler(test_js, "b") self.js_handler.create_puml() self.assertTrue(MockMethod.called)