add: ogrn and ogrnip

Author: sshaplygin

bik/bik_test.go

@@ -102,11 +102,13 @@ func TestValidate(t *testing.T) {
 }
 
 func Test_Generate(t *testing.T) {
-	bik := Generate()
-	isValid, err := Validate(bik)
-	require.NoError(t, err, fmt.Sprintf("invalid bik value: %s", bik))
+	for i := 0; i < 10; i++ {
+		bik := Generate()
+		isValid, err := Validate(bik)
+		require.NoError(t, err, fmt.Sprintf("invalid bik value: %s", bik))
 
-	assert.True(t, isValid)
+		assert.True(t, isValid)
+	}
 }
 
 func Test_Exists(t *testing.T) {

fts/fts.go

@@ -53,19 +53,8 @@ func (trc *TaxRegionCode) Ints() []int {
 
 	res := make([]int, subjectCodeLength+regionTaxServiceNumberLength)
 
-	nums := utils.CodeToInts(int(trc.subjectCode))
-	idx := subjectCodeLength - 1
-	for i := len(nums) - 1; i >= 0; i-- {
-		res[idx] = nums[i]
-		idx--
-	}
-
-	nums = utils.CodeToInts(int(trc.serviceNumber))
-	idx = len(res) - 1
-	for i := len(nums) - 1; i >= 0; i-- {
-		res[idx] = nums[i]
-		idx--
-	}
+	utils.FillSlice(utils.CodeToInts(int(trc.subjectCode)), res, subjectCodeLength-1)
+	utils.FillSlice(utils.CodeToInts(int(trc.serviceNumber)), res, len(res)-1)
 
 	return res
 }
@@ -154,6 +143,12 @@ func (csc ConstitutionRegionCode) GetName() string {
 	return codeName
 }
 
+func (csc ConstitutionRegionCode) Ints() []int {
+	res := make([]int, subjectCodeLength)
+	utils.FillSlice(utils.CodeToInts(int(csc)), res, len(res)-1)
+	return res
+}
+
 func GenerateConstitutionSubjectCode() ConstitutionRegionCode {
 	return regionsCodes[utils.Random(0, len(regionsCodes)-1)]
 }

generate.go

@@ -4,6 +4,8 @@ import (
 	"github.com/sshaplygin/docs-code/bik"
 	"github.com/sshaplygin/docs-code/inn"
 	"github.com/sshaplygin/docs-code/kpp"
+	"github.com/sshaplygin/docs-code/ogrn"
+	"github.com/sshaplygin/docs-code/ogrnip"
 )
 
 type GenerateFunc func() string
@@ -17,6 +19,10 @@ func Generate(docType DocType) string {
 		callFunc = inn.Generate
 	case KPP:
 		callFunc = kpp.Generate
+	case OGRN:
+		callFunc = ogrn.Generate
+	case OGRNIP:
+		callFunc = ogrnip.Generate
 	}
 
 	if callFunc == nil {

inn/models.go

@@ -44,42 +44,36 @@ const (
 )
 
 type SerialNumber struct {
-	val  int
-	size int
+	val int
+	len int
 }
 
 func (sn SerialNumber) String() string {
-	return utils.StrCode(sn.val, sn.size)
+	return utils.StrCode(sn.val, sn.len)
 }
 
 func (sn *SerialNumber) Ints() []int {
 	if sn == nil {
 		return nil
 	}
 
-	res := make([]int, sn.size)
-	nums := utils.CodeToInts(sn.val)
-
-	idx := len(res) - 1
-	for i := len(nums) - 1; i >= 0; i-- {
-		res[idx] = nums[i]
-		idx--
-	}
+	res := make([]int, sn.len)
+	utils.FillSlice(utils.CodeToInts(sn.val), res, len(res)-1)
 
 	return res
 }
 
 func GenerateSerailNumber(innType INNType) SerialNumber {
 	if innType == Physical {
 		return SerialNumber{
-			val:  int(utils.RandomDigits(physicalSerialNumberLength)),
-			size: physicalSerialNumberLength,
+			val: int(utils.RandomDigits(physicalSerialNumberLength)),
+			len: physicalSerialNumberLength,
 		}
 	}
 
 	return SerialNumber{
-		val:  int(utils.RandomDigits(legalSerialNumberLength)),
-		size: legalSerialNumberLength,
+		val: int(utils.RandomDigits(legalSerialNumberLength)),
+		len: legalSerialNumberLength,
 	}
 }
 
@@ -148,10 +142,10 @@ func ParseINN(inn string) (*INNStruct, error) {
 	}
 
 	t := Physical
-	snSize := physicalSerialNumberLength
+	snlen := physicalSerialNumberLength
 	parseIdx := len(inn) - 2
 	if len(inn) == legalLength {
-		snSize = legalSerialNumberLength
+		snlen = legalSerialNumberLength
 		t = Legal
 		parseIdx = len(inn) - 1
 		const foreignLegalStartWith = "9909"
@@ -173,8 +167,8 @@ func ParseINN(inn string) (*INNStruct, error) {
 	return &INNStruct{
 		taxRegionCode: taxRegionCode,
 		serialNumber: SerialNumber{
-			val:  utils.SliceToInt(serialNumberArr),
-			size: snSize,
+			val: utils.SliceToInt(serialNumberArr),
+			len: snlen,
 		},
 		checkSums: checkSums,
 		t:         t,

kpp/errors.go

@@ -12,6 +12,6 @@ var (
 	// ErrInvalidReasonCode invalid reason code
 	ErrInvalidReasonCode = errors.New("invalid reason code")
 
-	// ErrInvalidSerialNumber invalid serial number
-	ErrInvalidSerialNumber = errors.New("invalid serial number")
+	// ErrInvalidSerialNumbers invalid serial number
+	ErrInvalidSerialNumbers = errors.New("invalid serial number")
 )

kpp/kpp_test.go

@@ -130,11 +130,13 @@ func TestValidate(t *testing.T) {
 }
 
 func Test_Generate(t *testing.T) {
-	kpp := Generate()
-	isValid, err := Validate(kpp)
-	require.NoError(t, err, fmt.Sprintf("invalid kpp value: %s", kpp))
+	for i := 0; i < 10; i++ {
+		kpp := Generate()
+		isValid, err := Validate(kpp)
+		require.NoError(t, err, fmt.Sprintf("invalid kpp value: %s", kpp))
 
-	assert.True(t, isValid)
+		assert.True(t, isValid)
+	}
 }
 
 func BenchmarkValidateCorrect(b *testing.B) {

kpp/models.go

@@ -116,7 +116,7 @@ func (kpp *KPPStruct) IsValid() (bool, error) {
 	}
 
 	if !kpp.serialNumber.IsValid() {
-		return false, fmt.Errorf(validateErrorTmpl, ErrInvalidSerialNumber, kpp.serialNumber)
+		return false, fmt.Errorf(validateErrorTmpl, ErrInvalidSerialNumbers, kpp.serialNumber)
 	}
 
 	return true, nil

ogrn/errors.go

@@ -0,0 +1,20 @@
+package ogrn
+
+import "errors"
+
+var (
+	// ErrNilOGRN try call methods for nil ogrn struct
+	ErrNilOGRN = errors.New("nil ogrn struct")
+
+	// ErrInvalidCodeType invalid code type
+	ErrInvalidCodeType = errors.New("invalid code type")
+
+	// ErrInvalidYearsNumbers invalid years number code
+	ErrInvalidYearsNumbers = errors.New("invalid years number code")
+
+	// ErrInvalidRegion invalid region code
+	ErrInvalidRegion = errors.New("invalid region code")
+
+	// ErrInvalidSerialNumbers invalid serial numbers
+	ErrInvalidSerialNumbers = errors.New("invalid serial numbers")
+)