Silverfish/proton-bridge
1
0
Fork 1
mirror of https://github.com/ProtonMail/proton-bridge.git synced 2025-12-17 15:46:44 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: make store use ClientManager

Browse Source
This commit is contained in:
James Houlahan
2020-04-07 09:55:28 +02:00
parent f269be4291
commit 042c340881
43 changed files with 414 additions and 264 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
test/fakeapi/controller.go
View File

@ -44,8 +44,8 @@ type Controller struct {
log *logrus.Entry
}
func NewController() *Controller {
return &Controller{
func NewController(cm *pmapi.ClientManager) (cntrl *Controller) {
cntrl = &Controller{
lock: &sync.RWMutex{},
fakeAPIs: []*FakePMAPI{},
calls: []*fakeCall{},
@ -62,10 +62,12 @@ func NewController() *Controller {
log: logrus.WithField("pkg", "fakeapi-controller"),
}
}
func (cntrl *Controller) GetClient(userID string) *FakePMAPI {
fakeAPI := New(cntrl)
cntrl.fakeAPIs = append(cntrl.fakeAPIs, fakeAPI)
return fakeAPI
cm.SetClientConstructor(func(userID string) pmapi.Client {
fakeAPI := New(cntrl)
cntrl.fakeAPIs = append(cntrl.fakeAPIs, fakeAPI)
return fakeAPI
})
return
}