protoc_plugin/test/extension_generator_test.dart

// Copyright (c) 2015, the Dart project authors.  Please see the AUTHORS file
// for details. All rights reserved. Use of this source code is governed by a
// BSD-style license that can be found in the LICENSE file.

@TestOn('vm')
library;

import 'package:protoc_plugin/indenting_writer.dart';
import 'package:protoc_plugin/protoc.dart';
import 'package:protoc_plugin/src/gen/google/protobuf/compiler/plugin.pb.dart'
    as pb;
import 'package:protoc_plugin/src/gen/google/protobuf/descriptor.pb.dart' as pb;
import 'package:protoc_plugin/src/linker.dart';
import 'package:protoc_plugin/src/options.dart';
import 'package:test/test.dart';

import 'src/golden_file.dart';

void main() {
  test('testExtensionGenerator', () {
    final extensionFieldDescriptor =
        pb.FieldDescriptorProto()
          ..name = 'client_info'
          ..jsonName = 'clientInfo'
          ..number = 261486461
          ..label = pb.FieldDescriptorProto_Label.LABEL_OPTIONAL
          ..type = pb.FieldDescriptorProto_Type.TYPE_STRING
          ..extendee = '.Card';
    final messageDescriptor =
        pb.DescriptorProto()
          ..name = 'Card'
          ..extension.add(extensionFieldDescriptor);
    final fileDescriptor =
        pb.FileDescriptorProto()
          ..messageType.add(messageDescriptor)
          ..extension.add(extensionFieldDescriptor);

    final fileGenerator = FileGenerator(fileDescriptor, GenerationOptions());
    final options = parseGenerationOptions(
      pb.CodeGeneratorRequest(),
      pb.CodeGeneratorResponse(),
    );
    link(options, [fileGenerator]);
    final writer = IndentingWriter(
      generateMetadata: true,
      fileName: 'sample.proto',
    );
    fileGenerator.extensionGenerators.single.generate(writer);

    final actual = writer.emitSource(format: false);
    expectGolden(actual, 'extension.pb.dart');
    expectGolden(
      writer.sourceLocationInfo.toString(),
      'extension.pb.dart.meta',
    );
  });
}