Crystalnix /
omaha-server
| @@ 39-55 (lines=17) @@ | ||
| 36 | keys = conn.get_bucket('test').get_all_keys() |
|
| 37 | self.assertFalse(keys) |
|
| 38 | ||
| 39 | @moto.mock_s3 |
|
| 40 | def test_model_update(self): |
|
| 41 | conn = boto.connect_s3() |
|
| 42 | conn.create_bucket('test') |
|
| 43 | obj = self.factory() |
|
| 44 | new_obj = self.factory() |
|
| 45 | ||
| 46 | old_keys = conn.get_bucket('test').get_all_keys() |
|
| 47 | old_keys = [key.name for key in old_keys] |
|
| 48 | ||
| 49 | for field in self.file_fields: |
|
| 50 | self.assertIn(getattr(obj, field).name, old_keys) |
|
| 51 | setattr(obj, field, getattr(new_obj, field)) |
|
| 52 | obj.save() |
|
| 53 | ||
| 54 | new_keys = conn.get_bucket('test').get_all_keys() |
|
| 55 | self.assertFalse(set(old_keys) & set(new_keys)) |
|
| 56 | ||
| 57 | @moto.mock_s3 |
|
| 58 | def test_bulk_delete(self): |
|
| @@ 24-37 (lines=14) @@ | ||
| 21 | factory = None |
|
| 22 | file_fields = None |
|
| 23 | ||
| 24 | @moto.mock_s3 |
|
| 25 | def test_model_delete(self): |
|
| 26 | conn = boto.connect_s3() |
|
| 27 | conn.create_bucket('test') |
|
| 28 | obj = self.factory() |
|
| 29 | ||
| 30 | keys = conn.get_bucket('test').get_all_keys() |
|
| 31 | keys = [key.name for key in keys] |
|
| 32 | for field in self.file_fields: |
|
| 33 | self.assertIn(getattr(obj, field).name, keys) |
|
| 34 | ||
| 35 | obj.delete() |
|
| 36 | keys = conn.get_bucket('test').get_all_keys() |
|
| 37 | self.assertFalse(keys) |
|
| 38 | ||
| 39 | @moto.mock_s3 |
|
| 40 | def test_model_update(self): |
|
