Ejemplo n.º 1
0
 def test_with_no_result(self, authenticated, query_api):
     api = MockAPI()
     authenticated.return_value = False
     api = MockAPI(config["file"], 0)
     query_api.return_value = api.get_json()
     gist = Gist(config["user"], config["file"])
     filtered = [g for g in gist.filter_gists()]
     self.assertEqual(len(filtered), 0)
Ejemplo n.º 2
0
 def test_select_file_with_no_option(self, authenticated, query_api):
     api = MockAPI()
     authenticated.return_value = False
     api = MockAPI(config["file"], 0)
     query_api.return_value = api.get_json()
     gist = Gist(config["user"], config["file"])
     selected = gist.select_file([g for g in api.get_results()])
     self.assertFalse(selected)
Ejemplo n.º 3
0
 def test_select_file_with_two_options(self, authenticated, query_api, ask):
     api = MockAPI(config["file"], 2)
     authenticated.return_value = False
     query_api.return_value = api.get_json()
     ask.return_value = "2"
     gist = Gist(config["user"], config["file"])
     selected = gist.select_file([g for g in api.get_results()])
     self.assertEqual(selected["id"], "2")
     self.assertEqual(selected["raw_url"], "URL #2")
     self.assertEqual(selected["description"], "Gist #2")
Ejemplo n.º 4
0
 def test_with_sample_gists(self, authenticated, query_api):
     authenticated.return_value = False
     query_api.return_value = config["gists"]
     gist = Gist(config["user"], config["file"])
     filtered = [g for g in gist.filter_gists()]
     hashed_url = md5(filtered[0]["raw_url"].encode("utf-8")).hexdigest()
     self.assertEqual(len(filtered), 1)
     self.assertEqual(filtered[0]["id"], "409fac6ac23bf515f495")
     self.assertEqual(filtered[0]["description"], config["file"])
     self.assertEqual(hashed_url, "847fe81c7fdc3b6bd7184379fcd42773")
Ejemplo n.º 5
0
 def test_select_file_with_two_options(self, authenticated, filter_g, ask):
     authenticated.return_value = False
     two_options = [
         {"id": 12345, "description": "Gist #1", "raw_url": "URL #1"},
         {"id": 67890, "description": "Gist #2", "raw_url": "URL #2"},
     ]
     filter_g.return_value = two_options
     ask.return_value = "2"
     gist = Gist(config["user"], config["file"])
     self.assertEqual(gist.load_gist_info(), two_options[1])
Ejemplo n.º 6
0
 def test_select_file_with_one_option(self, authenticated, query_api):
     api = MockAPI()
     authenticated.return_value = False
     api = MockAPI(config["file"])
     query_api.return_value = api.get_json()
     results = [g for g in api.get_results()]
     gist = Gist(config["user"], config["file"])
     selected = gist.select_file(results)
     self.assertEqual(selected["id"], "1")
     self.assertEqual(selected["raw_url"], "URL #1")
     self.assertEqual(selected["description"], "Gist #1")
Ejemplo n.º 7
0
 def test_with_two_results(self, authenticated, query_api, ask):
     api = MockAPI()
     authenticated.return_value = False
     api = MockAPI(config["file"], 2)
     query_api.return_value = api.get_json()
     ask.return_value = 1
     gist = Gist(config["user"], config["file"])
     filtered = [g for g in gist.filter_gists()]
     self.assertEqual(len(filtered), 2, api.get_json())
     self.assertEqual(filtered[0]["id"], "1")
     self.assertEqual(filtered[0]["description"], "Gist #1")
     self.assertEqual(filtered[0]["raw_url"], "URL #1")
     self.assertEqual(filtered[1]["id"], "2")
     self.assertEqual(filtered[1]["description"], "Gist #2")
     self.assertEqual(filtered[1]["raw_url"], "URL #2")
Ejemplo n.º 8
0
 def test_select_file_with_no_option(self, authenticated, filter_gists):
     authenticated.return_value = False
     filter_gists.return_value = list()
     gist = Gist(config["user"], config["file"])
     self.assertFalse(gist.load_gist_info())
Ejemplo n.º 9
0
 def test_select_file_with_one_option(self, authenticated, filter_gists):
     authenticated.return_value = False
     one_option = [{"id": 12345, "description": "Gist #1", "raw_url": "URL #1"}]
     filter_gists.return_value = one_option
     gist = Gist(config["user"], config["file"])
     self.assertEqual(gist.load_gist_info(), one_option[0])
Ejemplo n.º 10
0
 def setUp(self, authenticated, query_api, curl):
     authenticated.return_value = False
     query_api.return_value = config['gists']
     curl.return_value = config['content']
     self.gist = Gist(config['user'], config['file'])
Ejemplo n.º 11
0
class TestDownload(TestCase):

    @patch('getgist.__main__.Gist.curl')
    @patch('getgist.__main__.Gist.query_api')
    @patch('getgist.__main__.Gist.authenticated')
    def setUp(self, authenticated, query_api, curl):
        authenticated.return_value = False
        query_api.return_value = config['gists']
        curl.return_value = config['content']
        self.gist = Gist(config['user'], config['file'])

    def tearDown(self):
        for name in os.listdir(self.gist.local_dir):
            if search(r'^({})(\.bkp(.\d)?)?$'.format(config['file']), name):
                os.remove(os.path.join(self.gist.local_dir, name))

    @patch('getgist.__main__.Gist.ask')
    def test_local_dir(self, mocked_ask):
        mocked_ask.return_value = 'y'
        self.assertTrue(os.path.exists(self.gist.local_dir))
        self.assertTrue(os.path.isdir(self.gist.local_dir))

    @patch('getgist.__main__.Gist.ask')
    @patch('getgist.__main__.Gist.curl')
    def test_download(self, curl, ask):
        ask.return_value = 'y'
        curl.return_value = config['content']
        self.gist.save()
        self.assertTrue(os.path.exists(self.gist.local_path))
        self.assertEqual(open(self.gist.local_path).read(), config['content'])

    @patch('getgist.__main__.Gist.ask')
    @patch('getgist.__main__.Gist.curl')
    def test_backup(self, curl,  ask):

        # mock input value
        ask.return_value = 'n'
        curl.return_value = config['content']

        # backup filenames
        backup1 = '{}.bkp'.format(self.gist.local_path)
        backup2 = '{}.bkp1'.format(self.gist.local_path)
        backup3 = '{}.bkp2'.format(self.gist.local_path)

        # 1st download
        self.gist.save()
        self.assertTrue(os.path.exists(self.gist.local_path))
        self.assertEqual(config['content'], open(self.gist.local_path).read())

        # 2nd download
        self.gist.save()
        self.assertTrue(os.path.exists(backup1))
        self.assertEqual(config['content'], open(backup1).read())

        # 3rd download
        self.gist.save()
        self.assertTrue(os.path.exists(backup2))
        self.assertEqual(config['content'], open(backup2).read())

        # 4th download
        self.gist.save()
        self.assertTrue(os.path.exists(backup3))
        self.assertEqual(config['content'], open(backup3).read())