forked from cwtch.im/libcwtch-go
125 lines
3.2 KiB
Go
125 lines
3.2 KiB
Go
package contact
|
|
|
|
import (
|
|
"cwtch.im/cwtch/model"
|
|
"git.openprivacy.ca/flutter/libcwtch-go/features"
|
|
"testing"
|
|
)
|
|
|
|
const ValidHostname = "openpravyvc6spbd4flzn4g2iqu4sxzsizbtb5aqec25t76dnoo5w7yd"
|
|
|
|
type MockPeer struct {
|
|
hasContact bool
|
|
addContact bool
|
|
peerRequest bool
|
|
}
|
|
|
|
func (m MockPeer) BlockUnknownConnections() {
|
|
panic("should never be called")
|
|
}
|
|
|
|
func (m MockPeer) AllowUnknownConnections() {
|
|
panic("should never be called")
|
|
}
|
|
|
|
func (m MockPeer) GetContacts() []string {
|
|
panic("should never be called")
|
|
}
|
|
|
|
func (m MockPeer) GetContact(s string) *model.PublicProfile {
|
|
if m.hasContact {
|
|
return &(model.GenerateNewProfile("").PublicProfile)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (m MockPeer) GetContactAttribute(s string, s2 string) (string, bool) {
|
|
panic("should never be called")
|
|
}
|
|
|
|
func (m *MockPeer) AddContact(nick, onion string, authorization model.Authorization) {
|
|
m.addContact = true
|
|
}
|
|
|
|
func (m MockPeer) SetContactAuthorization(s string, authorization model.Authorization) error {
|
|
panic("should never be called")
|
|
}
|
|
|
|
func (m MockPeer) SetContactAttribute(s string, s2 string, s3 string) {
|
|
panic("should never be called")
|
|
}
|
|
|
|
func (m MockPeer) DeleteContact(s string) {
|
|
panic("should never be called")
|
|
}
|
|
|
|
func (m *MockPeer) PeerWithOnion(s string) {
|
|
m.peerRequest = true
|
|
}
|
|
|
|
func (m MockPeer) JoinServer(s string) error {
|
|
panic("should never be called")
|
|
}
|
|
|
|
func TestContactFunctionality_InValidHostname(t *testing.T) {
|
|
cf, _ := FunctionalityGate(map[string]bool{})
|
|
|
|
peer := &MockPeer{
|
|
hasContact: false,
|
|
addContact: false,
|
|
peerRequest: false,
|
|
}
|
|
|
|
response := cf.HandleImportString(peer, "")
|
|
|
|
if peer.addContact || peer.peerRequest {
|
|
t.Fatalf("HandleImportString for a malformed import string should have no resulted in addContact or a peerRequest: %v", peer)
|
|
}
|
|
|
|
if response.Error() != features.ConstructResponse(addContactPrefix, "invalid_import_string").Error() {
|
|
t.Fatalf("Response to a successful import is malformed: %v", response)
|
|
}
|
|
|
|
}
|
|
|
|
func TestContactFunctionality_ValidHostnameExistingContact(t *testing.T) {
|
|
cf, _ := FunctionalityGate(map[string]bool{})
|
|
|
|
peer := &MockPeer{
|
|
hasContact: true,
|
|
addContact: false,
|
|
peerRequest: false,
|
|
}
|
|
|
|
response := cf.HandleImportString(peer, ValidHostname)
|
|
|
|
if peer.addContact || peer.peerRequest {
|
|
t.Fatalf("HandleImportString for a valid string should not call addContact or a peerRequest when the contact already exists: %v", peer)
|
|
}
|
|
|
|
if response.Error() != features.ConstructResponse(addContactPrefix, "contact_already_exists").Error() {
|
|
t.Fatalf("Response to a successful import is malformed: %v", response)
|
|
}
|
|
|
|
}
|
|
|
|
func TestContactFunctionality_ValidHostnameUnknownContact(t *testing.T) {
|
|
cf, _ := FunctionalityGate(map[string]bool{})
|
|
|
|
peer := &MockPeer{
|
|
hasContact: false,
|
|
addContact: false,
|
|
peerRequest: false,
|
|
}
|
|
|
|
response := cf.HandleImportString(peer, ValidHostname)
|
|
|
|
if peer.addContact && peer.peerRequest {
|
|
if response.Error() != features.ConstructResponse(addContactPrefix, "success").Error() {
|
|
t.Fatalf("Response to a successful import is malformed: %v", response)
|
|
}
|
|
} else {
|
|
t.Fatalf("HandleImportString for a valid import string should have resulted in addContact or a peerRequest: %v", peer)
|
|
}
|
|
}
|