mirror of
https://github.com/slackhq/nebula.git
synced 2026-07-01 02:50:29 +02:00
fixup tests
This commit is contained in:
@@ -22,12 +22,18 @@ func TestNewCipherStateDispatch(t *testing.T) {
|
|||||||
encA, _ := buildCipherStates(t, CipherAESGCM)
|
encA, _ := buildCipherStates(t, CipherAESGCM)
|
||||||
encC, _ := buildCipherStates(t, noise.CipherChaChaPoly)
|
encC, _ := buildCipherStates(t, noise.CipherChaChaPoly)
|
||||||
|
|
||||||
assert.IsType(t, &CipherStateAESGCM{}, NewCipherState(encA, CipherAESGCM))
|
if CipherAESGCM == noise.CipherAESGCM {
|
||||||
|
assert.IsType(t, &CipherStateAESGCM{}, NewCipherState(encA, CipherAESGCM))
|
||||||
|
} else {
|
||||||
|
// fips140
|
||||||
|
assert.IsType(t, encA.Cipher(), NewCipherState(encA, CipherAESGCM))
|
||||||
|
}
|
||||||
|
|
||||||
assert.IsType(t, &CipherStateChaChaPoly{}, NewCipherState(encC, noise.CipherChaChaPoly))
|
assert.IsType(t, &CipherStateChaChaPoly{}, NewCipherState(encC, noise.CipherChaChaPoly))
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestNewCipherStateUnsupportedPanics(t *testing.T) {
|
func TestNewCipherStateUnsupportedPanics(t *testing.T) {
|
||||||
enc, _ := buildCipherStates(t, CipherAESGCM)
|
enc, _ := buildCipherStates(t, noise.CipherChaChaPoly)
|
||||||
assert.Panics(t, func() {
|
assert.Panics(t, func() {
|
||||||
NewCipherState(enc, fakeCipher{})
|
NewCipherState(enc, fakeCipher{})
|
||||||
})
|
})
|
||||||
|
|||||||
Reference in New Issue
Block a user