Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
notifications-bot
Manage
Activity
Members
Labels
Code
Merge requests
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Deploy
Releases
Package registry
Model registry
Operate
Terraform modules
Analyze
Contributor analytics
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
GitLab community forum
Contribute to GitLab
Provide feedback
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
archives
notifications-bot
Commits
635e38f5
Commit
635e38f5
authored
Mar 24, 2023
by
Jonah Husson
Browse files
Options
Downloads
Patches
Plain Diff
Refactor testing code
parent
385c7102
No related branches found
No related tags found
2 merge requests
!38
Project/haven beta
,
!36
Update notifications bot to allow multiple identities & devices
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
storage/databaseImpl_test.go
+57
-284
57 additions, 284 deletions
storage/databaseImpl_test.go
with
57 additions
and
284 deletions
storage/databaseImpl_test.go
+
57
−
284
View file @
635e38f5
...
@@ -93,34 +93,14 @@ func TestDatabaseImpl_DeleteToken(t *testing.T) {
...
@@ -93,34 +93,14 @@ func TestDatabaseImpl_DeleteToken(t *testing.T) {
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
uid
,
err
:=
id
.
NewRandomID
(
csprng
.
NewSystemRNG
(),
id
.
User
)
identity
:=
generateTestIdentity
(
t
)
if
err
!=
nil
{
u
:=
generateTestUser
(
t
)
t
.
Fatal
(
err
)
}
iid
,
err
:=
ephemeral
.
GetIntermediaryId
(
uid
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
trsa
,
err
:=
rsa
.
GenerateKey
(
csprng
.
NewSystemRNG
(),
512
)
h
:=
hash
.
CMixHash
.
New
()
h
.
Write
(
trsa
.
GetPublic
()
.
Bytes
())
sig
:=
[]
byte
(
"fake signature"
)
u
:=
&
User
{
TransmissionRSAHash
:
h
.
Sum
(
nil
),
TransmissionRSA
:
trsa
.
GetPublic
()
.
Bytes
(),
Signature
:
sig
,
}
err
=
db
.
insertUser
(
u
)
err
=
db
.
insertUser
(
u
)
if
err
!=
nil
{
if
err
!=
nil
{
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
identity
:=
Identity
{
IntermediaryId
:
iid
,
OffsetNum
:
ephemeral
.
GetOffsetNum
(
ephemeral
.
GetOffset
(
iid
)),
}
err
=
db
.
insertIdentity
(
&
identity
)
err
=
db
.
insertIdentity
(
&
identity
)
if
err
!=
nil
{
if
err
!=
nil
{
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
...
@@ -162,15 +142,7 @@ func TestDatabaseImpl_insertUser(t *testing.T) {
...
@@ -162,15 +142,7 @@ func TestDatabaseImpl_insertUser(t *testing.T) {
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
trsa
,
err
:=
rsa
.
GenerateKey
(
csprng
.
NewSystemRNG
(),
512
)
u
:=
generateTestUser
(
t
)
h
:=
hash
.
CMixHash
.
New
()
h
.
Write
(
trsa
.
GetPublic
()
.
Bytes
())
sig
:=
[]
byte
(
"fake signature"
)
u
:=
&
User
{
TransmissionRSAHash
:
h
.
Sum
(
nil
),
TransmissionRSA
:
trsa
.
GetPublic
()
.
Bytes
(),
Signature
:
sig
,
}
err
=
db
.
insertUser
(
u
)
err
=
db
.
insertUser
(
u
)
if
err
!=
nil
{
if
err
!=
nil
{
...
@@ -205,15 +177,7 @@ func TestDatabaseImpl_GetUser(t *testing.T) {
...
@@ -205,15 +177,7 @@ func TestDatabaseImpl_GetUser(t *testing.T) {
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
trsa
,
err
:=
rsa
.
GenerateKey
(
csprng
.
NewSystemRNG
(),
512
)
u
:=
generateTestUser
(
t
)
h
:=
hash
.
CMixHash
.
New
()
h
.
Write
(
trsa
.
GetPublic
()
.
Bytes
())
sig
:=
[]
byte
(
"fake signature"
)
u
:=
&
User
{
TransmissionRSAHash
:
h
.
Sum
(
nil
),
TransmissionRSA
:
trsa
.
GetPublic
()
.
Bytes
(),
Signature
:
sig
,
}
receivedUser
,
err
:=
db
.
GetUser
(
u
.
TransmissionRSAHash
)
receivedUser
,
err
:=
db
.
GetUser
(
u
.
TransmissionRSAHash
)
if
err
==
nil
||
!
errors
.
Is
(
err
,
gorm
.
ErrRecordNotFound
)
{
if
err
==
nil
||
!
errors
.
Is
(
err
,
gorm
.
ErrRecordNotFound
)
{
...
@@ -240,15 +204,7 @@ func TestDatabaseImpl_deleteUser(t *testing.T) {
...
@@ -240,15 +204,7 @@ func TestDatabaseImpl_deleteUser(t *testing.T) {
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
trsa
,
err
:=
rsa
.
GenerateKey
(
csprng
.
NewSystemRNG
(),
512
)
u
:=
generateTestUser
(
t
)
h
:=
hash
.
CMixHash
.
New
()
h
.
Write
(
trsa
.
GetPublic
()
.
Bytes
())
sig
:=
[]
byte
(
"fake signature"
)
u
:=
&
User
{
TransmissionRSAHash
:
h
.
Sum
(
nil
),
TransmissionRSA
:
trsa
.
GetPublic
()
.
Bytes
(),
Signature
:
sig
,
}
receivedUser
,
err
:=
db
.
GetUser
(
u
.
TransmissionRSAHash
)
receivedUser
,
err
:=
db
.
GetUser
(
u
.
TransmissionRSAHash
)
if
err
==
nil
||
!
errors
.
Is
(
err
,
gorm
.
ErrRecordNotFound
)
{
if
err
==
nil
||
!
errors
.
Is
(
err
,
gorm
.
ErrRecordNotFound
)
{
...
@@ -295,19 +251,8 @@ func TestDatabaseImpl_GetAllUsers(t *testing.T) {
...
@@ -295,19 +251,8 @@ func TestDatabaseImpl_GetAllUsers(t *testing.T) {
expectedUsers
:=
5
expectedUsers
:=
5
for
i
:=
1
;
i
<=
expectedUsers
;
i
++
{
for
i
:=
1
;
i
<=
expectedUsers
;
i
++
{
newTrsa
,
err
:=
rsa
.
GenerateKey
(
csprng
.
NewSystemRNG
(),
512
)
u
:=
generateTestUser
(
t
)
if
err
!=
nil
{
err
=
db
.
insertUser
(
u
)
t
.
Fatal
(
err
)
}
newH
:=
hash
.
CMixHash
.
New
()
newH
.
Write
(
newTrsa
.
GetPublic
()
.
Bytes
())
sig
:=
[]
byte
(
"sig"
)
newU
:=
&
User
{
TransmissionRSAHash
:
newH
.
Sum
(
nil
),
TransmissionRSA
:
newTrsa
.
GetPublic
()
.
Bytes
(),
Signature
:
sig
,
}
err
=
db
.
insertUser
(
newU
)
if
err
!=
nil
{
if
err
!=
nil
{
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
...
@@ -329,19 +274,7 @@ func TestDatabaseImpl_getIdentity(t *testing.T) {
...
@@ -329,19 +274,7 @@ func TestDatabaseImpl_getIdentity(t *testing.T) {
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
uid
,
err
:=
id
.
NewRandomID
(
csprng
.
NewSystemRNG
(),
id
.
User
)
identity
:=
generateTestIdentity
(
t
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
iid
,
err
:=
ephemeral
.
GetIntermediaryId
(
uid
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
identity
:=
Identity
{
IntermediaryId
:
iid
,
OffsetNum
:
ephemeral
.
GetOffsetNum
(
ephemeral
.
GetOffset
(
iid
)),
}
err
=
db
.
insertIdentity
(
&
identity
)
err
=
db
.
insertIdentity
(
&
identity
)
if
err
!=
nil
{
if
err
!=
nil
{
...
@@ -363,19 +296,7 @@ func TestDatabaseImpl_getIdentitiesByOffset(t *testing.T) {
...
@@ -363,19 +296,7 @@ func TestDatabaseImpl_getIdentitiesByOffset(t *testing.T) {
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
uid
,
err
:=
id
.
NewRandomID
(
csprng
.
NewSystemRNG
(),
id
.
User
)
identity
:=
generateTestIdentity
(
t
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
iid
,
err
:=
ephemeral
.
GetIntermediaryId
(
uid
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
identity
:=
Identity
{
IntermediaryId
:
iid
,
OffsetNum
:
ephemeral
.
GetOffsetNum
(
ephemeral
.
GetOffset
(
iid
)),
}
err
=
db
.
insertIdentity
(
&
identity
)
err
=
db
.
insertIdentity
(
&
identity
)
if
err
!=
nil
{
if
err
!=
nil
{
...
@@ -406,19 +327,7 @@ func TestDatabaseImpl_GetOrphanedIdentities(t *testing.T) {
...
@@ -406,19 +327,7 @@ func TestDatabaseImpl_GetOrphanedIdentities(t *testing.T) {
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
uid
,
err
:=
id
.
NewRandomID
(
csprng
.
NewSystemRNG
(),
id
.
User
)
identity
:=
generateTestIdentity
(
t
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
iid
,
err
:=
ephemeral
.
GetIntermediaryId
(
uid
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
identity
:=
Identity
{
IntermediaryId
:
iid
,
OffsetNum
:
ephemeral
.
GetOffsetNum
(
ephemeral
.
GetOffset
(
iid
)),
}
err
=
db
.
insertIdentity
(
&
identity
)
err
=
db
.
insertIdentity
(
&
identity
)
if
err
!=
nil
{
if
err
!=
nil
{
...
@@ -433,18 +342,7 @@ func TestDatabaseImpl_GetOrphanedIdentities(t *testing.T) {
...
@@ -433,18 +342,7 @@ func TestDatabaseImpl_GetOrphanedIdentities(t *testing.T) {
t
.
Fatalf
(
"Did not receive expected count of orphaned identities
\n\t
Expected: %+v
\n\t
Received: %+v
\n
"
,
1
,
len
(
orphaned
))
t
.
Fatalf
(
"Did not receive expected count of orphaned identities
\n\t
Expected: %+v
\n\t
Received: %+v
\n
"
,
1
,
len
(
orphaned
))
}
}
uid2
,
err
:=
id
.
NewRandomID
(
csprng
.
NewSystemRNG
(),
id
.
User
)
identity2
:=
generateTestIdentity
(
t
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
iid2
,
err
:=
ephemeral
.
GetIntermediaryId
(
uid2
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
identity2
:=
Identity
{
IntermediaryId
:
iid2
,
OffsetNum
:
ephemeral
.
GetOffsetNum
(
ephemeral
.
GetOffset
(
iid2
)),
}
err
=
db
.
insertIdentity
(
&
identity2
)
err
=
db
.
insertIdentity
(
&
identity2
)
if
err
!=
nil
{
if
err
!=
nil
{
...
@@ -485,19 +383,7 @@ func TestDatabaseImpl_insertEphemeral(t *testing.T) {
...
@@ -485,19 +383,7 @@ func TestDatabaseImpl_insertEphemeral(t *testing.T) {
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
uid
,
err
:=
id
.
NewRandomID
(
csprng
.
NewSystemRNG
(),
id
.
User
)
identity
:=
generateTestIdentity
(
t
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
iid
,
err
:=
ephemeral
.
GetIntermediaryId
(
uid
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
identity
:=
Identity
{
IntermediaryId
:
iid
,
OffsetNum
:
ephemeral
.
GetOffsetNum
(
ephemeral
.
GetOffset
(
iid
)),
}
e1
:=
&
Ephemeral
{
e1
:=
&
Ephemeral
{
ID
:
0
,
ID
:
0
,
...
@@ -530,19 +416,7 @@ func TestDatabaseImpl_GetEphemeral(t *testing.T) {
...
@@ -530,19 +416,7 @@ func TestDatabaseImpl_GetEphemeral(t *testing.T) {
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
uid
,
err
:=
id
.
NewRandomID
(
csprng
.
NewSystemRNG
(),
id
.
User
)
identity
:=
generateTestIdentity
(
t
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
iid
,
err
:=
ephemeral
.
GetIntermediaryId
(
uid
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
identity
:=
Identity
{
IntermediaryId
:
iid
,
OffsetNum
:
ephemeral
.
GetOffsetNum
(
ephemeral
.
GetOffset
(
iid
)),
}
e1
:=
&
Ephemeral
{
e1
:=
&
Ephemeral
{
ID
:
0
,
ID
:
0
,
...
@@ -585,19 +459,7 @@ func TestDatabaseImpl_DeleteOldEphemerals(t *testing.T) {
...
@@ -585,19 +459,7 @@ func TestDatabaseImpl_DeleteOldEphemerals(t *testing.T) {
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
uid
,
err
:=
id
.
NewRandomID
(
csprng
.
NewSystemRNG
(),
id
.
User
)
identity
:=
generateTestIdentity
(
t
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
iid
,
err
:=
ephemeral
.
GetIntermediaryId
(
uid
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
identity
:=
Identity
{
IntermediaryId
:
iid
,
OffsetNum
:
ephemeral
.
GetOffsetNum
(
ephemeral
.
GetOffset
(
iid
)),
}
e1
:=
&
Ephemeral
{
e1
:=
&
Ephemeral
{
ID
:
0
,
ID
:
0
,
...
@@ -658,19 +520,7 @@ func TestDatabaseImpl_GetLatestEphemeral(t *testing.T) {
...
@@ -658,19 +520,7 @@ func TestDatabaseImpl_GetLatestEphemeral(t *testing.T) {
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
uid
,
err
:=
id
.
NewRandomID
(
csprng
.
NewSystemRNG
(),
id
.
User
)
identity
:=
generateTestIdentity
(
t
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
iid
,
err
:=
ephemeral
.
GetIntermediaryId
(
uid
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
identity
:=
Identity
{
IntermediaryId
:
iid
,
OffsetNum
:
ephemeral
.
GetOffsetNum
(
ephemeral
.
GetOffset
(
iid
)),
}
e1
:=
&
Ephemeral
{
e1
:=
&
Ephemeral
{
IntermediaryId
:
identity
.
IntermediaryId
,
IntermediaryId
:
identity
.
IntermediaryId
,
...
@@ -784,34 +634,14 @@ func TestDatabaseImpl_registerForNotifications(t *testing.T) {
...
@@ -784,34 +634,14 @@ func TestDatabaseImpl_registerForNotifications(t *testing.T) {
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
uid
,
err
:=
id
.
NewRandomID
(
csprng
.
NewSystemRNG
(),
id
.
User
)
identity
:=
generateTestIdentity
(
t
)
if
err
!=
nil
{
u
:=
generateTestUser
(
t
)
t
.
Fatal
(
err
)
}
iid
,
err
:=
ephemeral
.
GetIntermediaryId
(
uid
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
trsa
,
err
:=
rsa
.
GenerateKey
(
csprng
.
NewSystemRNG
(),
512
)
h
:=
hash
.
CMixHash
.
New
()
h
.
Write
(
trsa
.
GetPublic
()
.
Bytes
())
sig
:=
[]
byte
(
"fake signature"
)
u
:=
&
User
{
TransmissionRSAHash
:
h
.
Sum
(
nil
),
TransmissionRSA
:
trsa
.
GetPublic
()
.
Bytes
(),
Signature
:
sig
,
}
err
=
db
.
insertUser
(
u
)
err
=
db
.
insertUser
(
u
)
if
err
!=
nil
{
if
err
!=
nil
{
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
identity
:=
Identity
{
IntermediaryId
:
iid
,
OffsetNum
:
ephemeral
.
GetOffsetNum
(
ephemeral
.
GetOffset
(
iid
)),
}
err
=
db
.
insertIdentity
(
&
identity
)
err
=
db
.
insertIdentity
(
&
identity
)
if
err
!=
nil
{
if
err
!=
nil
{
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
...
@@ -831,33 +661,14 @@ func TestDatabaseImpl_registerForNotifications(t *testing.T) {
...
@@ -831,33 +661,14 @@ func TestDatabaseImpl_registerForNotifications(t *testing.T) {
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
uid2
,
err
:=
id
.
NewRandomID
(
csprng
.
NewSystemRNG
(),
id
.
User
)
identity2
:=
generateTestIdentity
(
t
)
if
err
!=
nil
{
u2
:=
generateTestUser
(
t
)
t
.
Fatal
(
err
)
}
iid2
,
err
:=
ephemeral
.
GetIntermediaryId
(
uid2
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
trsa2
,
err
:=
rsa
.
GenerateKey
(
csprng
.
NewSystemRNG
(),
512
)
h
.
Reset
()
h
.
Write
(
trsa2
.
GetPublic
()
.
Bytes
())
u2
:=
&
User
{
TransmissionRSAHash
:
h
.
Sum
(
nil
),
TransmissionRSA
:
trsa2
.
GetPublic
()
.
Bytes
(),
Signature
:
sig
,
}
err
=
db
.
insertUser
(
u2
)
err
=
db
.
insertUser
(
u2
)
if
err
!=
nil
{
if
err
!=
nil
{
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
identity2
:=
Identity
{
IntermediaryId
:
iid2
,
OffsetNum
:
ephemeral
.
GetOffsetNum
(
ephemeral
.
GetOffset
(
iid2
)),
}
err
=
db
.
insertIdentity
(
&
identity
)
err
=
db
.
insertIdentity
(
&
identity
)
if
err
!=
nil
{
if
err
!=
nil
{
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
...
@@ -920,34 +731,14 @@ func TestDatabaseImpl_unregisterIdentities(t *testing.T) {
...
@@ -920,34 +731,14 @@ func TestDatabaseImpl_unregisterIdentities(t *testing.T) {
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
uid
,
err
:=
id
.
NewRandomID
(
csprng
.
NewSystemRNG
(),
id
.
User
)
u
:=
generateTestUser
(
t
)
if
err
!=
nil
{
identity
:=
generateTestIdentity
(
t
)
t
.
Fatal
(
err
)
}
iid
,
err
:=
ephemeral
.
GetIntermediaryId
(
uid
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
trsa
,
err
:=
rsa
.
GenerateKey
(
csprng
.
NewSystemRNG
(),
512
)
h
:=
hash
.
CMixHash
.
New
()
h
.
Write
(
trsa
.
GetPublic
()
.
Bytes
())
sig
:=
[]
byte
(
"fake signature"
)
u
:=
&
User
{
TransmissionRSAHash
:
h
.
Sum
(
nil
),
TransmissionRSA
:
trsa
.
GetPublic
()
.
Bytes
(),
Signature
:
sig
,
}
err
=
db
.
insertUser
(
u
)
err
=
db
.
insertUser
(
u
)
if
err
!=
nil
{
if
err
!=
nil
{
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
identity
:=
Identity
{
IntermediaryId
:
iid
,
OffsetNum
:
ephemeral
.
GetOffsetNum
(
ephemeral
.
GetOffset
(
iid
)),
}
err
=
db
.
unregisterIdentities
(
u
,
[]
Identity
{
identity
})
err
=
db
.
unregisterIdentities
(
u
,
[]
Identity
{
identity
})
if
err
!=
nil
{
if
err
!=
nil
{
t
.
Fatalf
(
"Should not return error even if identity doesn't exist: %+v"
,
err
)
t
.
Fatalf
(
"Should not return error even if identity doesn't exist: %+v"
,
err
)
...
@@ -969,24 +760,7 @@ func TestDatabaseImpl_unregisterIdentities(t *testing.T) {
...
@@ -969,24 +760,7 @@ func TestDatabaseImpl_unregisterIdentities(t *testing.T) {
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
uid2
,
err
:=
id
.
NewRandomID
(
csprng
.
NewSystemRNG
(),
id
.
User
)
identity2
:=
generateTestIdentity
(
t
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
iid2
,
err
:=
ephemeral
.
GetIntermediaryId
(
uid2
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
identity2
:=
Identity
{
IntermediaryId
:
iid2
,
OffsetNum
:
ephemeral
.
GetOffsetNum
(
ephemeral
.
GetOffset
(
iid2
)),
}
err
=
db
.
insertIdentity
(
&
identity
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
token2
:=
"fcm:token2"
token2
:=
"fcm:token2"
err
=
db
.
registerForNotifications
(
u
,
identity2
,
token2
)
err
=
db
.
registerForNotifications
(
u
,
identity2
,
token2
)
...
@@ -1022,34 +796,14 @@ func TestDatabaseImpl_unregisterTokens(t *testing.T) {
...
@@ -1022,34 +796,14 @@ func TestDatabaseImpl_unregisterTokens(t *testing.T) {
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
uid
,
err
:=
id
.
NewRandomID
(
csprng
.
NewSystemRNG
(),
id
.
User
)
identity
:=
generateTestIdentity
(
t
)
if
err
!=
nil
{
u
:=
generateTestUser
(
t
)
t
.
Fatal
(
err
)
}
iid
,
err
:=
ephemeral
.
GetIntermediaryId
(
uid
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
trsa
,
err
:=
rsa
.
GenerateKey
(
csprng
.
NewSystemRNG
(),
512
)
h
:=
hash
.
CMixHash
.
New
()
h
.
Write
(
trsa
.
GetPublic
()
.
Bytes
())
sig
:=
[]
byte
(
"fake signature"
)
u
:=
&
User
{
TransmissionRSAHash
:
h
.
Sum
(
nil
),
TransmissionRSA
:
trsa
.
GetPublic
()
.
Bytes
(),
Signature
:
sig
,
}
err
=
db
.
insertUser
(
u
)
err
=
db
.
insertUser
(
u
)
if
err
!=
nil
{
if
err
!=
nil
{
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
identity
:=
Identity
{
IntermediaryId
:
iid
,
OffsetNum
:
ephemeral
.
GetOffsetNum
(
ephemeral
.
GetOffset
(
iid
)),
}
token
:=
"apnstoken02"
token
:=
"apnstoken02"
err
=
db
.
unregisterTokens
(
u
,
[]
Token
{
Token
{
Token
:
token
}})
err
=
db
.
unregisterTokens
(
u
,
[]
Token
{
Token
{
Token
:
token
}})
if
err
!=
nil
{
if
err
!=
nil
{
...
@@ -1076,19 +830,7 @@ func TestDatabaseImpl_unregisterTokens(t *testing.T) {
...
@@ -1076,19 +830,7 @@ func TestDatabaseImpl_unregisterTokens(t *testing.T) {
t
.
Fatal
(
err
)
t
.
Fatal
(
err
)
}
}
uid2
,
err
:=
id
.
NewRandomID
(
csprng
.
NewSystemRNG
(),
id
.
User
)
identity2
:=
generateTestIdentity
(
t
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
iid2
,
err
:=
ephemeral
.
GetIntermediaryId
(
uid2
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
identity2
:=
Identity
{
IntermediaryId
:
iid2
,
OffsetNum
:
ephemeral
.
GetOffsetNum
(
ephemeral
.
GetOffset
(
iid2
)),
}
err
=
db
.
insertIdentity
(
&
identity
)
err
=
db
.
insertIdentity
(
&
identity
)
if
err
!=
nil
{
if
err
!=
nil
{
...
@@ -1130,7 +872,7 @@ func TestDatabaseImpl_unregisterTokens(t *testing.T) {
...
@@ -1130,7 +872,7 @@ func TestDatabaseImpl_unregisterTokens(t *testing.T) {
t
.
Fatalf
(
"Expected gorm.ErrRecordNotFound when no user exists, instead got %+v"
,
err
)
t
.
Fatalf
(
"Expected gorm.ErrRecordNotFound when no user exists, instead got %+v"
,
err
)
}
}
_
,
err
=
db
.
getIdentity
(
i
i
d
)
_
,
err
=
db
.
getIdentity
(
i
dentity
.
IntermediaryI
d
)
if
err
!=
nil
{
if
err
!=
nil
{
t
.
Fatalf
(
"Failed to get identity: %+v"
,
err
)
t
.
Fatalf
(
"Failed to get identity: %+v"
,
err
)
}
}
...
@@ -1142,3 +884,34 @@ func TestDatabaseImpl_unregisterTokens(t *testing.T) {
...
@@ -1142,3 +884,34 @@ func TestDatabaseImpl_unregisterTokens(t *testing.T) {
// t.Fatal(err)
// t.Fatal(err)
// }
// }
//}
//}
func
generateTestIdentity
(
t
*
testing
.
T
)
Identity
{
uid
,
err
:=
id
.
NewRandomID
(
csprng
.
NewSystemRNG
(),
id
.
User
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
iid
,
err
:=
ephemeral
.
GetIntermediaryId
(
uid
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
identity
:=
Identity
{
IntermediaryId
:
iid
,
OffsetNum
:
ephemeral
.
GetOffsetNum
(
ephemeral
.
GetOffset
(
iid
)),
}
return
identity
}
func
generateTestUser
(
t
*
testing
.
T
)
*
User
{
trsa
,
err
:=
rsa
.
GenerateKey
(
csprng
.
NewSystemRNG
(),
512
)
if
err
!=
nil
{
t
.
Fatal
(
err
)
}
h
:=
hash
.
CMixHash
.
New
()
h
.
Write
(
trsa
.
GetPublic
()
.
Bytes
())
sig
:=
[]
byte
(
"fake signature"
)
u
:=
&
User
{
TransmissionRSAHash
:
h
.
Sum
(
nil
),
TransmissionRSA
:
trsa
.
GetPublic
()
.
Bytes
(),
Signature
:
sig
,
}
return
u
}
This diff is collapsed.
Click to expand it.
Preview
0%
Loading
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment