Manage cookie (#81)

* Add get cookies
* Add get cookie by name
* Add cookies

Author: atthaboon

Examples/browser-management/cookies.robot

@@ -4,13 +4,40 @@ Suite Teardown    Close Puppeteer
 Test Teardown    Close All Browser
 
 *** Variables ***
-${DEFAULT_BROWSER}    chrome
+${DEFAULT_BROWSER}    webkit
 
 
 *** Test Cases ***
+Get all cookies
+    ${BROWSER} =     Get variable value    ${BROWSER}    ${DEFAULT_BROWSER}
+    ${HEADLESS} =    Get variable value    ${HEADLESS}    ${False}
+    &{options} =    create dictionary   headless=${HEADLESS}
+    Open browser    https://www.blognone.com/    browser=${BROWSER}   options=${options}
+    &{cookies} =    Get Cookies
+    Should Not Be Empty    ${cookies}[_gat]
+    
+Get cookie
+    ${BROWSER} =     Get variable value    ${BROWSER}    ${DEFAULT_BROWSER}
+    ${HEADLESS} =    Get variable value    ${HEADLESS}    ${False}
+    &{options} =    create dictionary   headless=${HEADLESS}
+    Open browser    https://www.blognone.com/    browser=${BROWSER}   options=${options}
+    ${cookie value} =    Get Cookie    _gat
+    Should Not Be Empty    ${cookie value}
+    
+Add cookie
+    ${BROWSER} =     Get variable value    ${BROWSER}    ${DEFAULT_BROWSER}
+    ${HEADLESS} =    Get variable value    ${HEADLESS}    ${False}
+    &{options} =    create dictionary   headless=${HEADLESS}
+    Open browser    https://www.blognone.com/    browser=${BROWSER}   options=${options}
+    Add Cookie    Test    1111
+    &{cookies} =    Get Cookies
+    Should Be Equal As Strings    1111    ${cookies}[Test]    
+
 Delete all cookies
     ${BROWSER} =     Get variable value    ${BROWSER}    ${DEFAULT_BROWSER}
     ${HEADLESS} =    Get variable value    ${HEADLESS}    ${False}
     &{options} =    create dictionary   headless=${HEADLESS}
     Open browser    http://127.0.0.1:7272/basic-html-elements.html    browser=${BROWSER}   options=${options}
     Delete All Cookies
+
+    

PuppeteerLibrary/ikeywords/ibrowsermanagement_async.py

@@ -46,5 +46,14 @@ def unselect_iframe(self):
     ##############################
     # Cookies
     ##############################
+    async def get_cookie(self, name: str):
+        pass
+
+    async def get_cookies(self):
+        pass
+    
+    async def add_cookie(self, name: str, value: str):
+        pass
+
     async def delete_all_cookies(self):
         pass

PuppeteerLibrary/keywords/browsermanagement.py

@@ -173,6 +173,28 @@ def unselect_frame(self):
     ##############################
     # Cookies
     ##############################
+    @keyword
+    def get_cookie(self, name):
+        """ Get cookie by name
+
+            Returns Cookie value
+        """
+        return self.loop.run_until_complete(self.get_async_keyword_group().get_cookie(name))
+
+    @keyword
+    def get_cookies(self):
+        """ Get all cookies
+
+            Returns Dictionary for all cookies of the current page.
+        """
+        return self.loop.run_until_complete(self.get_async_keyword_group().get_cookies())
+
+    @keyword
+    def add_cookie(self, name, value):
+        """ Add cookie
+        """
+        return self.loop.run_until_complete(self.get_async_keyword_group().add_cookie(name, value))
+
     @keyword
     def delete_all_cookies(self):
         """ Deletes all cookies.

PuppeteerLibrary/playwright/async_keywords/playwright_browsermanagement.py

@@ -93,5 +93,24 @@ def unselect_iframe(self):
     ##############################
     # Cookies
     ##############################
+    async def get_cookie(self, name: str):
+        cookies = await self.get_cookies()
+        return cookies[name]
+
+    async def get_cookies(self):
+        cookies = await self.library_ctx.get_browser_context().contexts[0].cookies()
+        pairs = {}
+        for cookie in cookies:
+            pairs[cookie["name"]] = cookie["value"]
+        return pairs
+
+    async def add_cookie(self, name: str, value: str):
+        url = self.library_ctx.get_current_page().get_page().url
+        await self.library_ctx.get_browser_context().contexts[0].addCookies([{
+            'url': url,
+            'name': name,
+            'value': value
+        }])
+
     async def delete_all_cookies(self):
         await self.library_ctx.get_browser_context().contexts[0].clearCookies()

PuppeteerLibrary/puppeteer/async_keywords/puppeteer_browsermanagement.py

@@ -105,8 +105,24 @@ def unselect_iframe(self):
     ##############################
     # Cookies
     ##############################
+    async def get_cookie(self, name: str):
+        cookies = await self.get_cookies()
+        return cookies[name]
+    
+    async def get_cookies(self):
+        cookies = await self.library_ctx.get_current_page().get_page().cookies()
+        pairs = {}
+        for cookie in cookies:
+            pairs[cookie["name"]] = cookie["value"]
+        return pairs
+
+    async def add_cookie(self, name: str, value: str):
+        await self.library_ctx.get_current_page().get_page().setCookie({
+            'name': name,
+            'value': value,
+        })
+
     async def delete_all_cookies(self):
         cookies = await self.library_ctx.get_current_page().get_page().cookies()
         for cookie in cookies:
             await self.library_ctx.get_current_page().get_page().deleteCookie(cookie)
-