class TestDistrictTeamsApiController(unittest2.TestCase):
def setUp(self):
app = webapp2.WSGIApplication([webapp2.Route(r'/<district_abbrev:>/<year:([0-9]*)>', ApiDistrictTeamsController, methods=['GET'])], debug=True)
self.testapp = webtest.TestApp(app)
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.testbed.init_urlfetch_stub()
self.testbed.init_memcache_stub()
ndb.get_context().clear_cache() # Prevent data from leaking between tests
self.testbed.init_taskqueue_stub(root_path=".")
self.team = Team(
id="frc281",
name="Michelin / Caterpillar / Greenville Technical College /\
jcpenney / Baldor / ASME / Gastroenterology Associates /\
Laserflex South & Greenville County Schools & Greenville\
Technical Charter High School",
team_number=281,
nickname="EnTech GreenVillians",
city="Greenville",
state_prov="SC",
country="USA",
website="www.entech.org",
motto = "Infiltrating Young Minds One Robot at a Time",
)
self.district_team = DistrictTeam(
id="2015ne_frc281",
team=self.team.key,
year=2015,
district=3
)
self.team.put()
self.district_team.put()
def tearDown(self):
self.testbed.deactivate()
def assertTeamJson(self, team):
team = team[0]
self.assertEqual(team["key"], self.team.key_name)
self.assertEqual(team["team_number"], self.team.team_number)
self.assertEqual(team["nickname"], self.team.nickname)
self.assertEqual(team["location"], self.team.location)
self.assertEqual(team["locality"], "Greenville")
self.assertEqual(team["country_name"], "USA")
self.assertEqual(team["region"], "SC")
self.assertEqual(team["website"], self.team.website)
self.assertEqual(team["motto"], self.team.motto)
def testDistrictsApi(self):
response = self.testapp.get('/ne/2015', headers={"X-TBA-App-Id": "tba-tests:team-districts-controller-test:v01"})
teams = json.loads(response.body)
self.assertTeamJson(teams)
class TestDistrictTeamsApiController(unittest2.TestCase):
def setUp(self):
app = webapp2.WSGIApplication(
[webapp2.Route(r"/<district_abbrev:>/<year:([0-9]*)>", ApiDistrictTeamsController, methods=["GET"])],
debug=True,
)
self.testapp = webtest.TestApp(app)
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.testbed.init_urlfetch_stub()
self.testbed.init_memcache_stub()
ndb.get_context().clear_cache() # Prevent data from leaking between tests
self.testbed.init_taskqueue_stub(root_path=".")
self.team = Team(
id="frc281",
name="Michelin / Caterpillar / Greenville Technical College /\
jcpenney / Baldor / ASME / Gastroenterology Associates /\
Laserflex South & Greenville County Schools & Greenville\
Technical Charter High School",
team_number=281,
nickname="EnTech GreenVillians",
city="Greenville",
state_prov="SC",
country="USA",
website="www.entech.org",
motto="Infiltrating Young Minds One Robot at a Time",
)
self.district_team = DistrictTeam(id="2015ne_frc281", team=self.team.key, year=2015, district=3)
self.team.put()
self.district_team.put()
def tearDown(self):
self.testbed.deactivate()
def assertTeamJson(self, team):
team = team[0]
self.assertEqual(team["key"], self.team.key_name)
self.assertEqual(team["team_number"], self.team.team_number)
self.assertEqual(team["nickname"], self.team.nickname)
self.assertEqual(team["location"], self.team.location)
self.assertEqual(team["locality"], "Greenville")
self.assertEqual(team["country_name"], "USA")
self.assertEqual(team["region"], "SC")
self.assertEqual(team["website"], self.team.website)
self.assertEqual(team["motto"], self.team.motto)
def testDistrictsApi(self):
response = self.testapp.get(
"/ne/2015", headers={"X-TBA-App-Id": "tba-tests:team-districts-controller-test:v01"}
)
teams = json.loads(response.body)
self.assertTeamJson(teams)
class TestTeamHistoryDistrictsApiController(unittest2.TestCase):
def setUp(self):
app = webapp2.WSGIApplication([
webapp2.Route(r'/<team_key:>',
ApiTeamHistoryDistrictsController,
methods=['GET'])
],
debug=True)
self.testapp = webtest.TestApp(app)
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.testbed.init_urlfetch_stub()
self.testbed.init_memcache_stub()
ndb.get_context().clear_cache(
) # Prevent data from leaking between tests
self.testbed.init_taskqueue_stub(root_path=".")
self.team = Team(
id="frc1124",
name="UberBots",
team_number=1124,
nickname="UberBots",
)
self.district_team = DistrictTeam(id="2015ne_frc1124",
team=self.team.key,
year=2015,
district_key=ndb.Key(
District, '2015ne'))
self.district = District(
id='2015ne',
year=2015,
)
self.team.put()
self.district_team.put()
self.district.put()
def tearDown(self):
self.testbed.deactivate()
def testDistrictsApi(self):
response = self.testapp.get(
'/frc1124',
headers={
"X-TBA-App-Id":
"tba-tests:team-history-districts-controller-test:v01"
})
district_dict = json.loads(response.body)
self.assertTrue("2015" in district_dict)
district_key = district_dict["2015"]
self.assertEqual(district_key, "2015ne")
class TestTeamHistoryDistrictsApiController(unittest2.TestCase):
def setUp(self):
app = webapp2.WSGIApplication([webapp2.Route(r'/<team_key:>', ApiTeamHistoryDistrictsController, methods=['GET'])], debug=True)
self.testapp = webtest.TestApp(app)
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.testbed.init_urlfetch_stub()
self.testbed.init_memcache_stub()
ndb.get_context().clear_cache() # Prevent data from leaking between tests
self.testbed.init_taskqueue_stub(root_path=".")
self.team = Team(
id="frc1124",
name="UberBots",
team_number=1124,
nickname="UberBots"
)
self.district_team = DistrictTeam(
id="2015ne_frc1124",
team=self.team.key,
year=2015,
district_key=ndb.Key(District, '2015ne')
)
self.district = District(
id='2015ne',
year=2015
)
self.team.put()
self.district_team.put()
self.district.put()
def tearDown(self):
self.testbed.deactivate()
def testDistrictsApi(self):
response = self.testapp.get('/frc1124', headers={"X-TBA-App-Id": "tba-tests:team-history-districts-controller-test:v01"})
district_dict = json.loads(response.body)
self.assertTrue("2015" in district_dict)
district_key = district_dict["2015"]
self.assertEqual(district_key, "2015ne")
class TestDatabaseCacheClearer(unittest2.TestCase):
def setUp(self):
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.testbed.init_memcache_stub()
self.testbed.init_taskqueue_stub(root_path=".")
self.eventteam_2015casj_frc254 = EventTeam(
id='2015casj_frc254',
event=ndb.Key(Event, '2015casj'),
team=ndb.Key(Team, 'frc254'),
year=2015,
)
self.eventteam_2015cama_frc604 = EventTeam(
id='2015cama_frc604',
event=ndb.Key(Event, '2015cama'),
team=ndb.Key(Team, 'frc604'),
year=2015,
)
self.eventteam_2010cama_frc604 = EventTeam(
id='2010cama_frc604',
event=ndb.Key(Event, '2010cama'),
team=ndb.Key(Team, 'frc604'),
year=2010,
)
self.eventteam_2015casj_frc254.put()
self.eventteam_2015cama_frc604.put()
self.eventteam_2010cama_frc604.put()
self.districtteam_2015fim_frc254 = DistrictTeam(
id='2015fim_frc254',
district=DistrictType.MICHIGAN,
team=ndb.Key(Team, 'frc254'),
year=2015,
)
self.districtteam_2015mar_frc604 = DistrictTeam(
id='2015mar_frc604',
district=DistrictType.MID_ATLANTIC,
team=ndb.Key(Team, 'frc604'),
year=2015,
)
self.districtteam_2015fim_frc254.put()
self.districtteam_2015mar_frc604.put()
def tearDown(self):
self.testbed.deactivate()
def test_award_updated(self):
affected_refs = {
'event': {ndb.Key(Event, '2015casj'), ndb.Key(Event, '2015cama')},
'team_list': {ndb.Key(Team, 'frc254'), ndb.Key(Team, 'frc604')},
'year': {2014, 2015}
}
cache_keys = [q.cache_key for q in get_affected_queries.award_updated(affected_refs)]
self.assertEqual(len(cache_keys), 12)
self.assertTrue(EventAwardsQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventAwardsQuery('2015cama').cache_key in cache_keys)
self.assertTrue(TeamAwardsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamAwardsQuery('frc604').cache_key in cache_keys)
self.assertTrue(TeamYearAwardsQuery('frc254', 2014).cache_key in cache_keys)
self.assertTrue(TeamYearAwardsQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(TeamYearAwardsQuery('frc604', 2014).cache_key in cache_keys)
self.assertTrue(TeamYearAwardsQuery('frc604', 2015).cache_key in cache_keys)
self.assertTrue(TeamEventAwardsQuery('frc254', '2015casj').cache_key in cache_keys)
self.assertTrue(TeamEventAwardsQuery('frc254', '2015cama').cache_key in cache_keys)
self.assertTrue(TeamEventAwardsQuery('frc604', '2015casj').cache_key in cache_keys)
self.assertTrue(TeamEventAwardsQuery('frc604', '2015cama').cache_key in cache_keys)
def test_event_updated(self):
affected_refs = {
'key': {ndb.Key(Event, '2015casj'), ndb.Key(Event, '2015cama')},
'year': {2014, 2015},
'event_district_key': {'2015fim', '2014mar'}
}
cache_keys = [q.cache_key for q in get_affected_queries.event_updated(affected_refs)]
self.assertEqual(len(cache_keys), 8)
self.assertTrue(EventListQuery(2014).cache_key in cache_keys)
self.assertTrue(EventListQuery(2015).cache_key in cache_keys)
self.assertTrue(DistrictEventsQuery('2015fim').cache_key in cache_keys)
self.assertTrue(DistrictEventsQuery('2014mar').cache_key in cache_keys)
self.assertTrue(TeamEventsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamEventsQuery('frc604').cache_key in cache_keys)
self.assertTrue(TeamYearEventsQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(TeamYearEventsQuery('frc604', 2015).cache_key in cache_keys)
def test_match_updated(self):
affected_refs = {
'event': {ndb.Key(Event, '2015casj'), ndb.Key(Event, '2015cama')},
'team_keys': {ndb.Key(Team, 'frc254'), ndb.Key(Team, 'frc604')},
'year': {2014, 2015},
}
cache_keys = [q.cache_key for q in get_affected_queries.match_updated(affected_refs)]
self.assertEqual(len(cache_keys), 10)
self.assertTrue(EventMatchesQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventMatchesQuery('2015cama').cache_key in cache_keys)
self.assertTrue(TeamEventMatchesQuery('frc254', '2015casj').cache_key in cache_keys)
self.assertTrue(TeamEventMatchesQuery('frc254', '2015cama').cache_key in cache_keys)
self.assertTrue(TeamEventMatchesQuery('frc604', '2015casj').cache_key in cache_keys)
self.assertTrue(TeamEventMatchesQuery('frc604', '2015cama').cache_key in cache_keys)
self.assertTrue(TeamYearMatchesQuery('frc254', 2014).cache_key in cache_keys)
self.assertTrue(TeamYearMatchesQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(TeamYearMatchesQuery('frc604', 2014).cache_key in cache_keys)
self.assertTrue(TeamYearMatchesQuery('frc604', 2015).cache_key in cache_keys)
def test_media_updated(self):
affected_refs = {
'references': {ndb.Key(Team, 'frc254'), ndb.Key(Team, 'frc604')},
'year': {2014, 2015},
}
cache_keys = [q.cache_key for q in get_affected_queries.media_updated(affected_refs)]
self.assertEqual(len(cache_keys), 10)
self.assertTrue(TeamYearMediaQuery('frc254', 2014).cache_key in cache_keys)
self.assertTrue(TeamYearMediaQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(TeamSocialMediaQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamYearMediaQuery('frc604', 2014).cache_key in cache_keys)
self.assertTrue(TeamYearMediaQuery('frc604', 2015).cache_key in cache_keys)
self.assertTrue(TeamSocialMediaQuery('frc604').cache_key in cache_keys)
self.assertTrue(EventTeamsMediasQuery('2015cama').cache_key in cache_keys)
self.assertTrue(EventTeamsMediasQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventTeamsPreferredMediasQuery('2015cama').cache_key in cache_keys)
self.assertTrue(EventTeamsPreferredMediasQuery('2015casj').cache_key in cache_keys)
def test_robot_updated(self):
affected_refs = {
'team': {ndb.Key(Team, 'frc254'), ndb.Key(Team, 'frc604')},
}
cache_keys = [q.cache_key for q in get_affected_queries.robot_updated(affected_refs)]
self.assertEqual(len(cache_keys), 2)
self.assertTrue(TeamRobotsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamRobotsQuery('frc604').cache_key in cache_keys)
def test_team_updated(self):
affected_refs = {
'key': {ndb.Key(Team, 'frc254'), ndb.Key(Team, 'frc604')},
}
cache_keys = [q.cache_key for q in get_affected_queries.team_updated(affected_refs)]
self.assertEqual(len(cache_keys), 10)
self.assertTrue(TeamListQuery(0).cache_key in cache_keys)
self.assertTrue(TeamListQuery(1).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2015, 0).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2015, 1).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2010, 1).cache_key in cache_keys)
self.assertTrue(DistrictTeamsQuery('2015fim').cache_key in cache_keys)
self.assertTrue(DistrictTeamsQuery('2015mar').cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2015cama').cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2010cama').cache_key in cache_keys)
def test_eventteam_updated(self):
affected_refs = {
'event': {ndb.Key(Event, '2015casj'), ndb.Key(Event, '2015cama')},
'team': {ndb.Key(Team, 'frc254'), ndb.Key(Team, 'frc604')},
'year': {2014, 2015}
}
cache_keys = [q.cache_key for q in get_affected_queries.eventteam_updated(affected_refs)]
self.assertEqual(len(cache_keys), 18)
self.assertTrue(TeamEventsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamEventsQuery('frc604').cache_key in cache_keys)
self.assertTrue(TeamParticipationQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamParticipationQuery('frc604').cache_key in cache_keys)
self.assertTrue(TeamYearEventsQuery('frc254', 2014).cache_key in cache_keys)
self.assertTrue(TeamYearEventsQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(TeamYearEventsQuery('frc604', 2014).cache_key in cache_keys)
self.assertTrue(TeamYearEventsQuery('frc604', 2015).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2014, 0).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2014, 1).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2015, 0).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2015, 1).cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2015cama').cache_key in cache_keys)
self.assertTrue(EventTeamsMediasQuery('2015cama').cache_key in cache_keys)
self.assertTrue(EventTeamsMediasQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventTeamsPreferredMediasQuery('2015cama').cache_key in cache_keys)
self.assertTrue(EventTeamsPreferredMediasQuery('2015casj').cache_key in cache_keys)
def test_districtteam_updated(self):
affected_refs = {
'district_key': {'2015fim', '2015mar'},
'team': {ndb.Key(Team, 'frc254'), ndb.Key(Team, 'frc604')}
}
cache_keys = [q.cache_key for q in get_affected_queries.districtteam_updated(affected_refs)]
self.assertEqual(len(cache_keys), 4)
self.assertTrue(DistrictTeamsQuery('2015fim').cache_key in cache_keys)
self.assertTrue(DistrictTeamsQuery('2015mar').cache_key in cache_keys)
self.assertTrue(TeamDistrictsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamDistrictsQuery('frc604').cache_key in cache_keys)
class TestDatabaseCacheClearer(unittest2.TestCase):
def setUp(self):
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.testbed.init_memcache_stub()
self.testbed.init_taskqueue_stub(root_path=".")
self.eventteam_2015casj_frc254 = EventTeam(
id='2015casj_frc254',
event=ndb.Key(Event, '2015casj'),
team=ndb.Key(Team, 'frc254'),
year=2015,
)
self.eventteam_2015cama_frc604 = EventTeam(
id='2015cama_frc604',
event=ndb.Key(Event, '2015cama'),
team=ndb.Key(Team, 'frc604'),
year=2015,
)
self.eventteam_2010cama_frc604 = EventTeam(
id='2010cama_frc604',
event=ndb.Key(Event, '2010cama'),
team=ndb.Key(Team, 'frc604'),
year=2010,
)
self.eventteam_2015casj_frc254.put()
self.eventteam_2015cama_frc604.put()
self.eventteam_2010cama_frc604.put()
self.districtteam_2015fim_frc254 = DistrictTeam(
id='2015fim_frc254',
district=DistrictType.MICHIGAN,
team=ndb.Key(Team, 'frc254'),
year=2015,
)
self.districtteam_2015mar_frc604 = DistrictTeam(
id='2015mar_frc604',
district=DistrictType.MID_ATLANTIC,
team=ndb.Key(Team, 'frc604'),
year=2015,
)
self.districtteam_2015fim_frc254.put()
self.districtteam_2015mar_frc604.put()
def tearDown(self):
self.testbed.deactivate()
def test_award_updated(self):
affected_refs = {
'event': {ndb.Key(Event, '2015casj'),
ndb.Key(Event, '2015cama')},
'team_list': {ndb.Key(Team, 'frc254'),
ndb.Key(Team, 'frc604')},
'year': {2014, 2015}
}
cache_keys = [
q.cache_key
for q in get_affected_queries.award_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 12)
self.assertTrue(EventAwardsQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventAwardsQuery('2015cama').cache_key in cache_keys)
self.assertTrue(TeamAwardsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamAwardsQuery('frc604').cache_key in cache_keys)
self.assertTrue(
TeamYearAwardsQuery('frc254', 2014).cache_key in cache_keys)
self.assertTrue(
TeamYearAwardsQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(
TeamYearAwardsQuery('frc604', 2014).cache_key in cache_keys)
self.assertTrue(
TeamYearAwardsQuery('frc604', 2015).cache_key in cache_keys)
self.assertTrue(
TeamEventAwardsQuery('frc254', '2015casj').cache_key in cache_keys)
self.assertTrue(
TeamEventAwardsQuery('frc254', '2015cama').cache_key in cache_keys)
self.assertTrue(
TeamEventAwardsQuery('frc604', '2015casj').cache_key in cache_keys)
self.assertTrue(
TeamEventAwardsQuery('frc604', '2015cama').cache_key in cache_keys)
def test_event_updated(self):
affected_refs = {
'key': {ndb.Key(Event, '2015casj'),
ndb.Key(Event, '2015cama')},
'year': {2014, 2015},
'event_district_key': {'2015fim', '2014mar'}
}
cache_keys = [
q.cache_key
for q in get_affected_queries.event_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 8)
self.assertTrue(EventListQuery(2014).cache_key in cache_keys)
self.assertTrue(EventListQuery(2015).cache_key in cache_keys)
self.assertTrue(DistrictEventsQuery('2015fim').cache_key in cache_keys)
self.assertTrue(DistrictEventsQuery('2014mar').cache_key in cache_keys)
self.assertTrue(TeamEventsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamEventsQuery('frc604').cache_key in cache_keys)
self.assertTrue(
TeamYearEventsQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(
TeamYearEventsQuery('frc604', 2015).cache_key in cache_keys)
def test_match_updated(self):
affected_refs = {
'event': {ndb.Key(Event, '2015casj'),
ndb.Key(Event, '2015cama')},
'team_keys': {ndb.Key(Team, 'frc254'),
ndb.Key(Team, 'frc604')},
'year': {2014, 2015},
}
cache_keys = [
q.cache_key
for q in get_affected_queries.match_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 10)
self.assertTrue(EventMatchesQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventMatchesQuery('2015cama').cache_key in cache_keys)
self.assertTrue(
TeamEventMatchesQuery('frc254', '2015casj').cache_key in
cache_keys)
self.assertTrue(
TeamEventMatchesQuery('frc254', '2015cama').cache_key in
cache_keys)
self.assertTrue(
TeamEventMatchesQuery('frc604', '2015casj').cache_key in
cache_keys)
self.assertTrue(
TeamEventMatchesQuery('frc604', '2015cama').cache_key in
cache_keys)
self.assertTrue(
TeamYearMatchesQuery('frc254', 2014).cache_key in cache_keys)
self.assertTrue(
TeamYearMatchesQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(
TeamYearMatchesQuery('frc604', 2014).cache_key in cache_keys)
self.assertTrue(
TeamYearMatchesQuery('frc604', 2015).cache_key in cache_keys)
def test_media_updated(self):
affected_refs = {
'references': {ndb.Key(Team, 'frc254'),
ndb.Key(Team, 'frc604')},
'year': {2014, 2015},
}
cache_keys = [
q.cache_key
for q in get_affected_queries.media_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 10)
self.assertTrue(
TeamYearMediaQuery('frc254', 2014).cache_key in cache_keys)
self.assertTrue(
TeamYearMediaQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(TeamSocialMediaQuery('frc254').cache_key in cache_keys)
self.assertTrue(
TeamYearMediaQuery('frc604', 2014).cache_key in cache_keys)
self.assertTrue(
TeamYearMediaQuery('frc604', 2015).cache_key in cache_keys)
self.assertTrue(TeamSocialMediaQuery('frc604').cache_key in cache_keys)
self.assertTrue(
EventTeamsMediasQuery('2015cama').cache_key in cache_keys)
self.assertTrue(
EventTeamsMediasQuery('2015casj').cache_key in cache_keys)
self.assertTrue(
EventTeamsPreferredMediasQuery('2015cama').cache_key in cache_keys)
self.assertTrue(
EventTeamsPreferredMediasQuery('2015casj').cache_key in cache_keys)
def test_robot_updated(self):
affected_refs = {
'team': {ndb.Key(Team, 'frc254'),
ndb.Key(Team, 'frc604')},
}
cache_keys = [
q.cache_key
for q in get_affected_queries.robot_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 2)
self.assertTrue(TeamRobotsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamRobotsQuery('frc604').cache_key in cache_keys)
def test_team_updated(self):
affected_refs = {
'key': {ndb.Key(Team, 'frc254'),
ndb.Key(Team, 'frc604')},
}
cache_keys = [
q.cache_key
for q in get_affected_queries.team_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 10)
self.assertTrue(TeamListQuery(0).cache_key in cache_keys)
self.assertTrue(TeamListQuery(1).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2015, 0).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2015, 1).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2010, 1).cache_key in cache_keys)
self.assertTrue(DistrictTeamsQuery('2015fim').cache_key in cache_keys)
self.assertTrue(DistrictTeamsQuery('2015mar').cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2015cama').cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2010cama').cache_key in cache_keys)
def test_eventteam_updated(self):
affected_refs = {
'event': {ndb.Key(Event, '2015casj'),
ndb.Key(Event, '2015cama')},
'team': {ndb.Key(Team, 'frc254'),
ndb.Key(Team, 'frc604')},
'year': {2014, 2015}
}
cache_keys = [
q.cache_key
for q in get_affected_queries.eventteam_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 18)
self.assertTrue(TeamEventsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamEventsQuery('frc604').cache_key in cache_keys)
self.assertTrue(
TeamParticipationQuery('frc254').cache_key in cache_keys)
self.assertTrue(
TeamParticipationQuery('frc604').cache_key in cache_keys)
self.assertTrue(
TeamYearEventsQuery('frc254', 2014).cache_key in cache_keys)
self.assertTrue(
TeamYearEventsQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(
TeamYearEventsQuery('frc604', 2014).cache_key in cache_keys)
self.assertTrue(
TeamYearEventsQuery('frc604', 2015).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2014, 0).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2014, 1).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2015, 0).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2015, 1).cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2015cama').cache_key in cache_keys)
self.assertTrue(
EventTeamsMediasQuery('2015cama').cache_key in cache_keys)
self.assertTrue(
EventTeamsMediasQuery('2015casj').cache_key in cache_keys)
self.assertTrue(
EventTeamsPreferredMediasQuery('2015cama').cache_key in cache_keys)
self.assertTrue(
EventTeamsPreferredMediasQuery('2015casj').cache_key in cache_keys)
def test_districtteam_updated(self):
affected_refs = {
'district_key': {'2015fim', '2015mar'},
'team': {ndb.Key(Team, 'frc254'),
ndb.Key(Team, 'frc604')}
}
cache_keys = [
q.cache_key
for q in get_affected_queries.districtteam_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 4)
self.assertTrue(DistrictTeamsQuery('2015fim').cache_key in cache_keys)
self.assertTrue(DistrictTeamsQuery('2015mar').cache_key in cache_keys)
self.assertTrue(TeamDistrictsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamDistrictsQuery('frc604').cache_key in cache_keys)
class TestDatabaseCacheClearer(unittest2.TestCase):
def setUp(self):
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.testbed.init_memcache_stub()
ndb.get_context().clear_cache() # Prevent data from leaking between tests
self.testbed.init_taskqueue_stub(root_path=".")
self.eventteam_2015casj_frc254 = EventTeam(
id='2015casj_frc254',
event=ndb.Key(Event, '2015casj'),
team=ndb.Key(Team, 'frc254'),
year=2015,
)
self.eventteam_2015cama_frc604 = EventTeam(
id='2015cama_frc604',
event=ndb.Key(Event, '2015cama'),
team=ndb.Key(Team, 'frc604'),
year=2015,
)
self.eventteam_2010cama_frc604 = EventTeam(
id='2010cama_frc604',
event=ndb.Key(Event, '2010cama'),
team=ndb.Key(Team, 'frc604'),
year=2010,
)
self.eventteam_2016necmp_frc125 = EventTeam(
id='2016necmp_frc125',
event=ndb.Key(Event, '2016necmp'),
team=ndb.Key(Team, 'frc125'),
year=2016,
)
self.eventteam_2015casj_frc254.put()
self.eventteam_2015cama_frc604.put()
self.eventteam_2010cama_frc604.put()
self.eventteam_2016necmp_frc125.put()
self.districtteam_2015fim_frc254 = DistrictTeam(
id='2015fim_frc254',
district_key=ndb.Key(District, '2015fim'),
team=ndb.Key(Team, 'frc254'),
year=2015,
)
self.districtteam_2015mar_frc604 = DistrictTeam(
id='2015mar_frc604',
district_key=ndb.Key(District, '2015mar'),
team=ndb.Key(Team, 'frc604'),
year=2015,
)
self.districtteam_2016ne_frc604 = DistrictTeam(
id='2016ne_frc604',
district_key=ndb.Key(District, '2016ne'),
team=ndb.Key(Team, 'frc604'),
year=2016,
)
self.districtteam_2015fim_frc254.put()
self.districtteam_2015mar_frc604.put()
self.districtteam_2016ne_frc604.put()
self.district_2015ne = District(
id='2015ne',
year=2015,
abbreviation='ne',
)
self.district_2016chs = District(
id='2016chs',
year=2016,
abbreviation='chs',
)
self.district_2015ne.put()
self.district_2016chs.put()
self.event_2016necmp = Event(
id='2016necmp',
year=2016,
district_key=ndb.Key(District, '2016ne'),
event_short='necmp',
event_type_enum=EventType.DISTRICT_CMP,
)
self.event_2016necmp.put()
def tearDown(self):
self.testbed.deactivate()
def test_award_updated(self):
affected_refs = {
'event': {ndb.Key(Event, '2015casj'), ndb.Key(Event, '2015cama')},
'team_list': {ndb.Key(Team, 'frc254'), ndb.Key(Team, 'frc604')},
'year': {2014, 2015}
}
cache_keys = [q.cache_key for q in get_affected_queries.award_updated(affected_refs)]
self.assertEqual(len(cache_keys), 12)
self.assertTrue(EventAwardsQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventAwardsQuery('2015cama').cache_key in cache_keys)
self.assertTrue(TeamAwardsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamAwardsQuery('frc604').cache_key in cache_keys)
self.assertTrue(TeamYearAwardsQuery('frc254', 2014).cache_key in cache_keys)
self.assertTrue(TeamYearAwardsQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(TeamYearAwardsQuery('frc604', 2014).cache_key in cache_keys)
self.assertTrue(TeamYearAwardsQuery('frc604', 2015).cache_key in cache_keys)
self.assertTrue(TeamEventAwardsQuery('frc254', '2015casj').cache_key in cache_keys)
self.assertTrue(TeamEventAwardsQuery('frc254', '2015cama').cache_key in cache_keys)
self.assertTrue(TeamEventAwardsQuery('frc604', '2015casj').cache_key in cache_keys)
self.assertTrue(TeamEventAwardsQuery('frc604', '2015cama').cache_key in cache_keys)
def test_event_updated(self):
affected_refs = {
'key': {ndb.Key(Event, '2015casj'), ndb.Key(Event, '2015cama')},
'year': {2014, 2015},
'district_key': {ndb.Key(District, '2015fim'), ndb.Key(District, '2014mar')}
}
cache_keys = [q.cache_key for q in get_affected_queries.event_updated(affected_refs)]
self.assertEqual(len(cache_keys), 10)
self.assertTrue(EventQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventQuery('2015cama').cache_key in cache_keys)
self.assertTrue(EventListQuery(2014).cache_key in cache_keys)
self.assertTrue(EventListQuery(2015).cache_key in cache_keys)
self.assertTrue(DistrictEventsQuery('2015fim').cache_key in cache_keys)
self.assertTrue(DistrictEventsQuery('2014mar').cache_key in cache_keys)
self.assertTrue(TeamEventsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamEventsQuery('frc604').cache_key in cache_keys)
self.assertTrue(TeamYearEventsQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(TeamYearEventsQuery('frc604', 2015).cache_key in cache_keys)
def test_event_details_updated(self):
affected_refs = {
'key': {ndb.Key(EventDetails, '2015casj'), ndb.Key(EventDetails, '2015cama')},
}
cache_keys = [q.cache_key for q in get_affected_queries.event_details_updated(affected_refs)]
self.assertEqual(len(cache_keys), 2)
self.assertTrue(EventDetailsQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventDetailsQuery('2015cama').cache_key in cache_keys)
def test_match_updated(self):
affected_refs = {
'key': {ndb.Key(Match, '2015casj_qm1'), ndb.Key(Match, '2015casj_qm2')},
'event': {ndb.Key(Event, '2015casj'), ndb.Key(Event, '2015cama')},
'team_keys': {ndb.Key(Team, 'frc254'), ndb.Key(Team, 'frc604')},
'year': {2014, 2015},
}
cache_keys = [q.cache_key for q in get_affected_queries.match_updated(affected_refs)]
self.assertEqual(len(cache_keys), 12)
self.assertTrue(MatchQuery('2015casj_qm1').cache_key in cache_keys)
self.assertTrue(MatchQuery('2015casj_qm2').cache_key in cache_keys)
self.assertTrue(EventMatchesQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventMatchesQuery('2015cama').cache_key in cache_keys)
self.assertTrue(TeamEventMatchesQuery('frc254', '2015casj').cache_key in cache_keys)
self.assertTrue(TeamEventMatchesQuery('frc254', '2015cama').cache_key in cache_keys)
self.assertTrue(TeamEventMatchesQuery('frc604', '2015casj').cache_key in cache_keys)
self.assertTrue(TeamEventMatchesQuery('frc604', '2015cama').cache_key in cache_keys)
self.assertTrue(TeamYearMatchesQuery('frc254', 2014).cache_key in cache_keys)
self.assertTrue(TeamYearMatchesQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(TeamYearMatchesQuery('frc604', 2014).cache_key in cache_keys)
self.assertTrue(TeamYearMatchesQuery('frc604', 2015).cache_key in cache_keys)
def test_media_updated_team(self):
affected_refs = {
'references': {ndb.Key(Team, 'frc254'), ndb.Key(Team, 'frc604')},
'year': {2014, 2015},
}
cache_keys = [q.cache_key for q in get_affected_queries.media_updated(affected_refs)]
self.assertEqual(len(cache_keys), 10)
self.assertTrue(TeamYearMediaQuery('frc254', 2014).cache_key in cache_keys)
self.assertTrue(TeamYearMediaQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(TeamSocialMediaQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamYearMediaQuery('frc604', 2014).cache_key in cache_keys)
self.assertTrue(TeamYearMediaQuery('frc604', 2015).cache_key in cache_keys)
self.assertTrue(TeamSocialMediaQuery('frc604').cache_key in cache_keys)
self.assertTrue(EventTeamsMediasQuery('2015cama').cache_key in cache_keys)
self.assertTrue(EventTeamsMediasQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventTeamsPreferredMediasQuery('2015cama').cache_key in cache_keys)
self.assertTrue(EventTeamsPreferredMediasQuery('2015casj').cache_key in cache_keys)
def test_media_updated_event(self):
affected_refs = {
'references': {ndb.Key(Event, '2016necmp')},
'year': {2016},
}
cache_keys = [q.cache_key for q in get_affected_queries.media_updated(affected_refs)]
self.assertEqual(len(cache_keys), 1)
self.assertTrue(EventMediasQuery('2016necmp').cache_key in cache_keys)
def test_robot_updated(self):
affected_refs = {
'team': {ndb.Key(Team, 'frc254'), ndb.Key(Team, 'frc604')},
}
cache_keys = [q.cache_key for q in get_affected_queries.robot_updated(affected_refs)]
self.assertEqual(len(cache_keys), 2)
self.assertTrue(TeamRobotsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamRobotsQuery('frc604').cache_key in cache_keys)
def test_team_updated(self):
affected_refs = {
'key': {ndb.Key(Team, 'frc254'), ndb.Key(Team, 'frc604')},
}
cache_keys = [q.cache_key for q in get_affected_queries.team_updated(affected_refs)]
self.assertEqual(len(cache_keys), 13)
self.assertTrue(TeamQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamQuery('frc604').cache_key in cache_keys)
self.assertTrue(TeamListQuery(0).cache_key in cache_keys)
self.assertTrue(TeamListQuery(1).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2015, 0).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2015, 1).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2010, 1).cache_key in cache_keys)
self.assertTrue(DistrictTeamsQuery('2015fim').cache_key in cache_keys)
self.assertTrue(DistrictTeamsQuery('2015mar').cache_key in cache_keys)
self.assertTrue(DistrictTeamsQuery('2016ne').cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2015cama').cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2010cama').cache_key in cache_keys)
def test_eventteam_updated(self):
affected_refs = {
'event': {ndb.Key(Event, '2015casj'), ndb.Key(Event, '2015cama')},
'team': {ndb.Key(Team, 'frc254'), ndb.Key(Team, 'frc604')},
'year': {2014, 2015}
}
cache_keys = [q.cache_key for q in get_affected_queries.eventteam_updated(affected_refs)]
self.assertEqual(len(cache_keys), 18)
self.assertTrue(TeamEventsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamEventsQuery('frc604').cache_key in cache_keys)
self.assertTrue(TeamParticipationQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamParticipationQuery('frc604').cache_key in cache_keys)
self.assertTrue(TeamYearEventsQuery('frc254', 2014).cache_key in cache_keys)
self.assertTrue(TeamYearEventsQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(TeamYearEventsQuery('frc604', 2014).cache_key in cache_keys)
self.assertTrue(TeamYearEventsQuery('frc604', 2015).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2014, 0).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2014, 1).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2015, 0).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2015, 1).cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2015cama').cache_key in cache_keys)
self.assertTrue(EventTeamsMediasQuery('2015cama').cache_key in cache_keys)
self.assertTrue(EventTeamsMediasQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventTeamsPreferredMediasQuery('2015cama').cache_key in cache_keys)
self.assertTrue(EventTeamsPreferredMediasQuery('2015casj').cache_key in cache_keys)
def test_districtteam_updated(self):
affected_refs = {
'district_key': {ndb.Key(District, '2015fim'), ndb.Key(District, '2015mar')},
'team': {ndb.Key(Team, 'frc254'), ndb.Key(Team, 'frc604')}
}
cache_keys = [q.cache_key for q in get_affected_queries.districtteam_updated(affected_refs)]
self.assertEqual(len(cache_keys), 4)
self.assertTrue(DistrictTeamsQuery('2015fim').cache_key in cache_keys)
self.assertTrue(DistrictTeamsQuery('2015mar').cache_key in cache_keys)
self.assertTrue(TeamDistrictsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamDistrictsQuery('frc604').cache_key in cache_keys)
def test_district_updated(self):
affected_refs = {
'key': {ndb.Key(District, '2016ne')},
'year': {2015, 2016},
'abbreviation': {'ne', 'chs'}
}
cache_keys = [q.cache_key for q in get_affected_queries.district_updated(affected_refs)]
self.assertEqual(len(cache_keys), 11)
self.assertTrue(DistrictsInYearQuery(2015).cache_key in cache_keys)
self.assertTrue(DistrictsInYearQuery(2016).cache_key in cache_keys)
self.assertTrue(DistrictHistoryQuery('ne').cache_key in cache_keys)
self.assertTrue(DistrictHistoryQuery('chs').cache_key in cache_keys)
self.assertTrue(DistrictQuery('2016ne').cache_key in cache_keys)
self.assertTrue(TeamDistrictsQuery('frc604').cache_key in cache_keys)
# Necessary because APIv3 Event models include the District model
self.assertTrue(EventQuery('2016necmp').cache_key in cache_keys)
self.assertTrue(EventListQuery(2016).cache_key in cache_keys)
self.assertTrue(DistrictEventsQuery('2016ne').cache_key in cache_keys)
self.assertTrue(TeamEventsQuery('frc125').cache_key in cache_keys)
self.assertTrue(TeamYearEventsQuery('frc125', 2016).cache_key in cache_keys)
class TestDatabaseCacheClearer(unittest2.TestCase):
def setUp(self):
self.testbed = testbed.Testbed()
self.testbed.activate()
self.testbed.init_datastore_v3_stub()
self.testbed.init_memcache_stub()
ndb.get_context().clear_cache(
) # Prevent data from leaking between tests
self.testbed.init_taskqueue_stub(root_path=".")
self.eventteam_2015casj_frc254 = EventTeam(
id='2015casj_frc254',
event=ndb.Key(Event, '2015casj'),
team=ndb.Key(Team, 'frc254'),
year=2015,
)
self.eventteam_2015cama_frc604 = EventTeam(
id='2015cama_frc604',
event=ndb.Key(Event, '2015cama'),
team=ndb.Key(Team, 'frc604'),
year=2015,
)
self.eventteam_2010cama_frc604 = EventTeam(
id='2010cama_frc604',
event=ndb.Key(Event, '2010cama'),
team=ndb.Key(Team, 'frc604'),
year=2010,
)
self.eventteam_2016necmp_frc125 = EventTeam(
id='2016necmp_frc125',
event=ndb.Key(Event, '2016necmp'),
team=ndb.Key(Team, 'frc125'),
year=2016,
)
self.eventteam_2015casj_frc254.put()
self.eventteam_2015cama_frc604.put()
self.eventteam_2010cama_frc604.put()
self.eventteam_2016necmp_frc125.put()
self.districtteam_2015fim_frc254 = DistrictTeam(
id='2015fim_frc254',
district_key=ndb.Key(District, '2015fim'),
team=ndb.Key(Team, 'frc254'),
year=2015,
)
self.districtteam_2015mar_frc604 = DistrictTeam(
id='2015mar_frc604',
district_key=ndb.Key(District, '2015mar'),
team=ndb.Key(Team, 'frc604'),
year=2015,
)
self.districtteam_2016ne_frc604 = DistrictTeam(
id='2016ne_frc604',
district_key=ndb.Key(District, '2016ne'),
team=ndb.Key(Team, 'frc604'),
year=2016,
)
self.districtteam_2015fim_frc254.put()
self.districtteam_2015mar_frc604.put()
self.districtteam_2016ne_frc604.put()
self.district_2015ne = District(
id='2015ne',
year=2015,
abbreviation='ne',
)
self.district_2016chs = District(
id='2016chs',
year=2016,
abbreviation='chs',
)
self.district_2015ne.put()
self.district_2016chs.put()
self.event_2016necmp = Event(
id='2016necmp',
year=2016,
district_key=ndb.Key(District, '2016ne'),
event_short='necmp',
event_type_enum=EventType.DISTRICT_CMP,
)
self.event_2016necmp.put()
def tearDown(self):
self.testbed.deactivate()
def test_award_updated(self):
affected_refs = {
'event': {ndb.Key(Event, '2015casj'),
ndb.Key(Event, '2015cama')},
'team_list': {ndb.Key(Team, 'frc254'),
ndb.Key(Team, 'frc604')},
'year': {2014, 2015}
}
cache_keys = [
q.cache_key
for q in get_affected_queries.award_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 12)
self.assertTrue(EventAwardsQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventAwardsQuery('2015cama').cache_key in cache_keys)
self.assertTrue(TeamAwardsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamAwardsQuery('frc604').cache_key in cache_keys)
self.assertTrue(
TeamYearAwardsQuery('frc254', 2014).cache_key in cache_keys)
self.assertTrue(
TeamYearAwardsQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(
TeamYearAwardsQuery('frc604', 2014).cache_key in cache_keys)
self.assertTrue(
TeamYearAwardsQuery('frc604', 2015).cache_key in cache_keys)
self.assertTrue(
TeamEventAwardsQuery('frc254', '2015casj').cache_key in cache_keys)
self.assertTrue(
TeamEventAwardsQuery('frc254', '2015cama').cache_key in cache_keys)
self.assertTrue(
TeamEventAwardsQuery('frc604', '2015casj').cache_key in cache_keys)
self.assertTrue(
TeamEventAwardsQuery('frc604', '2015cama').cache_key in cache_keys)
def test_event_updated(self):
affected_refs = {
'key': {ndb.Key(Event, '2015casj'),
ndb.Key(Event, '2015cama')},
'year': {2014, 2015},
'district_key':
{ndb.Key(District, '2015fim'),
ndb.Key(District, '2014mar')}
}
cache_keys = [
q.cache_key
for q in get_affected_queries.event_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 10)
self.assertTrue(EventQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventQuery('2015cama').cache_key in cache_keys)
self.assertTrue(EventListQuery(2014).cache_key in cache_keys)
self.assertTrue(EventListQuery(2015).cache_key in cache_keys)
self.assertTrue(DistrictEventsQuery('2015fim').cache_key in cache_keys)
self.assertTrue(DistrictEventsQuery('2014mar').cache_key in cache_keys)
self.assertTrue(TeamEventsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamEventsQuery('frc604').cache_key in cache_keys)
self.assertTrue(
TeamYearEventsQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(
TeamYearEventsQuery('frc604', 2015).cache_key in cache_keys)
def test_event_details_updated(self):
affected_refs = {
'key': {
ndb.Key(EventDetails, '2015casj'),
ndb.Key(EventDetails, '2015cama')
},
}
cache_keys = [
q.cache_key
for q in get_affected_queries.event_details_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 2)
self.assertTrue(EventDetailsQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventDetailsQuery('2015cama').cache_key in cache_keys)
def test_match_updated(self):
affected_refs = {
'key':
{ndb.Key(Match, '2015casj_qm1'),
ndb.Key(Match, '2015casj_qm2')},
'event': {ndb.Key(Event, '2015casj'),
ndb.Key(Event, '2015cama')},
'team_keys': {ndb.Key(Team, 'frc254'),
ndb.Key(Team, 'frc604')},
'year': {2014, 2015},
}
cache_keys = [
q.cache_key
for q in get_affected_queries.match_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 12)
self.assertTrue(MatchQuery('2015casj_qm1').cache_key in cache_keys)
self.assertTrue(MatchQuery('2015casj_qm2').cache_key in cache_keys)
self.assertTrue(EventMatchesQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventMatchesQuery('2015cama').cache_key in cache_keys)
self.assertTrue(
TeamEventMatchesQuery('frc254', '2015casj').cache_key in
cache_keys)
self.assertTrue(
TeamEventMatchesQuery('frc254', '2015cama').cache_key in
cache_keys)
self.assertTrue(
TeamEventMatchesQuery('frc604', '2015casj').cache_key in
cache_keys)
self.assertTrue(
TeamEventMatchesQuery('frc604', '2015cama').cache_key in
cache_keys)
self.assertTrue(
TeamYearMatchesQuery('frc254', 2014).cache_key in cache_keys)
self.assertTrue(
TeamYearMatchesQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(
TeamYearMatchesQuery('frc604', 2014).cache_key in cache_keys)
self.assertTrue(
TeamYearMatchesQuery('frc604', 2015).cache_key in cache_keys)
def test_media_updated_team(self):
affected_refs = {
'references': {ndb.Key(Team, 'frc254'),
ndb.Key(Team, 'frc604')},
'year': {2014, 2015},
}
cache_keys = [
q.cache_key
for q in get_affected_queries.media_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 10)
self.assertTrue(
TeamYearMediaQuery('frc254', 2014).cache_key in cache_keys)
self.assertTrue(
TeamYearMediaQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(TeamSocialMediaQuery('frc254').cache_key in cache_keys)
self.assertTrue(
TeamYearMediaQuery('frc604', 2014).cache_key in cache_keys)
self.assertTrue(
TeamYearMediaQuery('frc604', 2015).cache_key in cache_keys)
self.assertTrue(TeamSocialMediaQuery('frc604').cache_key in cache_keys)
self.assertTrue(
EventTeamsMediasQuery('2015cama').cache_key in cache_keys)
self.assertTrue(
EventTeamsMediasQuery('2015casj').cache_key in cache_keys)
self.assertTrue(
EventTeamsPreferredMediasQuery('2015cama').cache_key in cache_keys)
self.assertTrue(
EventTeamsPreferredMediasQuery('2015casj').cache_key in cache_keys)
def test_media_updated_event(self):
affected_refs = {
'references': {ndb.Key(Event, '2016necmp')},
'year': {2016},
}
cache_keys = [
q.cache_key
for q in get_affected_queries.media_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 1)
self.assertTrue(EventMediasQuery('2016necmp').cache_key in cache_keys)
def test_robot_updated(self):
affected_refs = {
'team': {ndb.Key(Team, 'frc254'),
ndb.Key(Team, 'frc604')},
}
cache_keys = [
q.cache_key
for q in get_affected_queries.robot_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 2)
self.assertTrue(TeamRobotsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamRobotsQuery('frc604').cache_key in cache_keys)
def test_team_updated(self):
affected_refs = {
'key': {ndb.Key(Team, 'frc254'),
ndb.Key(Team, 'frc604')},
}
cache_keys = [
q.cache_key
for q in get_affected_queries.team_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 13)
self.assertTrue(TeamQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamQuery('frc604').cache_key in cache_keys)
self.assertTrue(TeamListQuery(0).cache_key in cache_keys)
self.assertTrue(TeamListQuery(1).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2015, 0).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2015, 1).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2010, 1).cache_key in cache_keys)
self.assertTrue(DistrictTeamsQuery('2015fim').cache_key in cache_keys)
self.assertTrue(DistrictTeamsQuery('2015mar').cache_key in cache_keys)
self.assertTrue(DistrictTeamsQuery('2016ne').cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2015cama').cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2010cama').cache_key in cache_keys)
def test_eventteam_updated(self):
affected_refs = {
'event': {ndb.Key(Event, '2015casj'),
ndb.Key(Event, '2015cama')},
'team': {ndb.Key(Team, 'frc254'),
ndb.Key(Team, 'frc604')},
'year': {2014, 2015}
}
cache_keys = [
q.cache_key
for q in get_affected_queries.eventteam_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 18)
self.assertTrue(TeamEventsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamEventsQuery('frc604').cache_key in cache_keys)
self.assertTrue(
TeamParticipationQuery('frc254').cache_key in cache_keys)
self.assertTrue(
TeamParticipationQuery('frc604').cache_key in cache_keys)
self.assertTrue(
TeamYearEventsQuery('frc254', 2014).cache_key in cache_keys)
self.assertTrue(
TeamYearEventsQuery('frc254', 2015).cache_key in cache_keys)
self.assertTrue(
TeamYearEventsQuery('frc604', 2014).cache_key in cache_keys)
self.assertTrue(
TeamYearEventsQuery('frc604', 2015).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2014, 0).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2014, 1).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2015, 0).cache_key in cache_keys)
self.assertTrue(TeamListYearQuery(2015, 1).cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2015casj').cache_key in cache_keys)
self.assertTrue(EventTeamsQuery('2015cama').cache_key in cache_keys)
self.assertTrue(
EventTeamsMediasQuery('2015cama').cache_key in cache_keys)
self.assertTrue(
EventTeamsMediasQuery('2015casj').cache_key in cache_keys)
self.assertTrue(
EventTeamsPreferredMediasQuery('2015cama').cache_key in cache_keys)
self.assertTrue(
EventTeamsPreferredMediasQuery('2015casj').cache_key in cache_keys)
def test_districtteam_updated(self):
affected_refs = {
'district_key':
{ndb.Key(District, '2015fim'),
ndb.Key(District, '2015mar')},
'team': {ndb.Key(Team, 'frc254'),
ndb.Key(Team, 'frc604')}
}
cache_keys = [
q.cache_key
for q in get_affected_queries.districtteam_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 4)
self.assertTrue(DistrictTeamsQuery('2015fim').cache_key in cache_keys)
self.assertTrue(DistrictTeamsQuery('2015mar').cache_key in cache_keys)
self.assertTrue(TeamDistrictsQuery('frc254').cache_key in cache_keys)
self.assertTrue(TeamDistrictsQuery('frc604').cache_key in cache_keys)
def test_district_updated(self):
affected_refs = {
'key': {ndb.Key(District, '2016ne')},
'year': {2015, 2016},
'abbreviation': {'ne', 'chs'}
}
cache_keys = [
q.cache_key
for q in get_affected_queries.district_updated(affected_refs)
]
self.assertEqual(len(cache_keys), 11)
self.assertTrue(DistrictsInYearQuery(2015).cache_key in cache_keys)
self.assertTrue(DistrictsInYearQuery(2016).cache_key in cache_keys)
self.assertTrue(DistrictHistoryQuery('ne').cache_key in cache_keys)
self.assertTrue(DistrictHistoryQuery('chs').cache_key in cache_keys)
self.assertTrue(DistrictQuery('2016ne').cache_key in cache_keys)
self.assertTrue(TeamDistrictsQuery('frc604').cache_key in cache_keys)
# Necessary because APIv3 Event models include the District model
self.assertTrue(EventQuery('2016necmp').cache_key in cache_keys)
self.assertTrue(EventListQuery(2016).cache_key in cache_keys)
self.assertTrue(DistrictEventsQuery('2016ne').cache_key in cache_keys)
self.assertTrue(TeamEventsQuery('frc125').cache_key in cache_keys)
self.assertTrue(
TeamYearEventsQuery('frc125', 2016).cache_key in cache_keys)
