def test_for_for_asset_type(self): APISchema._discovey_documents_map = { 'compute': [{ 'id': 'compute.v1', 'schemas': { 'Instance': { 'properties': { 'property-1': { 'type': 'string', 'description': 'description-1.' } } } } }] } schema = APISchema.bigquery_schema_for_asset_type( 'google.compute.Instance', True, True) self.assertEqual(len(schema), 4)
def element_to_schema(self, element): return APISchema.bigquery_schema_for_asset_type( element['asset_type'], 'resource' in element and 'data' in element['resource'], 'iam_policy' in element)