edit weekly

test/class/weekly.class.php

@@ -4,26 +4,101 @@ class weeklyTest
     public function __construct()
     {
          global $tester;
-         $this->objectModel = $tester->loadModel('weekly');
+         $this->objectModel  = $tester->loadModel('weekly');
+         $this->projectModel = $tester->loadModel('project');
     }
 
-    public function getPageNavTest($project, $date)
+
+    /**
+     * GetPageNav
+     *
+     * @param  int    $projectID
+     * @param  int    $date
+     * @access public
+     * @return string
+     */
+
+    public function getPageNavTest($projectID, $date)
     {
-        $objects = $this->objectModel->getPageNav($project, $date);
+        if($date == 1)
+        {
+            $date = date('Y-m-d');
+        }
+        elseif($date == 2)
+        {
+            $date = date('Y-m-d', strtotime(date('Y-m-d')."- 6 days"));
+        }
+        else
+        {
+            $date = date('Y-m-d', strtotime(date('Y-m-d') . "+ 6 days"));
+        }
+
+        $project = $this->projectModel->getById($projectID);
+        $pageNav = $this->objectModel->getPageNav($project, $date);
+
+        if(dao::isError()) return dao::getError();
+
+        $objects = strstr($pageNav,"</div>",true);
+        return $objects;
+    }
+
+    /**
+     * GetWeekPairs
+     *
+     * @param  int    $begin
+     * @access public
+     * @return array
+     */
+
+    public function getWeekPairsTest($begin, $end)
+    {
+        switch($begin)
+        {
+        case '1':
+           $begin = date('Y-m-d');
+           break;
+        case '2':
+           $begin = date('Y-m-d', strtotime(date('Y-m-d')."- 6 days"));
+           break;
+        case '3':
+           $begin = date('Y-m-d', strtotime(date('Y-m-d') . "+ 6 days"));
+           break;
+        case '':
+           $begin = '';
+           break;
+        }
+
+        switch($begin)
+        {
+        case '1':
+           $end = date('Y-m-d');
+           break;
+        case '2':
+           $end = date('Y-m-d', strtotime(date('Y-m-d')."- 6 days"));
+           break;
+        case '3':
+           $end = date('Y-m-d', strtotime(date('Y-m-d') . "+ 6 days"));
+           break;
+        case '':
+           $end = '';
+           break;
+        }
+
+        $objects = $this->objectModel->getWeekPairs($begin, $end);
 
         if(dao::isError()) return dao::getError();
 
         return $objects;
     }
 
-    public function getWeekPairsTest($begin, $end = '')
-    {
-        $objects = $this->objectModel->getWeekPairs($begin, $end = '');
-
-        if(dao::isError()) return dao::getError();
-
-        return $objects;
-    }
+    /**
+     * GetFromDB
+     *
+     * @param  int    $project
+     * @param  int    $date
+     * @access public
+     * @return object
+     */
 
     public function getFromDBTest($project, $date)
     {
@@ -34,15 +109,34 @@ class weeklyTest
         return $objects;
     }
 
+     /**
+     * Save data.
+     *
+     * @param  int    $project
+     * @param  int    $date
+     * @access public
+     * @return void
+     */
+
     public function saveTest($project, $date)
     {
         $objects = $this->objectModel->save($project, $date);
 
         if(dao::isError()) return dao::getError();
 
-        return $objects;
+        $weekly = $this->objectModel->getFromDB($project, $date);
+        return $weekly;
     }
 
+    /**
+     * GetWeekSN
+     *
+     * @param  int    $begin
+     * @param  int    $date
+     * @access public
+     * @return int
+     */
+
     public function getWeekSNTest($begin, $date)
     {
         $objects = $this->objectModel->getWeekSN($begin, $date);
@@ -52,6 +146,14 @@ class weeklyTest
         return $objects;
     }
 
+    /**
+     * Get monday for a date.
+     *
+     * @param  int $date
+     * @access public
+     * @return date
+     */
+
     public function getThisMondayTest($date)
     {
         $objects = $this->objectModel->getThisMonday($date);
@@ -61,6 +163,14 @@ class weeklyTest
         return $objects;
     }
 
+    /**
+     * GetThisSunday
+     *
+     * @param  int    $date
+     * @access public
+     * @return date
+     */
+
     public function getThisSundayTest($date)
     {
         $objects = $this->objectModel->getThisSunday($date);
@@ -70,6 +180,14 @@ class weeklyTest
         return $objects;
     }
 
+    /**
+     * GetLastDay
+     *
+     * @param  int    $date
+     * @access public
+     * @return string
+     */
+
     public function getLastDayTest($date)
     {
         $objects = $this->objectModel->getLastDay($date);
@@ -186,4 +304,4 @@ class weeklyTest
 
         return $objects;
     }
-}
+}