Class Test_RubyParser

tests/test_parser/test_ruby.py:10–143

class Test_RubyParser(unittest.TestCase):
    def setUp(self) -> None:
        with open('tests/test_parser/test_sample/ruby_test_sample.rb', 'r') as file:
            self.code_sample = file.read()
            
        tree = parse_code(self.code_sample, 'ruby')
        self.root_node = tree.root_node

        return super().setUp()

    def test_get_function_list(self):
        root = self.root_node
        
        function_list = RubyParser.get_function_list(root)
        
        self.assertEqual(len(function_list), 2)

    def test_get_class_list(self):
        root = self.root_node
        
        class_list = RubyParser.get_class_list(root)
        
        self.assertEqual(len(class_list), 3)

    def test_get_docstring(self):
        code_sample = """
        module Encryption

            # Search for links.
            #
            # @param query [String] The search query.
            # @option options [String, RedditKit::Subreddit] subreddit The optional subreddit to search.
            def encrypt(string)
                Digest::SHA2.hexdigest(string)
            end
        end
           
        =begin 
        comment line 1
        comment line 2
        =end  
        class Orange
            def initialize
                @juice_available = 100
            end
            def squeeze
                @juice_available -= 50
            end
        end

        orange = Orange.new
        orange.squeeze
        """

        tree = parse_code(code_sample, 'ruby')
        root = tree.root_node
        
        fn = RubyParser.get_function_list(root)[0]