move the mockgen cleanup commands to the mockgen script

quic-go · Aug 11, 2018 · 76d4d43 · 76d4d43
1 parent b291b33
commit 76d4d43
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/mockgen.go b/mockgen.go
@@ -17,6 +17,3 @@ package quic
 //go:generate sh -c "./mockgen_private.sh quic mock_unknown_packet_handler_test.go github.com/lucas-clemente/quic-go unknownPacketHandler"
 //go:generate sh -c "./mockgen_private.sh quic mock_packet_handler_manager_test.go github.com/lucas-clemente/quic-go packetHandlerManager"
 //go:generate sh -c "./mockgen_private.sh quic mock_multiplexer_test.go github.com/lucas-clemente/quic-go multiplexer"
-
-//go:generate sh -c "find . -type f -name 'mock_*_test.go' | xargs sed -i '' 's/quic_go.//g'"
-//go:generate sh -c "goimports -w mock*_test.go"
diff --git a/mockgen_private.sh b/mockgen_private.sh
@@ -21,4 +21,8 @@ export GOPATH="$TEMP_DIR:$GOPATH"
 
 mockgen -package $1 -self_package $1 -destination $2 $3 $INTERFACE_NAME
 
+# mockgen imports quic-go as 'import quic_go github.com/lucas_clemente/quic-go'
+sed -i '' 's/quic_go.//g' $2
+goimports -w $2
+
 rm -r "$TEMP_DIR"