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_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)
