add: tests

Author: sshaplygin

generate.go

@@ -17,8 +17,6 @@ func Generate(docType DocType) string {
 		callFunc = inn.Generate
 	case KPP:
 		callFunc = kpp.Generate
-	default:
-		panic("not implemented method")
 	}
 
 	if callFunc == nil {

generate_test.go

@@ -1,7 +1,21 @@
 package docs_code
 
-import "testing"
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
 
 func Test_Generate(t *testing.T) {
+	inn := Generate(INN)
+	isValid, err := Validate(INN, inn)
+	require.NoError(t, err)
+
+	require.True(t, isValid)
+}
 
+func Test_Generate_Unsupported(t *testing.T) {
+	require.Panics(t, func() {
+		Generate(DocType(100500))
+	})
 }

validate.go

@@ -17,8 +17,6 @@ func Validate(docType DocType, code string) (bool, error) {
 		callFunc = inn.Validate
 	case KPP:
 		callFunc = kpp.Validate
-	default:
-		panic("not implemented method")
 	}
 
 	if callFunc == nil {

validate_test.go

@@ -1,7 +1,20 @@
 package docs_code
 
-import "testing"
+import (
+	"testing"
+
+	"github.com/stretchr/testify/require"
+)
 
 func Test_Validate(t *testing.T) {
+	isValid, err := Validate(INN, "7707083893")
+	require.NoError(t, err)
+
+	require.True(t, isValid)
+}
 
+func Test_Validate_Unsupported(t *testing.T) {
+	require.Panics(t, func() {
+		Validate(DocType(100500), "100500")
+	})
 }