File size: 699 Bytes
de67647
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import unittest
import jupyter_kernel_test

class MyKernelTests(jupyter_kernel_test.KernelTests):

    kernel_name = "codeql"
    language_name = "codeql"

    def test_codeql_kernel_register_database(self):
        self.flush_channels()
        reply, output_msgs = self.execute_helper(code='%set_database tests/codeql_db')
        self.assertEqual(output_msgs[0]["content"]["wait"], True)
        self.assertIn("Registering database ...", output_msgs[1]["content"]["data"]["text/plain"])
        self.assertEqual(output_msgs[2]["content"]["wait"], True)
        self.assertIn("Database registered!", output_msgs[3]["content"]["data"]["text/plain"])

if __name__ == '__main__':
    unittest.main()