Added a session id.
This commit is contained in:
		@@ -3,6 +3,7 @@
 | 
			
		||||
from urllib import request
 | 
			
		||||
from urllib.error import HTTPError
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class Page:
 | 
			
		||||
    """Class for getting an individual page."""
 | 
			
		||||
 | 
			
		||||
@@ -13,10 +14,17 @@ class Page:
 | 
			
		||||
    def request_url(self, url):
 | 
			
		||||
        """Make a page request."""
 | 
			
		||||
        try:
 | 
			
		||||
            self.res = request.urlopen(url)
 | 
			
		||||
            page = request.urlopen(url)
 | 
			
		||||
        except HTTPError as err:
 | 
			
		||||
            self.res = err
 | 
			
		||||
        else:
 | 
			
		||||
            with page:
 | 
			
		||||
                self.res = page
 | 
			
		||||
 | 
			
		||||
    def get_status_code(self):
 | 
			
		||||
        """Return the status code."""
 | 
			
		||||
        return self.res.status
 | 
			
		||||
 | 
			
		||||
    def get_header(self, header):
 | 
			
		||||
        """Did a session id get set."""
 | 
			
		||||
        return self.res.headers[header]
 | 
			
		||||
 
 | 
			
		||||
							
								
								
									
										25
									
								
								tests/step_defs/test_session.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										25
									
								
								tests/step_defs/test_session.py
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,25 @@
 | 
			
		||||
"""tests for server sessions."""
 | 
			
		||||
 | 
			
		||||
from pytest_bdd import given, scenarios, then, when
 | 
			
		||||
 | 
			
		||||
scenarios("../features/session.feature")
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@given("a running server")
 | 
			
		||||
def start_server(server):
 | 
			
		||||
    """Make a running server."""
 | 
			
		||||
    server.set_safe_port()
 | 
			
		||||
    server.start()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@when("the home page is accessed")
 | 
			
		||||
def access_home_page(server, page):
 | 
			
		||||
    """Access the home page."""
 | 
			
		||||
    url = f"http://{server.base_url}/"
 | 
			
		||||
    page.request_url(url)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@then("there is a session id")
 | 
			
		||||
def confirm_session(page):
 | 
			
		||||
    """Confirm session id exists."""
 | 
			
		||||
    assert page.get_header("set-cookie")
 | 
			
		||||
		Reference in New Issue
	
	Block a user