def test_create_block_when_give_a_multi_line_code(self):
patch_input = mock.MagicMock(return_value="i int}")
block_generator = BlockGenerator(patch_input)
block = block_generator.generate("type Counter struct {")
self.assertEqual(
block.get_codes(), ["type Counter struct {", "i int}"]
)
def test_create_block_when_give_a_multi_line_code_with_bracket(self):
patch_input = mock.MagicMock(return_value="i)")
block_generator = BlockGenerator(patch_input)
block = block_generator.generate("fmt.Println(")
self.assertEqual(
block.get_codes(), ["fmt.Println(", "i)"]
)
def test_create_block_when_give_a_simple_code(self):
block_generator = BlockGenerator(None)
block = block_generator.generate("a := 222")
self.assertEqual(block.get_codes(), ["a := 222"])
