|  | 
| 3 | 3 | """Tests for notebook cell parsing.""" | 
| 4 | 4 | 
 | 
| 5 | 5 | import unittest | 
| 6 |  | -import sys | 
| 7 | 6 | from kernel_gateway.notebook_http.swagger.parser import SwaggerCellParser | 
| 8 | 7 | 
 | 
| 9 | 8 | 
 | 
| @@ -142,51 +141,48 @@ def test_endpoint_response_concatenation(self): | 
| 142 | 141 |         self.assertEqual(endpoints['/foo/:bar']['get'], '# ResponseInfo operationId: get\n') | 
| 143 | 142 | 
 | 
| 144 | 143 |     def test_undeclared_operations(self): | 
| 145 |  | -        if sys.version_info[:2] >= (3,4): | 
| 146 |  | -            """Parser should warn about operations that aren't documented in the | 
| 147 |  | -            swagger cell | 
| 148 |  | -            """ | 
| 149 |  | -            source_cells = [ | 
| 150 |  | -                {"source":'```\n{"swagger":"2.0", "paths": {"/foo": {"put": {"operationId":"putbar","parameters": [{"name": "bar"}]},"post":{"operationId":"postbar"},"get": {"operationId":"get","parameters": [{"name": "bar"}]}}}}\n```\n'}, | 
| 151 |  | -                {"source":'# operationId: get'}, | 
| 152 |  | -                {"source":'# operationId: postbar '}, | 
| 153 |  | -                {"source":'# operationId: putbar'}, | 
| 154 |  | -                {"source":'# operationId: extraOperation'}, | 
| 155 |  | -            ] | 
| 156 |  | -            with self.assertLogs(level='WARNING') as warnings: | 
| 157 |  | -                SwaggerCellParser(comment_prefix='#', notebook_cells=source_cells) | 
| 158 |  | -                for output in warnings.output: | 
| 159 |  | -                    self.assertRegex(output, 'extraOperation') | 
|  | 144 | +        """Parser should warn about operations that aren't documented in the | 
|  | 145 | +        swagger cell | 
|  | 146 | +        """ | 
|  | 147 | +        source_cells = [ | 
|  | 148 | +            {"source":'```\n{"swagger":"2.0", "paths": {"/foo": {"put": {"operationId":"putbar","parameters": [{"name": "bar"}]},"post":{"operationId":"postbar"},"get": {"operationId":"get","parameters": [{"name": "bar"}]}}}}\n```\n'}, | 
|  | 149 | +            {"source":'# operationId: get'}, | 
|  | 150 | +            {"source":'# operationId: postbar '}, | 
|  | 151 | +            {"source":'# operationId: putbar'}, | 
|  | 152 | +            {"source":'# operationId: extraOperation'}, | 
|  | 153 | +        ] | 
|  | 154 | +        with self.assertLogs(level='WARNING') as warnings: | 
|  | 155 | +            SwaggerCellParser(comment_prefix='#', notebook_cells=source_cells) | 
|  | 156 | +            for output in warnings.output: | 
|  | 157 | +                self.assertRegex(output, 'extraOperation') | 
| 160 | 158 | 
 | 
