def test_store_data_should_use_uuid_in_key_name(self):
with mock.patch("mysqlapi.api.management.commands.s3.bucket"):
with mock.patch("boto.s3.key.Key") as Key:
key = Key.return_value
with mock.patch("uuid.uuid4"):
s3.store_data("data")
key.set_contents_from_string.assert_called_with(Key().name)
def test_store_data_should_store_last_key(self):
with mock.patch("mysqlapi.api.management.commands.s3.bucket"):
with mock.patch("uuid.uuid4") as uuid4:
uuid4.return_value = mock.Mock(hex="uuid")
key = s3.store_data("data")
self.assertEqual("uuid", key.name)
def test_store_data(self):
with mock.patch("mysqlapi.api.management.commands.s3.bucket"):
with mock.patch("boto.s3.key.Key") as Key:
key = Key.return_value
s3.store_data("data")
key.set_contents_from_string.assert_any_call("data")
def send_data(self, data):
s3.store_data(data)