Merge pull request #460 from grongor/fix-unsafe-pointer

Fix unsafe pointer

pkg/fixtures/unsafe.go

@@ -0,0 +1,7 @@
+package test
+
+import "unsafe"
+
+type UnsafeInterface interface {
+	Do(ptr *unsafe.Pointer)
+}

pkg/generator.go

@@ -342,6 +342,9 @@ func (g *Generator) renderType(ctx context.Context, typ types.Type) string {
 		}
 		return g.addPackageImport(ctx, o.Pkg()) + "." + o.Name()
 	case *types.Basic:
+		if t.Kind() == types.UnsafePointer {
+			return "unsafe.Pointer"
+		}
 		return t.Name()
 	case *types.Pointer:
 		return "*" + g.renderType(ctx, t.Elem())

pkg/generator_test.go

@@ -1659,10 +1659,35 @@ func NewImportsSameAsPackage(t testing.TB) *ImportsSameAsPackage {
 	s.checkGeneration("imports_same_as_package.go", "ImportsSameAsPackage", false, "", expected)
 }
 
+func (s *GeneratorSuite) TestGeneratorWithUnsafePointer() {
+	expected := `// UnsafeInterface is an autogenerated mock type for the UnsafeInterface type
+type UnsafeInterface struct {
+	mock.Mock
+}
+
+// Do provides a mock function with given fields: ptr
+func (_m *UnsafeInterface) Do(ptr *unsafe.Pointer) {
+	_m.Called(ptr)
+}
+
+// NewUnsafeInterface creates a new instance of UnsafeInterface. It also registers the testing.TB interface on the mock and a cleanup function to assert the mocks expectations.
+func NewUnsafeInterface(t testing.TB) *UnsafeInterface {
+	mock := &UnsafeInterface{}
+	mock.Mock.Test(t)
+
+	t.Cleanup(func() { mock.AssertExpectations(t) })
+
+	return mock
+}
+`
+	s.checkGeneration("unsafe.go", "UnsafeInterface", false, "", expected)
+}
+
 func (s *GeneratorSuite) TestPrologueWithImportSameAsLocalPackage() {
 	generator := s.getGenerator(
 		"imports_same_as_package.go", "ImportsSameAsPackage", false, "",
 	)
+
 	expected := `package mocks
 
 import fixtures "` + generator.iface.QualifiedName + `"