2018-05-19 17:12:37 +03:00
package models
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/tstranex/u2f"
)
func TestGetU2FRegistrationByID ( t * testing . T ) {
assert . NoError ( t , PrepareTestDatabase ( ) )
res , err := GetU2FRegistrationByID ( 1 )
assert . NoError ( t , err )
assert . Equal ( t , "U2F Key" , res . Name )
_ , err = GetU2FRegistrationByID ( 342432 )
assert . Error ( t , err )
assert . True ( t , IsErrU2FRegistrationNotExist ( err ) )
}
func TestGetU2FRegistrationsByUID ( t * testing . T ) {
assert . NoError ( t , PrepareTestDatabase ( ) )
res , err := GetU2FRegistrationsByUID ( 1 )
assert . NoError ( t , err )
assert . Len ( t , res , 1 )
assert . Equal ( t , "U2F Key" , res [ 0 ] . Name )
}
func TestU2FRegistration_TableName ( t * testing . T ) {
assert . Equal ( t , "u2f_registration" , U2FRegistration { } . TableName ( ) )
}
func TestU2FRegistration_UpdateCounter ( t * testing . T ) {
assert . NoError ( t , PrepareTestDatabase ( ) )
reg := AssertExistsAndLoadBean ( t , & U2FRegistration { ID : 1 } ) . ( * U2FRegistration )
reg . Counter = 1
assert . NoError ( t , reg . UpdateCounter ( ) )
AssertExistsIf ( t , true , & U2FRegistration { ID : 1 , Counter : 1 } )
}
2019-03-05 05:34:52 +03:00
func TestU2FRegistration_UpdateLargeCounter ( t * testing . T ) {
assert . NoError ( t , PrepareTestDatabase ( ) )
reg := AssertExistsAndLoadBean ( t , & U2FRegistration { ID : 1 } ) . ( * U2FRegistration )
reg . Counter = 0xffffffff
assert . NoError ( t , reg . UpdateCounter ( ) )
AssertExistsIf ( t , true , & U2FRegistration { ID : 1 , Counter : 0xffffffff } )
}
2018-05-19 17:12:37 +03:00
func TestCreateRegistration ( t * testing . T ) {
assert . NoError ( t , PrepareTestDatabase ( ) )
user := AssertExistsAndLoadBean ( t , & User { ID : 1 } ) . ( * User )
res , err := CreateRegistration ( user , "U2F Created Key" , & u2f . Registration { Raw : [ ] byte ( "Test" ) } )
assert . NoError ( t , err )
assert . Equal ( t , "U2F Created Key" , res . Name )
assert . Equal ( t , [ ] byte ( "Test" ) , res . Raw )
AssertExistsIf ( t , true , & U2FRegistration { Name : "U2F Created Key" , UserID : user . ID } )
}
func TestDeleteRegistration ( t * testing . T ) {
assert . NoError ( t , PrepareTestDatabase ( ) )
reg := AssertExistsAndLoadBean ( t , & U2FRegistration { ID : 1 } ) . ( * U2FRegistration )
assert . NoError ( t , DeleteRegistration ( reg ) )
AssertNotExistsBean ( t , & U2FRegistration { ID : 1 } )
}