Functional test code done

Author: Mike Spicer

testfunctionality.py

@@ -21,18 +21,19 @@
 wait = WebDriverWait(driver,15)
 df = pd.DataFrame()
 
-for item in range(1): 
+for item in range(10): 
     wait.until(EC.visibility_of_element_located((By.TAG_NAME,"body"))).send_keys(Keys.END)
-    time.sleep(15)
+    time.sleep(2)
 # for author in wait.until(EC.presence_of_all_elements_located((By.CSS_SELECTOR, "#ytd-comment-renderer"))):
 #     #author.append(comment.text)    
 #     df = df.append({'author': author}, ignore_index=True))
-df = pd.concat([pd.DataFrame([author.text], columns=['Author']) for author in wait.until(EC.presence_of_all_elements_located((By.CSS_SELECTOR, ".ytd-comment-renderer")) )], ignore_index=True)
+df = pd.concat([pd.DataFrame([author.text], columns=['Author']) for author in wait.until(EC.presence_of_all_elements_located((By.CSS_SELECTOR, "#author-text")) )], ignore_index=True)
 
-df = pd.concat([df([comment.text], columns=['Comment']) for comment in wait.until(EC.presence_of_all_elements_located((By.CSS_SELECTOR, "#content-text")) )], ignore_index=True)
+df['AuthorLink'] = [author.get_attribute('href') for author in wait.until(EC.presence_of_all_elements_located((By.CSS_SELECTOR, "#author-text")) )]
+
+df['Comment'] = [comment.text for comment in wait.until(EC.presence_of_all_elements_located((By.CSS_SELECTOR, "#content-text")) )]
 
 # for comment in wait.until(EC.presence_of_all_elements_located((By.CSS_SELECTOR, "#content-text"))):
 #     data.append(comment.text)
 
-df = pd.DataFrame(data, columns=['comment'])
-df.head()
+print(df.head())