package mastodon import ( "context" "crypto/ecdsa" "crypto/elliptic" "crypto/rand" "fmt" "net/http" "net/http/httptest" "testing" ) func TestGetNotifications(t *testing.T) { ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { switch r.URL.Path { case "/api/v1/notifications": fmt.Fprintln(w, `[{"id": 122, "action_taken": false}, {"id": 123, "action_taken": true}]`) return case "/api/v1/notifications/123": fmt.Fprintln(w, `{"id": 123, "action_taken": true}`) return case "/api/v1/notifications/clear": fmt.Fprintln(w, `{}`) return case "/api/v1/notifications/dismiss": fmt.Fprintln(w, `{}`) return } http.Error(w, http.StatusText(http.StatusNotFound), http.StatusNotFound) return })) defer ts.Close() client := NewClient(&Config{ Server: ts.URL, ClientID: "foo", ClientSecret: "bar", AccessToken: "zoo", }) ns, err := client.GetNotifications(context.Background(), nil) if err != nil { t.Fatalf("should not be fail: %v", err) } if len(ns) != 2 { t.Fatalf("result should be two: %d", len(ns)) } if ns[0].ID != "122" { t.Fatalf("want %v but %v", "122", ns[0].ID) } if ns[1].ID != "123" { t.Fatalf("want %v but %v", "123", ns[1].ID) } n, err := client.GetNotification(context.Background(), "123") if err != nil { t.Fatalf("should not be fail: %v", err) } if n.ID != "123" { t.Fatalf("want %v but %v", "123", n.ID) } err = client.ClearNotifications(context.Background()) if err != nil { t.Fatalf("should not be fail: %v", err) } err = client.DismissNotification(context.Background(), "123") if err != nil { t.Fatalf("should not be fail: %v", err) } } func TestPushSubscription(t *testing.T) { ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { switch r.URL.Path { case "/api/v1/push/subscription": fmt.Fprintln(w, ` {"id":1,"endpoint":"https://example.org","alerts":{"follow":true,"favourite":"true","reblog":"true","mention":"true"},"server_key":"foobar"}`) return } http.Error(w, http.StatusText(http.StatusNotFound), http.StatusNotFound) return })) defer ts.Close() client := NewClient(&Config{ Server: ts.URL, ClientID: "foo", ClientSecret: "bar", AccessToken: "zoo", }) enabled := new(Sbool) *enabled = true alerts := PushAlerts{Follow: enabled, Favourite: enabled, Reblog: enabled, Mention: enabled} priv, err := ecdsa.GenerateKey(elliptic.P256(), rand.Reader) if err != nil { t.Fatal(err) } shared := make([]byte, 16) _, err = rand.Read(shared) if err != nil { t.Fatal(err) } testSub := func(sub *PushSubscription, err error) { if err != nil { t.Fatalf("should not be fail: %v", err) } if sub.ID != "1" { t.Fatalf("want %v but %v", "1", sub.ID) } if sub.Endpoint != "https://example.org" { t.Fatalf("want %v but %v", "https://example.org", sub.Endpoint) } if sub.ServerKey != "foobar" { t.Fatalf("want %v but %v", "foobar", sub.ServerKey) } if *sub.Alerts.Favourite != true { t.Fatalf("want %v but %v", true, *sub.Alerts.Favourite) } if *sub.Alerts.Mention != true { t.Fatalf("want %v but %v", true, *sub.Alerts.Mention) } if *sub.Alerts.Reblog != true { t.Fatalf("want %v but %v", true, *sub.Alerts.Reblog) } if *sub.Alerts.Follow != true { t.Fatalf("want %v but %v", true, *sub.Alerts.Follow) } } sub, err := client.AddPushSubscription(context.Background(), "http://example.org", priv.PublicKey, shared, alerts) testSub(sub, err) sub, err = client.GetPushSubscription(context.Background()) testSub(sub, err) sub, err = client.UpdatePushSubscription(context.Background(), &alerts) testSub(sub, err) err = client.RemovePushSubscription(context.Background()) if err != nil { t.Fatalf("should not be fail: %v", err) } }