From 256c0cdb6be05e345b03f478d2a45b1233c6238f Mon Sep 17 00:00:00 2001
From: Akash <64683866+akamhy@users.noreply.github.com>
Date: Sat, 18 Jul 2020 16:39:35 +0530
Subject: [PATCH] update test - save

---
 tests/test_1.py | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/tests/test_1.py b/tests/test_1.py
index 5426c4a..b288745 100644
--- a/tests/test_1.py
+++ b/tests/test_1.py
@@ -3,6 +3,7 @@ import sys
 sys.path.append("..")
 import waybackpy
 import pytest
+import random
 
 user_agent = "Mozilla/5.0 (Windows NT 6.2; rv:20.0) Gecko/20121202 Firefox/20.0"
 
@@ -20,7 +21,17 @@ def test_url_check():
 
 def test_save():
     # Test for urls that exist and can be archived.
-    url1="https://github.com/akamhy/waybackpy"
+
+    url_list = [
+        "en.wikipedia.org",
+        "www.wikidata.org",
+        "commons.wikimedia.org",
+        "www.wiktionary.org",
+        "www.w3schools.com",
+        "www.youtube.com"
+    ]
+
+    url1=random.choices(url_list)
     target = waybackpy.Url(url1, "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_9_2) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/36.0.1944.0 Safari/537.36")
     archived_url1 = target.save()
     assert url1 in archived_url1