func TestPut(t *testing.T) { mb1, err := mailbox.Create("put1") if err != nil { t.Fatal(err) } mb2, err := mailbox.Create("put2") if err != nil { t.Fatal(err) } accessKey := mailbox.AccessKey{FullAccess: true} accessKey.Create() req := api.PutMessageRequest{ Mailboxes: []string{ mb1.Id, mb2.Id, }, Body: "TEST", } req.Sign(accessKey.Name, accessKey.Secret) var resp api.PutMessageResponse code := doRequest(t, req, &resp, "put") count, err := mb1.MessageCount() if err != nil { t.Fatal(err) } if count == 0 { t.Fatal("Message not added to mailbox") } count, err = mb2.MessageCount() if err != nil { t.Fatal(err) } if count == 0 { t.Fatal("Message not added to mailbox") } if code != 200 { t.Fatal("Server responded with", code) } message1, err := mb1.GetMessage() if err != nil { t.Fatal(err) } message2, err := mb2.GetMessage() if err != nil { t.Fatal(err) } if message1 == nil || message2 == nil { t.Fatal("Message is nil") } if message1.Body != "TEST" { t.Fatal("Incorrect message1 body", message1.Body) } if message2.Body != "TEST" { t.Fatal("Incorrect message2 body", message2.Body) } }
func TestPutBadToken(t *testing.T) { mb, _ := mailbox.Create("puttest.badtoken") accessKey := mailbox.AccessKey{MailboxId: mb.Id} accessKey.Create() req := api.PutMessageRequest{ Mailboxes: []string{mb.Id}, Body: "TEST MESSAGE", } req.Sign(accessKey.Name, accessKey.Secret) var resp api.PutMessageResponse code := doRequest(t, req, &resp, "put") if code == 200 { t.Fatal("Bad token should return error") } }
func TestPutByPattern(t *testing.T) { mb, _ := mailbox.Create("PATTERN") accessKey := mailbox.AccessKey{FullAccess: true} accessKey.Create() req := api.PutMessageRequest{Pattern: "P*"} req.Sign(accessKey.Name, accessKey.Secret) var resp api.PutMessageResponse code := doRequest(t, req, &resp, "put") count, err := mb.MessageCount() if err != nil { t.Fatal(err) } if count == 0 { t.Fatal("Message not added to mailbox") } if code != 200 { t.Fatal("Server responded with", code) } }
// Put sends a message to a series of mailboxes. An array of mailboxes can be // provided, as well as a pattern using '*' as wildcards. The message will by // sent to all matching mailboxes. func (client *Client) Put(mbxs []string, pattern string, msg string, deploymentName string, asset string) (*api.PutMessageResponse, error) { md5, _ := client.hashFile(asset) if asset != "" { exists, err := client.CheckRemoteFile(md5) if err != nil { return nil, err } if exists == true { log.Info("File exists on server, skipping upload") } else { _, err := client.Upload(asset) if err != nil { log.Debug(err.Error()) return nil, errors.New("Could not upload asset") } } } request := api.PutMessageRequest{ Mailboxes: mbxs, Body: msg, Pattern: pattern, DeploymentName: deploymentName, Asset: md5, } request.Sign(client.AccessKeyName, client.AccessKey) var response api.PutMessageResponse err := client.request("put", request, &response) if err != nil { return nil, err } if !response.Validate(client.AccessKey) { return nil, errors.New("Server responded with an invalid signature") } return &response, nil }