Create test.py

test.py

+import twint
+import os
+
+'''
+Test.py - Testing TWINT to make sure everything works.
+'''
+
+def test_reg(c, run):
+    print("[+] Beginning vanilla test in {}".format(str(run)))
+    run(c)
+
+def test_db(c, run):
+    print("[+] Beginning DB test in {}".format(str(run)))
+    c.Database = "test_twint.db"
+    run(c)
+
+def custom(c, run, _type):
+    print("[+] Beginning custom {} test in {}".format(_type, str(run)))
+    c.Custom = ["id", "username"]
+    run(c)
+
+def test_json(c, run):
+    c.Store_json = True
+    c.Output = "test_twint.json"
+    custom(c, run, "JSON")
+    print("[+] Beginning JSON test in {}".format(str(run)))
+    run(c)
+
+def test_csv(c, run):
+    c.Store_csv = True
+    c.Output = "test_twint.csv"
+    custom(c, run, "CSV")
+    print("[+] Beginning CSV test in {}".format(str(run)))
+    run(c)
+
+def main():
+    c = twint.Config()
+    c.Username = "verified"
+    c.Limit = 20
+    c.Store_object = True
+
+    # Seperate objects are neccessary.
+
+    f = twint.Config()
+    f.Username = "verified"
+    f.Limit = 20
+    f.Store_object = True
+    f.User_full = True
+    
+    runs = [twint.run.Following,
+            twint.run.Followers,
+            twint.run.Search,
+            twint.run.Profile,
+            twint.run.Favorites
+            ]
+    
+    tests = [test_reg, test_json, test_csv, test_db]
+    
+    # Something breaks if we don't split these up
+
+    for run in runs[:2]: 
+        for test in tests:
+            test(f, run)
+
+    for run in runs[2:]:
+        if run == twint.run.Search:
+            c.Since = "2012-1-1"
+            c.Until = "2017-1-1"
+        else:
+            c.Since = ""
+            c.Until = ""
+        
+        for test in tests:
+            test(c, run)
+
+    files = ["test_twint.db", "test_twint.json", "test_twint.csv"]
+    for _file in files:
+       os.remove(_file)
+
+
+    print("[+] Testing complete!")
+
+if __name__ == '__main__':
+    main()