| 161 | 159 |     def test_undeclared_operations_reversed(self): | 
| 162 |  | -        if sys.version_info[:2] >= (3,4): | 
| 163 |  | -            """Parser should warn about operations that aren't documented in the | 
| 164 |  | -            swagger cell | 
| 165 |  | -            """ | 
| 166 |  | -            source_cells = [ | 
| 167 |  | -                {"source":'# operationId: get'}, | 
| 168 |  | -                {"source":'# operationId: postbar '}, | 
| 169 |  | -                {"source":'# operationId: putbar'}, | 
| 170 |  | -                {"source":'# operationId: extraOperation'}, | 
| 171 |  | -                {"source":'```\n{"swagger":"2.0", "paths": {"/foo": {"put": {"operationId":"putbar","parameters": [{"name": "bar"}]},"post":{"operationId":"postbar"},"get": {"operationId":"get","parameters": [{"name": "bar"}]}}}}\n```\n'}, | 
| 172 |  | -            ] | 
| 173 |  | -            with self.assertLogs(level='WARNING') as warnings: | 
| 174 |  | -                SwaggerCellParser(comment_prefix='#', notebook_cells=source_cells) | 
| 175 |  | -                for output in warnings.output: | 
| 176 |  | -                    self.assertRegex(output, 'extraOperation') | 
|  | 160 | +        """Parser should warn about operations that aren't documented in the | 
|  | 161 | +        swagger cell | 
|  | 162 | +        """ | 
|  | 163 | +        source_cells = [ | 
|  | 164 | +            {"source":'# operationId: get'}, | 
|  | 165 | +            {"source":'# operationId: postbar '}, | 
|  | 166 | +            {"source":'# operationId: putbar'}, | 
|  | 167 | +            {"source":'# operationId: extraOperation'}, | 
|  | 168 | +            {"source":'```\n{"swagger":"2.0", "paths": {"/foo": {"put": {"operationId":"putbar","parameters": [{"name": "bar"}]},"post":{"operationId":"postbar"},"get": {"operationId":"get","parameters": [{"name": "bar"}]}}}}\n```\n'}, | 
|  | 169 | +        ] | 
|  | 170 | +        with self.assertLogs(level='WARNING') as warnings: | 
|  | 171 | +            SwaggerCellParser(comment_prefix='#', notebook_cells=source_cells) | 
|  | 172 | +            for output in warnings.output: | 
|  | 173 | +                self.assertRegex(output, 'extraOperation') | 
| 177 | 174 | 
 | 
| 178 | 175 |     def test_unreferenced_operations(self): | 
| 179 |  | -        if sys.version_info[:2] >= (3,4): | 
| 180 |  | -            """Parser should warn about documented operations that aren't referenced | 
| 181 |  | -            in a cell | 
| 182 |  | -            """ | 
| 183 |  | -            source_cells = [ | 
| 184 |  | -                {"source": '```\n{"swagger":"2.0", "paths": {"/foo": {"put": {"operationId":"putbar","parameters": [{"name": "bar"}]},"post":{"operationId":"postbar"},"get": {"operationId":"get","parameters": [{"name": "bar"}]}}}}\n```\n'}, | 
| 185 |  | -                {"source": '# operationId: get'}, | 
| 186 |  | -                {"source": '# operationId: putbar'}, | 
| 187 |  | -                {"source": '# operationId: putbar '} | 
| 188 |  | -            ] | 
| 189 |  | -            with self.assertLogs(level='WARNING') as warnings: | 
| 190 |  | -                SwaggerCellParser(comment_prefix='#', notebook_cells=source_cells) | 
| 191 |  | -                for output in warnings.output: | 
| 192 |  | -                    self.assertRegex(output, 'postbar') | 
|  | 176 | +        """Parser should warn about documented operations that aren't referenced | 
|  | 177 | +        in a cell | 
|  | 178 | +        """ | 
|  | 179 | +        source_cells = [ | 
|  | 180 | +            {"source": '```\n{"swagger":"2.0", "paths": {"/foo": {"put": {"operationId":"putbar","parameters": [{"name": "bar"}]},"post":{"operationId":"postbar"},"get": {"operationId":"get","parameters": [{"name": "bar"}]}}}}\n```\n'}, | 
|  | 181 | +            {"source": '# operationId: get'}, | 
|  | 182 | +            {"source": '# operationId: putbar'}, | 
|  | 183 | +            {"source": '# operationId: putbar '} | 
|  | 184 | +        ] | 
|  | 185 | +        with self.assertLogs(level='WARNING') as warnings: | 
|  | 186 | +            SwaggerCellParser(comment_prefix='#', notebook_cells=source_cells) | 
|  | 187 | +            for output in warnings.output: | 
|  | 188 | +                self.assertRegex(output, 'postbar') | 
0 commit comments