feat(model): ✨ add model for container

MerleLiuKun · MerleLiuKun · commit 1839bb1198ff · 2021-07-27T19:23:21.000+08:00
diff --git a/pyfacebook/models/__init__.py b/pyfacebook/models/__init__.py
@@ -18,4 +18,5 @@
     IgBusReply,
     IgBusComment,
     IgBusHashtag,
+    IgBusContainer,
 )
diff --git a/pyfacebook/models/ig_business_models.py b/pyfacebook/models/ig_business_models.py
@@ -132,3 +132,16 @@ class IgBusHashtag(BaseModel):
 
     id: Optional[str] = field(repr=True, compare=True)
     name: Optional[str] = field(repr=True)
+
+
+@dataclass
+class IgBusContainer(BaseModel):
+    """
+    A class representing the media container.
+
+    Refer: https://developers.facebook.com/docs/instagram-api/reference/ig-container
+    """
+
+    id: Optional[str] = field(repr=True, compare=True)
+    status: Optional[str] = field()
+    status_code: Optional[str] = field()
diff --git a/testdata/instagram/models/ig_container.json b/testdata/instagram/models/ig_container.json
@@ -1,4 +1 @@
-{
-  "status_code": "FINISHED",
-  "id": "17889615691921648"
-}
+{"status_code":"FINISHED","id":"17889615691921648"}
diff --git a/tests/test_models.py b/tests/test_models.py
@@ -158,3 +158,11 @@ def test_ig_bus_hashtag(helpers):
     hashtag = md.IgBusHashtag.new_from_json_dict(data)
     assert hashtag.id == "17841593698074073"
     assert hashtag.name == "coke"
+
+
+def test_ig_bus_container(helpers):
+    data = helpers.load_json("testdata/instagram/models/ig_container.json")
+
+    container = md.IgBusContainer.new_from_json_dict(data)
+    assert container.id == "17889615691921648"
+    assert container.status_code == "FINISHED"

Original file line number	Diff line number	Diff line change
`@@ -18,4 +18,5 @@`
`18`	`18`	`IgBusReply,`
`19`	`19`	`IgBusComment,`
`20`	`20`	`IgBusHashtag,`
	`21`	`+ IgBusContainer,`
`21`	`22`	`)`