Тестирование: загрузка страницы с расписанием

modules/ssau_parser/connecter.go

@@ -101,6 +101,9 @@ func DownloadShedule(uri string, week int) (Page, error) {
 	var page Page
 	var err error
 
+	if len(uri) < 15 {
+		return page, fmt.Errorf("uri too short, maybe its wrong: %s", uri)
+	}
 	page.ID, err = strconv.ParseInt(uri[14:], 0, 64)
 	if err != nil {
 		return page, err
@@ -120,6 +123,10 @@ func DownloadShedule(uri string, week int) (Page, error) {
 		return page, err
 	}
 
+	if resp.StatusCode != 200 {
+		return page, fmt.Errorf("responce: %s", resp.Status)
+	}
+
 	page.Doc, err = goquery.NewDocumentFromReader(resp.Body)
 	if err != nil {
 		return page, err

modules/ssau_parser/ssau_parser_test.go

@@ -41,3 +41,34 @@ func pingQuery(query string, t *testing.T) {
 		log.Println(query, list)
 	}
 }
+
+var groupUri = []string{
+	"/rasp?groupId=530996168",
+	"/rasp?staffId=59915001",
+	"/aaa",
+	"/aaaaaaaaaaaaaa",
+	"/rasp?groupId=123",
+}
+var weeks = []int{
+	1,
+	2,
+	100,
+}
+
+func TestDownloadShedule(t *testing.T) {
+	// headURL = "https://ssau.ru"
+	headURL = "http://127.0.0.1:5000"
+	for _, uri := range groupUri {
+		for _, week := range weeks {
+			if _, err := DownloadShedule(uri, week); err != nil {
+				log.Println(err)
+			}
+		}
+	}
+	if _, err := DownloadSheduleById(530996168, true, 1); err != nil {
+		log.Println(err)
+	}
+	if _, err := DownloadSheduleById(59915001, false, 1); err != nil {
+		log.Println(err)
+	}
+}