Write private data archives with safe extraction

[onechat] / whatsapp_send_test.go

diff --git a/whatsapp_send_test.go b/whatsapp_send_test.go
index 51651b2a52659ba44bc1ef1f65837f8c5e997387..469912b0e0806adcb4c80712de0a82498c1181cb 100644 (file)
--- a/whatsapp_send_test.go
+++ b/whatsapp_send_test.go
@@ -70,6 +70,7 @@ func TestWhatsAppSendToAGroup(t *testing.T) {
 func TestWhatsAppSendThatFails(t *testing.T) {
        whatsapp, fake := whatsappStartTestLinked(t)
        fake.sendErr = errors.New("server error for " + whatsappTestNumber)
 func TestWhatsAppSendThatFails(t *testing.T) {
        whatsapp, fake := whatsappStartTestLinked(t)
        fake.sendErr = errors.New("server error for " + whatsappTestNumber)

+

        daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
        whatsappTestConnect(t, fake)
 
        daemonStartTestWithWhatsApp(t, newModelFake(t), nil, whatsapp)
        whatsappTestConnect(t, fake)
 


@@ -170,6 +171,7 @@ func TestWhatsAppSendWhileDisconnected(t *testing.T) {
        require.EqualError(t, err, "WhatsApp is disconnected, so nothing was sent: try again later")
 
        require.True(t, fake.Emit(&events.Connected{}))
        require.EqualError(t, err, "WhatsApp is disconnected, so nothing was sent: try again later")
 
        require.True(t, fake.Emit(&events.Connected{}))

+

        answer, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "hi")
        require.NoError(t, err)
        require.Equal(t, "Sent to +15550002222 at 12:05.", answer)
        answer, err := whatsapp.Send(t.Context(), "keys", whatsappTestAna.String(), "hi")
        require.NoError(t, err)
        require.Equal(t, "Sent to +15550002222 at 12:05.", answer)


@@ -186,10 +188,12 @@ func TestWhatsAppSendWithoutALink(t *testing.T) {
 func TestAgentSendsAWhatsAppMessage(t *testing.T) {
        whatsapp, fake := whatsappStartTestLinked(t)
        whatsappTestContacts(t, whatsapp)
 func TestAgentSendsAWhatsAppMessage(t *testing.T) {
        whatsapp, fake := whatsappStartTestLinked(t)
        whatsappTestContacts(t, whatsapp)

+

        server := newModelFake(t)
        server.reply = func(w http.ResponseWriter, call ModelFakeCall) {
                if call.number == 0 {
                        modelWriteEvents(w, modelCompletedEvent([]any{modelFunctionCallItem("fc_1", "call_1", "whatsapp_send", `{"id":"keys","chat":"15550002222@s.whatsapp.net","text":"The keys are under the mat."}`)}, 100, 0, 5))
        server := newModelFake(t)
        server.reply = func(w http.ResponseWriter, call ModelFakeCall) {
                if call.number == 0 {
                        modelWriteEvents(w, modelCompletedEvent([]any{modelFunctionCallItem("fc_1", "call_1", "whatsapp_send", `{"id":"keys","chat":"15550002222@s.whatsapp.net","text":"The keys are under the mat."}`)}, 100, 0, 5))

+

                        return
                }
 
                        return
                }