Example #1
0
    def test_get_user_info(self):
        email = '*****@*****.**'
        log.info(f'Downloading user info for account {email}...')
        fpl = FPLPandas(email, 'TestMcTestFace')
        user_info = fpl.get_user_info()
        log.info(f'User info downloaded.')

        self.assertTrue(user_info.shape[0] > 0)
Example #2
0
    def test_get_user_info(self):
        test_data = {'player': {'entry': '123'}}
        expected_df = pd.DataFrame.from_dict([test_data['player']])

        fpl_mock = mock.MagicMock()

        async def mock_login(email, password):
            self.assertEqual(email, 'email')
            self.assertEqual(password, 'password')

        async def mock_get_user_info():
            return {'player': {'entry': '123'}}

        fpl_mock.get_user_info = mock_get_user_info
        fpl_mock.login = mock_login

        fpl = FPLPandas('email', 'password', fpl=fpl_mock)
        actual_df = fpl.get_user_info()

        assert_frame_equal(expected_df, actual_df)