def setUp(self): """ Create the list with which we'll test. """ # region play list setup. # Save and store a reference to two new plays. self.positivePlay = PlayTestHelper.makePlay(PlayTestHelper.Play1Vals) self.positivePlay.save() self.neutralPlay = PlayTestHelper.makePlay(PlayTestHelper.PlayPrepersistedVals) self.neutralPlay.save() self.negativePlay = PlayTestHelper.makePlay(PlayTestHelper.Play2Vals) self.negativePlay.save() self.allPlayList = PlayList() self.allPlayList.title = "All ReviewToRankTestCase Plays" self.allPlayList.url_title = self.PLAY_LIST_URL_TITLE self.allPlayList.is_dynamically_ordered = True self.allPlayList.num_to_order_dynamically = 10 # Order them all. self.allPlayList.save() positivePlayEntry = PlayListEntry() positivePlayEntry.play = self.positivePlay positivePlayEntry.play_list = self.allPlayList positivePlayEntry.play_list_order = None positivePlayEntry.save() neutralPlayEntry = PlayListEntry() neutralPlayEntry.play = self.neutralPlay neutralPlayEntry.play_list = self.allPlayList neutralPlayEntry.play_list_order = 200 neutralPlayEntry.save() negativePlayEntry = PlayListEntry() negativePlayEntry.play = self.negativePlay negativePlayEntry.play_list = self.allPlayList negativePlayEntry.play_list_order = 1 negativePlayEntry.save() # We should now have a list of three plays. self.assertEqual(self.allPlayList.plays.count(), 3) # endregion # region review plays Review.createFromText(SentimentTestHelper.POSITIVE_REVIEW_TEXT, self.positivePlay.url_title) Review.createFromText(SentimentTestHelper.POSITIVE_REVIEW_TEXT, self.positivePlay.url_title) Review.createFromText(SentimentTestHelper.POSITIVE_REVIEW_TEXT, self.neutralPlay.url_title) Review.createFromText(SentimentTestHelper.NEGATIVE_REVIEW_TEXT, self.neutralPlay.url_title) Review.createFromText(SentimentTestHelper.NEGATIVE_REVIEW_TEXT, self.negativePlay.url_title) Review.createFromText(SentimentTestHelper.NEGATIVE_REVIEW_TEXT, self.negativePlay.url_title) # We should now have six reviews in the system. self.assertEqual(Review.objects.count(), 6) # endregion self.allPlayList.refresh_from_db() self.positivePlay.refresh_from_db() self.neutralPlay.refresh_from_db() self.negativePlay.refresh_from_db()
def testHappyCaseConversion(self): # Create (and save) the PlayList we're working with. theList = PlayList() theList.title = "Test Play List" theList.url_title = "test-play-list" theList.save() # Create (and save) a pair of plays to populate the Play play1 = PlayTestHelper.makePlay(PlayTestHelper.Play1Vals); play2 = PlayTestHelper.makePlay(PlayTestHelper.Play2Vals); play1.save() play2.save() # Create* a pair of entries to bind the plays to the list playListEntry1 = PlayListEntry() playListEntry1.play = play1 playListEntry1.play_list = theList playListEntry1.play_list_order = 1 playListEntry2 = PlayListEntry() playListEntry2.play = play2 playListEntry2.play_list = theList playListEntry2.play_list_order = 2 # *(and save) playListEntry1.save() playListEntry2.save() # Update the list, so that it picks up changes theList.refresh_from_db() # Convert it to API Fields apiList = theList.getApiFields() # Check things. self.assertEquals(len(apiList), 2)