def test_update_friends(self, user_data, friends_data): user = User(data=user_data) user.update_friends(data=friends_data) assert len(user.friends) == 2 assert user.friends[0].id == '2345678901' assert user.friends[0].name == 'Test Friend' assert user.friends[1].id == '3456789012' assert user.friends[1].name == 'Test Friend2'
def test_update_posts(self, user_data, posts_data): user = User(data=user_data) user.update_posts(data=posts_data) assert len(user.posts) == 2 assert user.posts[0].id == "100005283221676_492510915650769" assert user.posts[0].message == 'Test Message' assert user.posts[0].created_time == "2014-05-02T23:11:33+0000" assert user.posts[1].id == "100005283221676_531093118725882" assert user.posts[1].story == 'Test Story' assert user.posts[1].created_time == "2012-04-01T07:00:00+0000"
def test_network_export(self, user_data, friends, posts_data): user = User(data=user_data) friends[0].update_posts(data=posts_data) friends[1].update_posts(data=posts_data) user.update_friends(friends=friends) path = user.walk() export_file_path = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'test_network_export' ) data = user.export(export_file_path)
def test_walk(self, user_data, friends, posts_data): user = User(data=user_data) friends[0].update_posts(data=posts_data) friends[1].update_posts(data=posts_data) user.update_friends(friends=friends) path = user.walk() assert len(path) == 6 assert isinstance(path[0], User) assert isinstance(path[1], Post) assert isinstance(path[2], Post) assert isinstance(path[3], User) assert isinstance(path[4], Post) assert isinstance(path[5], Post)