From 1526266bf39cd981397e0b5860acfbb6f408afbd Mon Sep 17 00:00:00 2001
From: Mads Marquart <madsmtm@gmail.com>
Date: Wed, 22 Jan 2020 00:57:02 +0100
Subject: [PATCH] Export MessageData model

---
 fbchat/__init__.py           | 2 +-
 tests/test_client_payload.py | 3 ++-
 tests/test_delta_class.py    | 3 ++-
 tests/test_delta_type.py     | 3 ++-
 tests/test_message.py        | 9 ++-------
 5 files changed, 9 insertions(+), 11 deletions(-)

diff --git a/fbchat/__init__.py b/fbchat/__init__.py
index cc44190..93f300d 100644
--- a/fbchat/__init__.py
+++ b/fbchat/__init__.py
@@ -27,7 +27,7 @@ from ._thread import ThreadLocation, ThreadABC, Thread
 from ._user import User, UserData, ActiveStatus
 from ._group import Group, GroupData
 from ._page import Page, PageData
-from ._message import EmojiSize, Mention, Message
+from ._message import EmojiSize, Mention, Message, MessageData
 from ._attachment import Attachment, UnsentMessage, ShareAttachment
 from ._sticker import Sticker
 from ._location import LocationAttachment, LiveLocationAttachment
diff --git a/tests/test_client_payload.py b/tests/test_client_payload.py
index c85018f..8994fc1 100644
--- a/tests/test_client_payload.py
+++ b/tests/test_client_payload.py
@@ -4,6 +4,8 @@ from fbchat import (
     ParseError,
     User,
     Group,
+    Message,
+    MessageData,
     UnknownEvent,
     ReactionEvent,
     UserStatusEvent,
@@ -11,7 +13,6 @@ from fbchat import (
     UnsendEvent,
     MessageReplyEvent,
 )
-from fbchat._message import Message, MessageData
 from fbchat._client_payload import parse_client_delta, parse_client_payloads
 
 
diff --git a/tests/test_delta_class.py b/tests/test_delta_class.py
index 90e3828..eba1e9c 100644
--- a/tests/test_delta_class.py
+++ b/tests/test_delta_class.py
@@ -4,6 +4,8 @@ from fbchat import (
     ParseError,
     User,
     Group,
+    Message,
+    MessageData,
     ThreadLocation,
     UnknownEvent,
     PeopleAdded,
@@ -15,7 +17,6 @@ from fbchat import (
     MessageEvent,
     ThreadFolder,
 )
-from fbchat._message import Message, MessageData
 from fbchat._delta_class import parse_delta
 
 
diff --git a/tests/test_delta_type.py b/tests/test_delta_type.py
index ac44ad2..52ed670 100644
--- a/tests/test_delta_type.py
+++ b/tests/test_delta_type.py
@@ -5,6 +5,8 @@ from fbchat import (
     ParseError,
     User,
     Group,
+    Message,
+    MessageData,
     Poll,
     PollOption,
     PlanData,
@@ -27,7 +29,6 @@ from fbchat import (
     PlanDeleted,
     PlanResponded,
 )
-from fbchat._message import Message, MessageData
 from fbchat._delta_type import parse_delta
 
 
diff --git a/tests/test_message.py b/tests/test_message.py
index 0cebf4a..e08514f 100644
--- a/tests/test_message.py
+++ b/tests/test_message.py
@@ -1,12 +1,7 @@
 import pytest
 import fbchat
-from fbchat._message import (
-    EmojiSize,
-    Mention,
-    Message,
-    MessageData,
-    graphql_to_extensible_attachment,
-)
+from fbchat import EmojiSize, Mention, Message, MessageData
+from fbchat._message import graphql_to_extensible_attachment
 
 
 @pytest.mark.parametrize(