From e8763ef71520f36365c60f6507ddd040f56aa405 Mon Sep 17 00:00:00 2001
From: Mat Peterson <mpeterson@edx.org>
Date: Wed, 13 Aug 2014 18:16:39 +0000
Subject: [PATCH] Do TODOs in test_contentstore.py

---
 .../contentstore/tests/test_contentstore.py         | 13 ++++++-------
 cms/templates/index.html                            |  6 +++---
 2 files changed, 9 insertions(+), 10 deletions(-)

diff --git a/cms/djangoapps/contentstore/tests/test_contentstore.py b/cms/djangoapps/contentstore/tests/test_contentstore.py
index 911d1a31bb..caf605fda3 100644
--- a/cms/djangoapps/contentstore/tests/test_contentstore.py
+++ b/cms/djangoapps/contentstore/tests/test_contentstore.py
@@ -1584,12 +1584,11 @@ class RerunCourseTest(ContentStoreTestCase):
 
     def create_course_listing_html(self, course_key):
         """Creates html fragment that is created for the given course_key in the course listing section"""
-        return '<a class="course-link" href="/course/{}"'.format(course_key)
+        return 'data-test-course="{}/{}/{}"'.format(course_key.org, course_key.course, course_key.run)
 
     def create_unsucceeded_course_action_html(self, course_key):
         """Creates html fragment that is created for the given course_key in the unsucceeded course action section"""
-        # TODO Update this once the Rerun UI LMS-11011 is implemented.
-        return '<div class="unsucceeded-course-action" href="/course/{}"'.format(course_key)
+        return 'data-test-unsucceeded="{}/{}/{}"'.format(course_key.org, course_key.course, course_key.run)
 
     def assertInCourseListing(self, course_key):
         """
@@ -1597,7 +1596,7 @@ class RerunCourseTest(ContentStoreTestCase):
         and NOT in the unsucceeded course action section of the html.
         """
         course_listing_html = self.client.get_html('/course/')
-        self.assertIn(course_key.run, course_listing_html.content)
+        self.assertIn(self.create_course_listing_html(course_key), course_listing_html.content)
         self.assertNotIn(self.create_unsucceeded_course_action_html(course_key), course_listing_html.content)
 
     def assertInUnsucceededCourseActions(self, course_key):
@@ -1606,8 +1605,8 @@ class RerunCourseTest(ContentStoreTestCase):
         and NOT in the accessible course listing section of the html.
         """
         course_listing_html = self.client.get_html('/course/')
-        self.assertNotIn(course_key.run, course_listing_html.content)
-        # TODO Verify the course is in the unsucceeded listing once LMS-11011 is implemented.
+        self.assertNotIn(self.create_course_listing_html(course_key), course_listing_html.content)
+        self.assertIn(self.create_unsucceeded_course_action_html(course_key), course_listing_html.content)
 
     def test_rerun_course_success(self):
 
@@ -1650,7 +1649,7 @@ class RerunCourseTest(ContentStoreTestCase):
         self.assertInCourseListing(existent_course_key)
 
         # Verify that the failed course is NOT in the course listings
-        self.assertInUnsucceededCourseActions(non_existent_course_key)
+        self.assertInUnsucceededCourseActions(destination_course_key)
 
     def test_rerun_course_fail_duplicate_course(self):
         existent_course_key = CourseFactory.create().id
diff --git a/cms/templates/index.html b/cms/templates/index.html
index 82497ed0b0..bda00eb77e 100644
--- a/cms/templates/index.html
+++ b/cms/templates/index.html
@@ -140,7 +140,7 @@ require(["domReady!", "jquery", "jquery.form", "js/index"], function(doc, $) {
             %for display_name, org, num, run, state_failed, state_in_progress, dismiss_link in sorted(unsucceeded_course_actions, key=lambda s: s[0].lower() if s[0] is not None else ''):
             <!-- STATE: re-run is processing -->
             %if state_in_progress:
-            <li class="wrapper-course has-status">
+            <li class="wrapper-course has-status" data-test-unsucceeded="${org}/${num}/${run}">
               <div class="course-item course-rerun is-processing">
                 <div class="course-details" href="#">
                   <h3 class="course-title">${display_name}</h3>
@@ -178,7 +178,7 @@ require(["domReady!", "jquery", "jquery.form", "js/index"], function(doc, $) {
 
             <!-- STATE: re-run has error -->
             %if state_failed:
-            <li class="wrapper-course has-status">
+            <li class="wrapper-course has-status" data-test-unsucceeded="${org}/${num}/${run}">
               <div class="course-item course-rerun has-error">
                 <div class="course-details" href="#">
                   <h3 class="course-title">${display_name}</h3>
@@ -229,7 +229,7 @@ require(["domReady!", "jquery", "jquery.form", "js/index"], function(doc, $) {
       <div class="courses">
         <ul class="list-courses">
           %for course, url, lms_link, rerun_link, org, num, run in sorted(courses, key=lambda s: s[0].lower() if s[0] is not None else ''):
-          <li class="course-item">
+          <li class="course-item" data-test-course="${org}/${num}/${run}">
             <a class="course-link" href="${url}">
               <h3 class="course-title">${course}</h3>