Code compiles, but tests fail

src/bin/protoc-gen-roto.rs

@@ -1,10 +1,9 @@
 use env_logger::init;
 use log::{error, info};
 use roto::generator::generate_rust_code;
-use roto::google::protobuf::descriptor::{
-    FileDescriptorSet
+use roto::proto_gen::google::protobuf::descriptor::{
+    CodeGeneratorRequest, CodeGeneratorResponse, FileDescriptorSet, ResponseFile,
 };
-use roto::google::protobuf::compiler::plugin::{CodeGeneratorRequest, CodeGeneratorResponse};
 use roto::ProtoBuilder;
 use std::io::{self, Read, Write};
 

src/generator.rs

@@ -1,8 +1,8 @@
 
-use crate::google::protobuf::descriptor::{
-    FileDescriptorSet, FieldDescriptorProto, DescriptorProto, EnumDescriptorProto, FileDescriptorProto
+
+use crate::proto_gen::google::protobuf::descriptor::{
+    CodeGeneratorRequest, CodeGeneratorResponse, FileDescriptorSet, ResponseFile, DescriptorProto, EnumDescriptorProto, FileDescriptorProto, FieldDescriptorProto
 };
-use crate::google::protobuf::compiler::plugin::{CodeGeneratorRequest, CodeGeneratorResponse};
 use crate::{ProtoAccessor, Result, RotoError};
 use std::str;
 use std::collections::{HashMap, HashSet};

src/lib.rs

@@ -1,6 +1,6 @@
 pub mod proto_gen;
 pub mod generator;
-pub mod google;
+// pub mod google;
 // Uncomment this to check if the code compiles
 // #[path = "../proto/google/protobuf/descriptor.rs"]
 // pub mod descriptor;