def test_create_query(self): """ _create_query should return a string with the movie title seperated by %20 instead of spaces and add the year if it is supplied """ movie = 'iron man 3' year = 2013 quality = '1080p' yify = YIFYAPI(None) query = yify._create_query(movie=movie, year=year) self.assertTrue(query.startswith('keywords=')) self.assertFalse(' ' in query) split_movie = movie.split(' ') for term in split_movie: self.assertTrue(term in query)
def test_get_magnet_movie(self): """ _get_magnet_movie should return a dict() with magnet and seeds key or raise an exception: MovieNotFound, QualityNotFound, LookupError """ movie = 'iron man 3' year = 2013 quality = '1080p' yify = YIFYAPI(None) yify._wanted_movie = movie query = yify._create_query(movie=movie, year=year) result = yify._get_magnet_movie(query=query, quality=quality, year=year) self.assertTrue('seeds' in result.keys()) self.assertTrue('magnet' in result.keys()) self.assertEqual(2, len(result)) self.assertTrue(result['magnet'].startswith('magnet:?')) self.assertGreaterEqual(int(result['seeds']), 0)