twitter-scrapper/auth_test.go
Alexander Sheiko e86c2fa4ce Improve login flow
support LoginAcid task
2023-05-10 21:48:41 +03:00

34 lines
761 B
Go

package twitterscraper_test
import (
"os"
"testing"
twitterscraper "github.com/n0madic/twitter-scraper"
)
var (
username = os.Getenv("TWITTER_USERNAME")
password = os.Getenv("TWITTER_PASSWORD")
email = os.Getenv("TWITTER_EMAIL")
)
func TestAuth(t *testing.T) {
scraper := twitterscraper.New()
if err := scraper.Login(username, password, email); err != nil {
t.Fatalf("Login() error = %v", err)
}
if !scraper.IsLoggedIn() {
t.Fatalf("Expected IsLoggedIn() = true")
}
cookies := scraper.GetCookies()
scraper.Logout()
if scraper.IsLoggedIn() {
t.Error("Expected IsLoggedIn() = false")
}
scraper2 := twitterscraper.New()
scraper2.SetCookies(cookies)
if !scraper2.IsLoggedIn() {
t.Error("Expected restored IsLoggedIn() = true")
}
}