You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
56 lines
1.6 KiB
56 lines
1.6 KiB
// Copyright (c) 2020 Proton Technologies AG |
|
// |
|
// This file is part of ProtonMail Bridge.Bridge. |
|
// |
|
// ProtonMail Bridge is free software: you can redistribute it and/or modify |
|
// it under the terms of the GNU General Public License as published by |
|
// the Free Software Foundation, either version 3 of the License, or |
|
// (at your option) any later version. |
|
// |
|
// ProtonMail Bridge is distributed in the hope that it will be useful, |
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
// GNU General Public License for more details. |
|
// |
|
// You should have received a copy of the GNU General Public License |
|
// along with ProtonMail Bridge. If not, see <https://www.gnu.org/licenses/>. |
|
|
|
package liveapi |
|
|
|
import ( |
|
"net/http" |
|
"sync" |
|
|
|
"github.com/ProtonMail/proton-bridge/pkg/pmapi" |
|
) |
|
|
|
type Controller struct { |
|
// Internal states. |
|
lock *sync.RWMutex |
|
calls []*fakeCall |
|
pmapiByUsername map[string]pmapi.Client |
|
messageIDsByUsername map[string][]string |
|
clientManager *pmapi.ClientManager |
|
|
|
// State controlled by test. |
|
noInternetConnection bool |
|
} |
|
|
|
func NewController(cm *pmapi.ClientManager) *Controller { |
|
controller := &Controller{ |
|
lock: &sync.RWMutex{}, |
|
calls: []*fakeCall{}, |
|
pmapiByUsername: map[string]pmapi.Client{}, |
|
messageIDsByUsername: map[string][]string{}, |
|
clientManager: cm, |
|
|
|
noInternetConnection: false, |
|
} |
|
|
|
cm.SetRoundTripper(&fakeTransport{ |
|
ctl: controller, |
|
transport: http.DefaultTransport, |
|
}) |
|
|
|
return controller |
|
}
|
|
|