Skip to content

Commit 2bef5f8

Browse files
author
B1nj0y
committed
add more unit tests for format_request function
1 parent aeef22e commit 2bef5f8

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

kernel_gateway/tests/notebook_http/test_request_utils.py

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -148,7 +148,17 @@ def test_format_request_without_a_kernel_language_arg(self):
148148
request_code = format_request(test_request)
149149
self.assertTrue(request_code.startswith("REQUEST"), "Call format_request without a kernel_language argument was not formatted correctly")
150150

151-
def test_format_request_with_a_kernel_language_arg(self):
151+
def test_format_request_with_a_kernel_language_python(self):
152+
test_request = ('''{"body": "", "headers": {}, "args": {}, "path": {}}''')
153+
request_code = format_request(test_request, 'python')
154+
self.assertTrue(request_code.startswith("REQUEST"), 'Call format_request without a kernel_language "python" was not formatted correctly')
155+
156+
def test_format_request_with_a_kernel_language_scala(self):
157+
test_request = ('''{"body": "", "headers": {}, "args": {}, "path": {}}''')
158+
request_code = format_request(test_request, 'scala')
159+
self.assertTrue(request_code.startswith("REQUEST"), 'Call format_request without a kernel_language "scala" was not formatted correctly')
160+
161+
def test_format_request_with_a_kernel_language_perl(self):
152162
test_request = ('''{"body": "", "headers": {}, "args": {}, "path": {}}''')
153163
request_code = format_request(test_request, 'perl')
154-
self.assertTrue(request_code.startswith("my $REQUEST"), "Call format_request with a kernel language argument was not formatted correctly")
164+
self.assertTrue(request_code.startswith("my $REQUEST"), 'Call format_request with a kernel language "perl" was not formatted correctly')

0 commit comments

Comments
 (0)