mirror of
				https://github.com/open-metadata/OpenMetadata.git
				synced 2025-11-04 12:36:23 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			61 lines
		
	
	
		
			2.0 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			61 lines
		
	
	
		
			2.0 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
#  Copyright 2021 Collate
 | 
						|
#  Licensed under the Apache License, Version 2.0 (the "License");
 | 
						|
#  you may not use this file except in compliance with the License.
 | 
						|
#  You may obtain a copy of the License at
 | 
						|
#  http://www.apache.org/licenses/LICENSE-2.0
 | 
						|
#  Unless required by applicable law or agreed to in writing, software
 | 
						|
#  distributed under the License is distributed on an "AS IS" BASIS,
 | 
						|
#  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
						|
#  See the License for the specific language governing permissions and
 | 
						|
#  limitations under the License.
 | 
						|
 | 
						|
"""
 | 
						|
Test GitHub Reader
 | 
						|
"""
 | 
						|
from unittest import TestCase
 | 
						|
 | 
						|
from metadata.generated.schema.security.credentials.bitbucketCredentials import (
 | 
						|
    BitBucketCredentials,
 | 
						|
)
 | 
						|
from metadata.ingestion.source.dashboard.looker.models import Includes, ViewName
 | 
						|
from metadata.ingestion.source.dashboard.looker.parser import LkmlParser
 | 
						|
from metadata.readers.file.bitbucket import BitBucketReader
 | 
						|
 | 
						|
 | 
						|
class TestLookMLBitBucketReader(TestCase):
 | 
						|
    """
 | 
						|
    Validate the github reader against the OM repo
 | 
						|
    """
 | 
						|
 | 
						|
    creds = BitBucketCredentials(
 | 
						|
        repositoryName="api",
 | 
						|
        repositoryOwner="pmbrull-trial-api",
 | 
						|
        branch="main",
 | 
						|
    )
 | 
						|
 | 
						|
    reader = BitBucketReader(creds)
 | 
						|
    parser = LkmlParser(reader)
 | 
						|
 | 
						|
    def test_lookml_read_and_parse(self):
 | 
						|
        """
 | 
						|
        We can parse the explore file.
 | 
						|
 | 
						|
        We'll expand and find views from https://bitbucket.org/pmbrull-trial-api/api/src/main
 | 
						|
        """
 | 
						|
 | 
						|
        explore_file = "cats.explore.lkml"
 | 
						|
        self.parser.parse_file(Includes(explore_file))
 | 
						|
 | 
						|
        contents = self.parser.parsed_files.get(Includes(explore_file))
 | 
						|
 | 
						|
        # Check file contents
 | 
						|
        self.assertIn("explore: cats", contents)
 | 
						|
 | 
						|
        view = self.parser.find_view(
 | 
						|
            view_name=ViewName("cats"), path=Includes(explore_file)
 | 
						|
        )
 | 
						|
 | 
						|
        # We can get views that are resolved even if the include does not contain `.lkml`
 | 
						|
        self.assertIsNotNone(view)
 | 
						|
        self.assertEqual(view.name, "cats")
 |