-
Notifications
You must be signed in to change notification settings - Fork 10
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* Update faker dependency * Add new types * Fix readme * Add valid Norwegian SSN * Add test for valid SSN * Remove number helpers * Prevent intermittent test failures by handling control digit of 10 * Enforce single digits from control digits
- Loading branch information
1 parent
1f2e375
commit d9e10c7
Showing
6 changed files
with
126 additions
and
1 deletion.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,68 @@ | ||
| package helpers | ||
|
|
||
| import ( | ||
| "fmt" | ||
| "math/rand" | ||
| "time" | ||
| ) | ||
|
|
||
| // generateControlDigits generates the control digits for a given Norwegian SSN | ||
| func generateControlDigits(ssn string) (int, int) { | ||
| // Weights for calculating control digits | ||
| var weights1 = []int{3, 7, 6, 1, 8, 9, 4, 5, 2} | ||
| var weights2 = []int{5, 4, 3, 2, 7, 6, 5, 4, 3, 2} | ||
|
|
||
| var sum1, sum2 int | ||
| for i, r := range ssn { | ||
| digit := int(r - '0') | ||
| if i < 9 { // First control digit calculation | ||
| sum1 += digit * weights1[i] | ||
| } | ||
| // Note: Second control digit calculation now includes the first control digit, but it's added outside this loop | ||
| if i < 10 { // Second control digit calculation | ||
| sum2 += digit * weights2[i] | ||
| } | ||
| } | ||
|
|
||
| // Calculate control digits | ||
| control1 := 11 - (sum1 % 11) | ||
| control2 := 11 - (sum2 % 11) | ||
|
|
||
| // Adjust for special cases and check for the value 10 | ||
| if control1 == 11 { | ||
| control1 = 0 | ||
| } else if control1 == 10 { | ||
| return -1, -1 // Indicate an error or invalid state | ||
| } | ||
|
|
||
| // Add first control digit to sum2 before calculating control2 | ||
| sum2 += control1 * weights2[9] | ||
| control2 = 11 - (sum2 % 11) | ||
|
|
||
| if control2 == 11 { | ||
| control2 = 0 | ||
| } else if control2 == 10 { | ||
| return -1, -1 // Indicate an error or invalid state | ||
| } | ||
|
|
||
| return control1, control2 | ||
| } | ||
|
|
||
| // generateFakeNorwegianSSN generates a fake (but valid) Norwegian SSN. | ||
| func generateFakeNorwegianSSN(dob time.Time) string { | ||
| year, month, day := dob.Date() | ||
| var ssn string | ||
| var control1, control2 int = -1, -1 | ||
|
|
||
| for control1 == -1 || control2 == -1 { | ||
| individualNumber := rand.Intn(500) | ||
| ssn = fmt.Sprintf("%02d%02d%02d%03d", day, month, year%100, individualNumber) | ||
|
|
||
| control1, control2 = generateControlDigits(ssn) | ||
| if control1 != -1 && control2 != -1 { | ||
| ssn += fmt.Sprintf("%01d%01d", control1, control2) | ||
| } | ||
| } | ||
|
|
||
| return ssn | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,29 @@ | ||
| package helpers | ||
|
|
||
| import ( | ||
| "testing" | ||
|
|
||
| "syreclabs.com/go/faker" | ||
| ) | ||
|
|
||
| func TestSSN(t *testing.T) { | ||
| ssn := generateFakeNorwegianSSN(faker.Date().Birthday(18, 90)) | ||
|
|
||
| if !isValidNorwegianSSN(ssn) { | ||
| t.Errorf("Generated SSN is invalid: %s", ssn) | ||
| } | ||
| } | ||
|
|
||
| // isValidNorwegianSSN checks if a given Norwegian SSN is valid (assumes the DOB is correct, as we generate that ourselves). | ||
| func isValidNorwegianSSN(ssn string) bool { | ||
| if len(ssn) != 11 { | ||
| return false | ||
| } | ||
|
|
||
| // Assuming generateControlDigits is implemented and returns control digits correctly | ||
| control1, control2 := generateControlDigits(ssn[:9]) | ||
| expectedControl1 := int(ssn[9] - '0') | ||
| expectedControl2 := int(ssn[10] - '0') | ||
|
|
||
| return control1 == expectedControl1 && control2 == expectedControl2 | ||
| } |