Exemplo n.º 1
0
    def setUp(self):
        atms_api.main.app.config['TESTING'] = True
        self.app = atms_api.main.app.test_client()
        self.auth = AuthFactory()
        self.Config = Config
        
        # Define user account to use for testing.
        self.username = "******"
        self.password = "******"

        # Set-up/register a new username/token with AuthFactory.
        self.token = self.auth.new(self.username)
Exemplo n.º 2
0
class TestAtmsApi(unittest.TestCase):

    def setUp(self):
        atms_api.main.app.config['TESTING'] = True
        self.app = atms_api.main.app.test_client()
        self.auth = AuthFactory()
        self.Config = Config
        
        # Define user account to use for testing.
        self.username = "******"
        self.password = "******"

        # Set-up/register a new username/token with AuthFactory.
        self.token = self.auth.new(self.username)

    def testSetUp(self):
        """
        Test the set up of the test client before proceeding.
        """
        assert self.app
        assert self.auth
        assert self.Config
        assert self.username
        assert self.password
        assert self.token
        assert self.auth.check(self.username, self.token)
    
    #
    # Main Tests
    #

    def testStatus(self):
        """
        Simple starter test to make sure the main app definition is responding.
        """

        resp = self.app.get("/status")

        data = json.loads(resp.get_data())

        assert "status" in data
        assert "message" in data
        assert data["status"] == "0"

    #
    # Core Tests
    #

    def testCoreStatus(self):
        """
        Simple test to make sure api_core blue print was registered.
        """

        resp = self.app.get("/core/status")

        data = json.loads(resp.get_data())
        
        assert "status" in data
        assert "message" in data

    def testCoreConfig(self):
        """
        Test the configuration route.
        """

        resp = self.app.get("/core/config",
                            headers={"X-Core-Token": "%s-%s" % 
                                     (self.username, self.token)})

        data = json.loads(resp.get_data())

        assert "atmsapi" in data
        assert "webrdp" in data

    def testCoreLoginDelete(self):
        """
        Since there was a login registered during initialization, test the log
        out functionality first.
        """

        resp = self.app.delete("/core/login",
                               headers={"X-Core-Token": "%s-%s" % 
                                        (self.username, self.token)})
        
        data = json.loads(resp.get_data())

        assert "status" in data
        assert "message" in data
        assert "username" in data
        assert data["status"] == "0"
        assert data["username"] == "admin"

    def testCoreLoginPost(self):
        """
        Re-login and verify new credentials.
        """

        jsonData = str(
            {
                "username": self.username, 
                "password": self.password
            }
        )

        resp = self.app.post("/core/login",
                             content_type="application/json",
                             data=jsonData.replace("'",'"'))

        data = json.loads(resp.get_data())

        self.token = data["token"].split("-")[1]

        assert "status" in data
        assert "message" in data
        assert "token" in data
        assert self.auth.check(self.username, self.token)

    def testCorePasswordPost(self):
        """
        Test password recovery.
        """

        jsonData = str(
            {
                "username": self.username,
            }
        )
        resp = self.app.post("/core/password",
                             content_type="application/json",
                             data=jsonData.replace("'",'"'))

        data = json.loads(resp.get_data())

        assert "status" in data
        # Can't check status code because admin user is not configured for
        # e-mailing in test DB.
        assert "message" in data

    def testCorePasswordPut(self):
#        """
#        Test password change.
#        """
        
        jsonData = str(
            {
                "username": self.username,
                "oldpassword": self.password,
                "newpassword": "******"
            }
        )

        resp = self.app.put("/core/password",
                            content_type="application/json",
                            headers={"X-Core-Token": "%s-%s" %
                                     (self.username, self.token)},
                            data=jsonData.replace("'",'"'))

        data = json.loads(resp.get_data())

        assert "status" in data
        assert data["status"] == "0"
        assert "message" in data

    def testCorePasswordPut2(self):
        """
        Revert the password change in the previous test.
        """

        jsonData = str(
            {
                "username": self.username,
                "oldpassword": "******",
                "newpassword": self.password
            }
        )

        resp = self.app.put("/core/password",
                            content_type="application/json",
                            headers={"X-Core-Token": "%s-%s" %
                                     (self.username, self.token)},
                            data=jsonData.replace("'",'"'))

        data = json.loads(resp.get_data())

        assert "status" in data
        assert data["status"] == "0"
        assert "message" in data

    def testCoreApplicationsGet(self):
        """
        Test the retrieval of a list of applications.
        """

        resp = self.app.get("/core/applications",
                               headers={"X-Core-Token": "%s-%s" % 
                                        (self.username, self.token)})
        
        data = json.loads(resp.get_data())

        assert "status" in data
        assert data["status"] == "0"
        assert "message" in data
        assert "application_list" in data
        assert "application_count" in data