import sys import write import hitter import api_call import servers # language = "Java" # query = "gradle" language = "python" query = "foobar" try: filename = sys.argv[1] except IndexError: filename = "best_matches.txt" # java_gradle_hrefs = api_call.hit_github(language, query) # java_gradle_hrefs = hitter.hit_github(language, query) java_gradle_hrefs = servers.hard_code(language, query) ### gets beyond the API limit write.add_repos_to_text_file(java_gradle_hrefs, filename)
def test_add_to_txt_file(self): write.add_repos_to_text_file(self.hrefs(), "test.txt") f = open("test.txt", "r") f.readline() self.assertEqual(f.readline().rstrip(), "http://test1.com") os.remove("test.txt") ## keep the directory clean