Start implementing UnmarshalJSON for Star
This commit is contained in:
parent
43014ca473
commit
d28ea1a30b
2 changed files with 61 additions and 0 deletions
|
@ -6,6 +6,7 @@ package forgefed
|
||||||
import (
|
import (
|
||||||
"code.gitea.io/gitea/modules/context"
|
"code.gitea.io/gitea/modules/context"
|
||||||
ap "github.com/go-ap/activitypub"
|
ap "github.com/go-ap/activitypub"
|
||||||
|
"github.com/valyala/fastjson"
|
||||||
)
|
)
|
||||||
|
|
||||||
type (
|
type (
|
||||||
|
@ -64,3 +65,23 @@ func (a Star) MarshalJSON() ([]byte, error) {
|
||||||
ap.JSONWrite(&b, '}')
|
ap.JSONWrite(&b, '}')
|
||||||
return b, nil
|
return b, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func JSONLoadStar(val *fastjson.Value, s *Star) error {
|
||||||
|
if err := ap.OnActivity(&s.Activity, func(a *ap.Activity) error {
|
||||||
|
return ap.JSONLoadActivity(val, a)
|
||||||
|
}); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
s.Source = SourceType(ap.JSONGetString(val, "source"))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Star) UnmarshalJSON(data []byte) error {
|
||||||
|
p := fastjson.Parser{}
|
||||||
|
val, err := p.ParseBytes(data)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
return JSONLoadStar(val, s)
|
||||||
|
}
|
||||||
|
|
|
@ -48,3 +48,43 @@ func Test_StarMarshalJSON(t *testing.T) {
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func Test_StarUnmarshalJSON(t *testing.T) {
|
||||||
|
type testPair struct {
|
||||||
|
item []byte
|
||||||
|
want Star
|
||||||
|
wantErr error
|
||||||
|
}
|
||||||
|
|
||||||
|
tests := map[string]testPair{
|
||||||
|
"empty": {
|
||||||
|
item: []byte(``),
|
||||||
|
want: Star{},
|
||||||
|
},
|
||||||
|
"with ID": {
|
||||||
|
item: []byte(`{"source":"forgejo","type":"Star","actor":"https://repo.prod.meissa.de/api/activitypub/user-id/1","object":"https://codeberg.org/api/activitypub/repository-id/1"}`),
|
||||||
|
want: Star{
|
||||||
|
Source: "forgejo",
|
||||||
|
Activity: ap.Activity{
|
||||||
|
Actor: ap.IRI("https://repo.prod.meissa.de/api/activitypub/user-id/1"),
|
||||||
|
Type: "Star",
|
||||||
|
Object: ap.IRI("https://codeberg.org/api/activitypub/repository-id/1"),
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for name, tt := range tests {
|
||||||
|
t.Run(name, func(t *testing.T) {
|
||||||
|
got := new(Star)
|
||||||
|
err := got.UnmarshalJSON(tt.item)
|
||||||
|
if (err != nil || tt.wantErr != nil) && tt.wantErr.Error() != err.Error() {
|
||||||
|
t.Errorf("UnmarshalJSON() error = \"%v\", wantErr \"%v\"", err, tt.wantErr)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !reflect.DeepEqual(got, tt.want) {
|
||||||
|
t.Errorf("UnmarshalJSON() got = %q, want %q", got, tt.want)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
Loading…
Reference in a new issue