Silverfish/proton-bridge
1
0
Fork 1
mirror of https://github.com/ProtonMail/proton-bridge.git synced 2025-12-10 04:36:43 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat(GODT-2734): Add testing steps to modify account settings.

Browse Source
This commit is contained in:
Romain LE JEUNE
2023-09-04 16:48:59 +02:00
parent 635b81314a
commit 79524185a8
2 changed files with 71 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/steps_test.go
View File

@ -109,6 +109,12 @@ func (s *scenario) steps(ctx *godog.ScenarioContext) {
ctx.Step(`^the bridge password of user "([^"]*)" is changed to "([^"]*)"`, s.bridgePasswordOfUserIsChangedTo)
ctx.Step(`^the bridge password of user "([^"]*)" is equal to "([^"]*)"`, s.bridgePasswordOfUserIsEqualTo)
// ==== ACCOUNT SETTINGS ====
ctx.Step(`^the account "([^"]*)" has public key attachment "([^"]*)"`, s.accountHasPublicKeyAttachment)
ctx.Step(`^the account "([^"]*)" has sign external messages "([^"]*)"`, s.accountHasSignExternalMessages)
ctx.Step(`^the account "([^"]*)" has default draft format "([^"]*)"`, s.accountHasDefaultDraftFormat)
ctx.Step(`^the account "([^"]*)" has default PGP schema "([^"]*)"`, s.accountHasDefaultPGPSchema)
// ==== IMAP ====
ctx.Step(`^user "([^"]*)" connects IMAP client "([^"]*)"$`, s.userConnectsIMAPClient)
ctx.Step(`^user "([^"]*)" connects IMAP client "([^"]*)" on port (\d+)$`, s.userConnectsIMAPClientOnPort)

65
tests/user_test.go
View File

@ -578,3 +578,68 @@ func (s *scenario) createUserAccount(username, password string, disabled bool) e
return nil
})
}
func (s *scenario) accountHasPublicKeyAttachment(account, enabled string) error {
value := true
switch {
case enabled == "enabled":
value = true
case enabled == "disabled":
value = false
default:
return errors.New("parameter should either be 'enabled' or 'disabled'")
}
return s.t.withClient(context.Background(), account, func(ctx context.Context, c *proton.Client) error {
_, err := c.SetAttachPublicKey(ctx, proton.SetAttachPublicKeyReq{AttachPublicKey: proton.Bool(value)})
return err
})
}
func (s *scenario) accountHasSignExternalMessages(account, enabled string) error {
value := proton.SignExternalMessagesDisabled
switch {
case enabled == "enabled":
value = proton.SignExternalMessagesEnabled
case enabled == "disabled":
value = proton.SignExternalMessagesDisabled
default:
return errors.New("parameter should either be 'enabled' or 'disabled'")
}
return s.t.withClient(context.Background(), account, func(ctx context.Context, c *proton.Client) error {
_, err := c.SetSignExternalMessages(ctx, proton.SetSignExternalMessagesReq{Sign: value})
return err
})
}
func (s *scenario) accountHasDefaultDraftFormat(account, format string) error {
value := rfc822.TextPlain
switch {
case format == "plain":
value = rfc822.TextPlain
case format == "HTML":
value = rfc822.TextHTML
default:
return errors.New("parameter should either be 'plain' or 'HTML'")
}
return s.t.withClient(context.Background(), account, func(ctx context.Context, c *proton.Client) error {
_, err := c.SetDraftMIMEType(ctx, proton.SetDraftMIMETypeReq{MIMEType: value})
return err
})
}
func (s *scenario) accountHasDefaultPGPSchema(account, schema string) error {
value := proton.PGPInlineScheme
switch {
case schema == "inline":
value = proton.PGPInlineScheme
case schema == "MIME":
value = proton.PGPMIMEScheme
default:
return errors.New("parameter should either be 'inline' or 'MIME'")
}
return s.t.withClient(context.Background(), account, func(ctx context.Context, c *proton.Client) error {
_, err := c.SetDefaultPGPScheme(ctx, proton.SetDefaultPGPSchemeReq{PGPScheme: value})
return err
})
}