def test_store_retrieve_team(ddb): """Test to see if we can store and retrieve the same team.""" team = create_test_team('1', 'rocket2.0', 'Rocket 2.0') assert ddb.store(team) another_team = ddb.retrieve(Team, '1') assert team == another_team
def test_delete_team(self): team = create_test_team('1', 'rocket-2.0', 'Rocket 2.0') self.assertTrue(self.ddb.store(team)) self.assertEqual(len(self.ddb.query(Team)), 1) self.ddb.delete(Team, '1') self.assertEqual(len(self.ddb.query(Team)), 0)
def test_store_retrieve_team(self): """Test to see if we can store and retrieve the same team.""" team = create_test_team('1', 'rocket2.0', 'Rocket 2.0') self.assertTrue(self.ddb.store(team)) another_team = self.ddb.retrieve(Team, '1') self.assertEqual(team, another_team)
def test_delete_team(ddb): """Test to see if we can successfully delete a team.""" team = create_test_team('1', 'rocket-2.0', 'Rocket 2.0') ddb.store(team) assert len(ddb.query(Team)) == 1 ddb.delete(Team, '1') assert len(ddb.query(Team)) == 0
def test_query_or_teams(self): """Test edge cases like set inclusion in query_or.""" teams = [create_test_team(str(i), "glob", "name") for i in range(10)] member_to_check = list(teams[0].members)[0] param = [('members', member_to_check)] for team in teams: self.assertTrue(self.ddb.store(team)) queried_teams = self.ddb.query_or(Team, param) for team in teams: self.assertIn(team, queried_teams)
def test_query_team(ddb): """Test to see if we can store and query the same team.""" team = create_test_team('1', 'rocket2.0', 'Rocket 2.0') team2 = create_test_team('2', 'lame-o', 'Lame-O Team') team2.add_member('apple') ddb.store(team) ddb.store(team2) another_team = ddb.query(Team, [('display_name', 'Rocket 2.0')]) same_team = ddb.query(Team, [('platform', 'slack')]) multiple_queries = ddb.query(Team, [('display_name', 'Rocket 2.0'), ('platform', 'slack')]) member_team = ddb.query(Team, [('members', 'abc_123'), ('members', 'apple')]) all_team = ddb.query(Team) assert team == another_team[0] assert team == all_team[0] assert team == same_team[0] assert team == multiple_queries[0] assert team2 == member_team[0]
def test_query_team(self): """Test to see if we can store and query the same team.""" team = create_test_team('1', 'rocket2.0', 'Rocket 2.0') team2 = create_test_team('2', 'lame-o', 'Lame-O Team') team2.add_member('apple') self.assertTrue(self.ddb.store(team)) self.assertTrue(self.ddb.store(team2)) another_team = self.ddb.query(Team, [('displayname', 'Rocket 2.0')]) same_team = self.ddb.query(Team, [('platform', 'slack')]) multi_queries = self.ddb.query(Team, [('displayname', 'Rocket 2.0'), ('platform', 'slack')]) member_team = self.ddb.query(Team, [('members', 'abc_123'), ('members', 'apple')]) all_team = self.ddb.query(Team) self.assertEqual(team, another_team[0]) self.assertEqual(team, all_team[0]) self.assertEqual(team, same_team[0]) self.assertEqual(team, multi_queries[0]) self.assertEqual(team2, member_team[0])
def test_valid_team(): """Test the Team static class method is_valid().""" team = Team("1", "", "Brussel Sprouts") assert not Team.is_valid(team) team = create_test_team("1", "brussel-sprouts", "Brussel Sprouts") assert Team.is_valid(team)
def test_store_team(ddb): """Test storing team calls correct functions.""" dbf = DBFacade(ddb) test_team = create_test_team('1', 'brussel-sprouts', 'Brussel Sprouts') dbf.store(test_team) ddb.store.assert_called_with(test_team)