def setUp(self):
super(TestServiceUsagesApi, self).setUp()
self.resource = 'serviceusages'
self.user = User.objects.create_user(
'ralph',
'*****@*****.**',
'ralph'
)
date2datetime = lambda d: datetime.datetime.combine(
d, datetime.datetime.min.time()
)
self.date = datetime.date(2013, 10, 10)
self.datetime = date2datetime(self.date)
self.today = datetime.date.today()
self.today_datetime = date2datetime(self.today)
self.service = Service(name='Service1', symbol='s1')
self.service.save()
self.venture1 = Venture(
name='Venture1',
venture_id=1,
symbol='v1',
is_active=True,
)
self.venture1.save()
self.venture2 = Venture(
name='Venture2',
venture_id=2,
symbol='v2',
is_active=True,
)
self.venture2.save()
self.venture3 = Venture(
name='Venture3',
venture_id=3,
symbol='v3',
is_active=True,
)
self.venture3.save()
self.inactive_venture = Venture(
name='Venture4',
venture_id=4,
symbol='v4',
is_active=False,
)
self.inactive_venture.save()
self.usage_type1 = UsageType(name='UsageType1', symbol='ut1')
self.usage_type1.save()
self.usage_type2 = UsageType(name='UsageType2', symbol='ut2')
self.usage_type2.save()
self.api_key = self.create_apikey(
self.user.username,
self.user.api_key.key,
)
def setUp(self):
self.venture = get_or_create_venture()
self.venture2 = get_or_create_venture()
self.device = get_or_create_device(device_id=1)
self.usage_type = UsageType(
name='Usage1',
symbol='ut1',
)
self.usage_type.save()
self.today = date.today()
def setUp(self):
super(TestServiceUsagesApi, self).setUp()
self.resource = 'serviceusages'
self.user = User.objects.create_user('ralph', '*****@*****.**',
'ralph')
date2datetime = lambda d: datetime.datetime.combine(
d, datetime.datetime.min.time())
self.date = datetime.date(2013, 10, 10)
self.datetime = date2datetime(self.date)
self.today = datetime.date.today()
self.today_datetime = date2datetime(self.today)
self.service = Service(name='Service1', symbol='s1')
self.service.save()
self.venture1 = Venture(
name='Venture1',
venture_id=1,
symbol='v1',
is_active=True,
)
self.venture1.save()
self.venture2 = Venture(
name='Venture2',
venture_id=2,
symbol='v2',
is_active=True,
)
self.venture2.save()
self.venture3 = Venture(
name='Venture3',
venture_id=3,
symbol='v3',
is_active=True,
)
self.venture3.save()
self.inactive_venture = Venture(
name='Venture4',
venture_id=4,
symbol='v4',
is_active=False,
)
self.inactive_venture.save()
self.usage_type1 = UsageType(name='UsageType1', symbol='ut1')
self.usage_type1.save()
self.usage_type2 = UsageType(name='UsageType2', symbol='ut2')
self.usage_type2.save()
self.api_key = self.create_apikey(
self.user.username,
self.user.api_key.key,
)
class TestServiceUsagesApi(ResourceTestCase):
def setUp(self):
super(TestServiceUsagesApi, self).setUp()
self.resource = 'serviceusages'
self.user = User.objects.create_user('ralph', '*****@*****.**',
'ralph')
date2datetime = lambda d: datetime.datetime.combine(
d, datetime.datetime.min.time())
self.date = datetime.date(2013, 10, 10)
self.datetime = date2datetime(self.date)
self.today = datetime.date.today()
self.today_datetime = date2datetime(self.today)
self.service = Service(name='Service1', symbol='s1')
self.service.save()
self.venture1 = Venture(
name='Venture1',
venture_id=1,
symbol='v1',
is_active=True,
)
self.venture1.save()
self.venture2 = Venture(
name='Venture2',
venture_id=2,
symbol='v2',
is_active=True,
)
self.venture2.save()
self.venture3 = Venture(
name='Venture3',
venture_id=3,
symbol='v3',
is_active=True,
)
self.venture3.save()
self.inactive_venture = Venture(
name='Venture4',
venture_id=4,
symbol='v4',
is_active=False,
)
self.inactive_venture.save()
self.usage_type1 = UsageType(name='UsageType1', symbol='ut1')
self.usage_type1.save()
self.usage_type2 = UsageType(name='UsageType2', symbol='ut2')
self.usage_type2.save()
self.api_key = self.create_apikey(
self.user.username,
self.user.api_key.key,
)
def _get_sample_service_usages_object(self, overwrite=None):
service_usages = ServiceUsageObject(
date=self.date,
service=self.service.symbol,
venture_usages=[
VentureUsageObject(
venture=self.venture1.symbol,
usages=[
UsageObject(symbol=self.usage_type1.symbol, value=123),
UsageObject(symbol=self.usage_type2.symbol, value=1.2),
]),
VentureUsageObject(
venture=self.venture2.symbol,
usages=[
UsageObject(symbol=self.usage_type1.symbol, value=3.3),
UsageObject(symbol=self.usage_type2.symbol, value=44),
])
])
if overwrite is not None:
service_usages.overwrite = overwrite
return service_usages
def test_save_usages(self):
service_usages = self._get_sample_service_usages_object()
ServiceUsageResource.save_usages(service_usages)
self.assertEquals(DailyUsage.objects.count(), 4)
daily_usage_1 = DailyUsage.objects.order_by('id')[0]
self.assertEquals(daily_usage_1.pricing_venture, self.venture1)
self.assertEquals(daily_usage_1.date, self.datetime)
self.assertEquals(daily_usage_1.type, self.usage_type1)
self.assertEquals(daily_usage_1.value, 123)
def test_to_dict(self):
service_usages = self._get_sample_service_usages_object()
self.assertEquals(
service_usages.to_dict(), {
'service':
self.service.symbol,
'date':
self.date,
'overwrite':
'no',
'venture_usages': [{
'venture':
self.venture1.symbol,
'usages': [
{
'symbol': self.usage_type1.symbol,
'value': 123,
},
{
'symbol': self.usage_type2.symbol,
'value': 1.2,
},
]
}, {
'venture':
self.venture2.symbol,
'usages': [
{
'symbol': self.usage_type1.symbol,
'value': 3.3,
},
{
'symbol': self.usage_type2.symbol,
'value': 44,
},
]
}]
})
def test_api(self):
service_usages = self._get_sample_service_usages_object()
data = service_usages.to_dict()
resp = self.api_client.post('/scrooge/api/v0.9/{0}/'.format(
self.resource),
format='json',
authentication=self.api_key,
data=data)
self.assertEquals(resp.status_code, 201)
self.assertEquals(DailyUsage.objects.count(), 4)
daily_usage_1 = DailyUsage.objects.order_by('id')[0]
self.assertEquals(daily_usage_1.pricing_venture, self.venture1)
self.assertEquals(daily_usage_1.date, self.datetime)
self.assertEquals(daily_usage_1.type, self.usage_type1)
self.assertEquals(daily_usage_1.value, 123)
def test_api_invalid_service(self):
service_usages = self._get_sample_service_usages_object()
data = service_usages.to_dict()
service_symbol = 'invalid_service'
data['service'] = service_symbol
resp = self.api_client.post('/scrooge/api/v0.9/{0}/'.format(
self.resource),
format='json',
authentication=self.api_key,
data=data)
self.assertEquals(resp.status_code, 400)
self.assertEquals(resp.content,
'Invalid service symbol: {}'.format(service_symbol))
self.assertEquals(DailyUsage.objects.count(), 0)
def test_api_invalid_venture(self):
service_usages = self._get_sample_service_usages_object()
data = service_usages.to_dict()
venture_symbol = 'invalid_venture'
data['venture_usages'][1]['venture'] = venture_symbol
resp = self.api_client.post('/scrooge/api/v0.9/{0}/'.format(
self.resource),
format='json',
authentication=self.api_key,
data=data)
self.assertEquals(resp.status_code, 400)
self.assertEquals(
resp.content,
'Invalid or inactive venture (symbol: {})'.format(venture_symbol))
self.assertEquals(DailyUsage.objects.count(), 0)
def test_api_invalid_usage(self):
service_usages = self._get_sample_service_usages_object()
data = service_usages.to_dict()
usage_type_symbol = 'invalid_usage'
data['venture_usages'][1]['usages'][1]['symbol'] = usage_type_symbol
resp = self.api_client.post('/scrooge/api/v0.9/{0}/'.format(
self.resource),
format='json',
authentication=self.api_key,
data=data)
self.assertEquals(resp.status_code, 400)
self.assertEquals(
resp.content,
'Invalid usage type symbol: {}'.format(usage_type_symbol))
self.assertEquals(DailyUsage.objects.count(), 0)
def test_api_inactive_venture(self):
service_usages = self._get_sample_service_usages_object()
service_usages.venture_usages[0].venture = self.inactive_venture.symbol
data = service_usages.to_dict()
resp = self.api_client.post('/scrooge/api/v0.9/{0}/'.format(
self.resource),
format='json',
authentication=self.api_key,
data=data)
self.assertEquals(resp.status_code, 400)
self.assertEquals(
resp.content, 'Invalid or inactive venture (symbol: {})'.format(
self.inactive_venture.symbol, ))
self.assertEquals(DailyUsage.objects.count(), 0)
def _basic_call(self):
service_usages = self._get_sample_service_usages_object()
data = service_usages.to_dict()
resp = self.api_client.post('/scrooge/api/v0.9/{0}/'.format(
self.resource),
format='json',
authentication=self.api_key,
data=data)
self.assertEquals(resp.status_code, 201)
usages = DailyUsage.objects.filter(type=self.usage_type1).values_list(
'pricing_venture__symbol',
'value',
)
self.assertEquals(dict(usages), {
self.venture1.symbol: 123.0,
self.venture2.symbol: 3.3,
})
def test_overwrite_values_only(self):
self._basic_call()
service_usages = self._get_sample_service_usages_object(
overwrite='values_only')
service_usages.venture_usages[0].venture = self.venture3.symbol
data = service_usages.to_dict()
resp = self.api_client.post('/scrooge/api/v0.9/{0}/'.format(
self.resource),
format='json',
authentication=self.api_key,
data=data)
self.assertEquals(resp.status_code, 201)
usages = DailyUsage.objects.filter(type=self.usage_type1).values_list(
'pricing_venture__symbol',
'value',
)
self.assertEquals(
dict(usages), {
self.venture1.symbol: 123.0,
self.venture2.symbol: 3.3,
self.venture3.symbol: 123.0,
})
def test_overwrite_delete_all_previous(self):
self._basic_call()
service_usages = self._get_sample_service_usages_object(
overwrite='delete_all_previous')
service_usages.venture_usages[0].venture = self.venture3.symbol
data = service_usages.to_dict()
resp = self.api_client.post('/scrooge/api/v0.9/{0}/'.format(
self.resource),
format='json',
authentication=self.api_key,
data=data)
self.assertEquals(resp.status_code, 201)
usages = DailyUsage.objects.filter(type=self.usage_type1).values_list(
'pricing_venture__symbol',
'value',
)
self.assertEquals(dict(usages), {
self.venture2.symbol: 3.3,
self.venture3.symbol: 123.0,
})
def test_not_overwriting(self):
self._basic_call()
service_usages = self._get_sample_service_usages_object(overwrite='no')
service_usages.venture_usages[0].venture = self.venture3.symbol
data = service_usages.to_dict()
resp = self.api_client.post('/scrooge/api/v0.9/{0}/'.format(
self.resource),
format='json',
authentication=self.api_key,
data=data)
self.assertEquals(resp.status_code, 201)
usages = DailyUsage.objects.filter(type=self.usage_type1).values_list(
'pricing_venture__symbol',
'value',
)
self.assertEquals([a for a in usages], [(self.venture1.symbol, 123.0),
(self.venture2.symbol, 3.3),
(self.venture3.symbol, 123.0),
(self.venture2.symbol, 3.3)])
class TestServiceUsagesApi(ResourceTestCase):
def setUp(self):
super(TestServiceUsagesApi, self).setUp()
self.resource = 'serviceusages'
self.user = User.objects.create_user(
'ralph',
'*****@*****.**',
'ralph'
)
date2datetime = lambda d: datetime.datetime.combine(
d, datetime.datetime.min.time()
)
self.date = datetime.date(2013, 10, 10)
self.datetime = date2datetime(self.date)
self.today = datetime.date.today()
self.today_datetime = date2datetime(self.today)
self.service = Service(name='Service1', symbol='s1')
self.service.save()
self.venture1 = Venture(
name='Venture1',
venture_id=1,
symbol='v1',
is_active=True,
)
self.venture1.save()
self.venture2 = Venture(
name='Venture2',
venture_id=2,
symbol='v2',
is_active=True,
)
self.venture2.save()
self.inactive_venture = Venture(
name='Venture3',
venture_id=3,
symbol='v3',
is_active=False,
)
self.inactive_venture.save()
self.usage_type1 = UsageType(name='UsageType1', symbol='ut1')
self.usage_type1.save()
self.usage_type2 = UsageType(name='UsageType2', symbol='ut2')
self.usage_type2.save()
self.api_key = self.create_apikey(
self.user.username,
self.user.api_key.key,
)
def _get_sample_service_usages_object(self):
service_usages = ServiceUsageObject(
date=self.date,
service=self.service.symbol,
venture_usages=[
VentureUsageObject(
venture=self.venture1.symbol,
usages=[
UsageObject(symbol=self.usage_type1.symbol, value=123),
UsageObject(symbol=self.usage_type2.symbol, value=1.2),
]
),
VentureUsageObject(
venture=self.venture2.symbol,
usages=[
UsageObject(symbol=self.usage_type1.symbol, value=3.3),
UsageObject(symbol=self.usage_type2.symbol, value=44),
]
)
]
)
return service_usages
def test_save_usages(self):
service_usages = self._get_sample_service_usages_object()
ServiceUsageResource.save_usages(service_usages)
self.assertEquals(DailyUsage.objects.count(), 4)
daily_usage_1 = DailyUsage.objects.order_by('id')[0]
self.assertEquals(daily_usage_1.pricing_venture, self.venture1)
self.assertEquals(daily_usage_1.date, self.datetime)
self.assertEquals(daily_usage_1.type, self.usage_type1)
self.assertEquals(daily_usage_1.value, 123)
def test_to_dict(self):
service_usages = self._get_sample_service_usages_object()
self.assertEquals(service_usages.to_dict(), {
'service': self.service.symbol,
'date': self.date,
'venture_usages': [
{
'venture': self.venture1.symbol,
'usages': [
{
'symbol': self.usage_type1.symbol,
'value': 123,
},
{
'symbol': self.usage_type2.symbol,
'value': 1.2,
},
]
},
{
'venture': self.venture2.symbol,
'usages': [
{
'symbol': self.usage_type1.symbol,
'value': 3.3,
},
{
'symbol': self.usage_type2.symbol,
'value': 44,
},
]
}
]
})
def test_api(self):
service_usages = self._get_sample_service_usages_object()
data = service_usages.to_dict()
resp = self.api_client.post(
'/scrooge/api/v0.9/{0}/'.format(self.resource),
format='json',
authentication=self.api_key,
data=data
)
self.assertEquals(resp.status_code, 201)
self.assertEquals(DailyUsage.objects.count(), 4)
daily_usage_1 = DailyUsage.objects.order_by('id')[0]
self.assertEquals(daily_usage_1.pricing_venture, self.venture1)
self.assertEquals(daily_usage_1.date, self.datetime)
self.assertEquals(daily_usage_1.type, self.usage_type1)
self.assertEquals(daily_usage_1.value, 123)
def test_api_invalid_service(self):
service_usages = self._get_sample_service_usages_object()
data = service_usages.to_dict()
data['service'] = 'invalid_service'
resp = self.api_client.post(
'/scrooge/api/v0.9/{0}/'.format(self.resource),
format='json',
authentication=self.api_key,
data=data
)
self.assertEquals(resp.status_code, 400)
self.assertEquals(resp.content, 'Invalid service symbol')
self.assertEquals(DailyUsage.objects.count(), 0)
def test_api_invalid_venture(self):
service_usages = self._get_sample_service_usages_object()
data = service_usages.to_dict()
data['venture_usages'][1]['venture'] = 'invalid_venture'
resp = self.api_client.post(
'/scrooge/api/v0.9/{0}/'.format(self.resource),
format='json',
authentication=self.api_key,
data=data
)
self.assertEquals(resp.status_code, 400)
self.assertEquals(
resp.content,
'Invalid venture symbol or venture is inactive'
)
self.assertEquals(DailyUsage.objects.count(), 0)
def test_api_invalid_usage(self):
service_usages = self._get_sample_service_usages_object()
data = service_usages.to_dict()
data['venture_usages'][1]['usages'][1]['symbol'] = 'invalid_usage'
resp = self.api_client.post(
'/scrooge/api/v0.9/{0}/'.format(self.resource),
format='json',
authentication=self.api_key,
data=data
)
self.assertEquals(resp.status_code, 400)
self.assertEquals(resp.content, 'Invalid usage type symbol')
self.assertEquals(DailyUsage.objects.count(), 0)
def test_api_inactive_venture(self):
service_usages = self._get_sample_service_usages_object()
service_usages.venture_usages[0].venture = self.inactive_venture.symbol
data = service_usages.to_dict()
resp = self.api_client.post(
'/scrooge/api/v0.9/{0}/'.format(self.resource),
format='json',
authentication=self.api_key,
data=data
)
self.assertEquals(resp.status_code, 400)
self.assertEquals(
resp.content,
'Invalid venture symbol or venture is inactive'
)
self.assertEquals(DailyUsage.objects.count(), 0)
class TestSharesPlugin(TestCase):
def setUp(self):
self.venture = get_or_create_venture()
self.venture2 = get_or_create_venture()
self.device = get_or_create_device(device_id=1)
self.usage_type = UsageType(
name='Usage1',
symbol='ut1',
)
self.usage_type.save()
self.today = date.today()
def test_update_usage(self):
update_usage(self.device, self.venture, self.usage_type, self.today, 8)
self.assertEqual(DailyUsage.objects.count(), 1)
daily_usage = DailyUsage.objects.all()[0]
self.assertEqual(daily_usage.pricing_venture, self.venture)
self.assertEqual(daily_usage.pricing_device, self.device)
self.assertEqual(daily_usage.value, 8)
self.assertEqual(daily_usage.date.date(), self.today)
self.assertEqual(daily_usage.type, self.usage_type)
def test_update_usage_append(self):
update_usage(self.device, self.venture, self.usage_type, self.today, 8)
update_usage(self.device, self.venture, self.usage_type, self.today, 8)
self.assertEqual(DailyUsage.objects.count(), 1)
daily_usage = DailyUsage.objects.all()[0]
self.assertEqual(daily_usage.pricing_venture, self.venture)
self.assertEqual(daily_usage.pricing_device, self.device)
self.assertEqual(daily_usage.value, 16)
self.assertEqual(daily_usage.date.date(), self.today)
self.assertEqual(daily_usage.type, self.usage_type)
def test_update(self):
data = {
'mount_device_id': self.device.device_id,
'size': 10
}
get_or_create_dailydevice(
date=self.today,
device=self.device,
venture=self.venture,
)
result = update(data, self.today, self.usage_type, None)
self.assertTrue(result)
self.assertEqual(DailyUsage.objects.count(), 1)
daily_usage = DailyUsage.objects.all()[0]
self.assertEqual(daily_usage.pricing_venture, self.venture)
self.assertEqual(daily_usage.pricing_device, self.device)
self.assertEqual(daily_usage.value, 10)
self.assertEqual(daily_usage.date.date(), self.today)
self.assertEqual(daily_usage.type, self.usage_type)
def test_update_without_device_id(self):
data = {
'mount_device_id': None,
'size': 10
}
result = update(data, self.today, self.usage_type, None)
self.assertFalse(result)
def test_update_device_not_found(self):
data = {
'mount_device_id': self.device.device_id + 1,
'size': 10
}
result = update(data, self.today, self.usage_type, None)
self.assertFalse(result)
def test_update_dailydevice_not_found(self):
data = {
'mount_device_id': self.device.device_id,
'size': 10
}
result = update(data, self.today, self.usage_type, None)
self.assertFalse(result)
def test_update_device_not_found_unknown_venture(self):
data = {
'mount_device_id': self.device.device_id + 1,
'size': 10
}
result = update(data, self.today, self.usage_type, self.venture2)
self.assertTrue(result)
self.assertEqual(DailyUsage.objects.count(), 1)
daily_usage = DailyUsage.objects.all()[0]
self.assertEqual(daily_usage.pricing_venture, self.venture2)
self.assertIsNone(daily_usage.pricing_device)
self.assertEqual(daily_usage.value, 10)
self.assertEqual(daily_usage.date.date(), self.today)
self.assertEqual(daily_usage.type, self.usage_type)
def test_update_dailydevice_not_found_unknown_venture(self):
data = {
'mount_device_id': self.device.device_id,
'size': 10
}
result = update(data, self.today, self.usage_type, self.venture2)
self.assertTrue(result)
self.assertEqual(DailyUsage.objects.count(), 1)
daily_usage = DailyUsage.objects.all()[0]
self.assertEqual(daily_usage.pricing_venture, self.venture2)
self.assertEqual(daily_usage.pricing_device, self.device)
self.assertEqual(daily_usage.value, 10)
self.assertEqual(daily_usage.date.date(), self.today)
self.assertEqual(daily_usage.type, self.usage_type)
class TestServiceUsagesApi(ResourceTestCase):
def setUp(self):
super(TestServiceUsagesApi, self).setUp()
self.resource = 'serviceusages'
self.user = User.objects.create_user(
'ralph',
'*****@*****.**',
'ralph'
)
date2datetime = lambda d: datetime.datetime.combine(
d, datetime.datetime.min.time()
)
self.date = datetime.date(2013, 10, 10)
self.datetime = date2datetime(self.date)
self.today = datetime.date.today()
self.today_datetime = date2datetime(self.today)
self.service = Service(name='Service1', symbol='s1')
self.service.save()
self.venture1 = Venture(
name='Venture1',
venture_id=1,
symbol='v1',
is_active=True,
)
self.venture1.save()
self.venture2 = Venture(
name='Venture2',
venture_id=2,
symbol='v2',
is_active=True,
)
self.venture2.save()
self.venture3 = Venture(
name='Venture3',
venture_id=3,
symbol='v3',
is_active=True,
)
self.venture3.save()
self.inactive_venture = Venture(
name='Venture4',
venture_id=4,
symbol='v4',
is_active=False,
)
self.inactive_venture.save()
self.usage_type1 = UsageType(name='UsageType1', symbol='ut1')
self.usage_type1.save()
self.usage_type2 = UsageType(name='UsageType2', symbol='ut2')
self.usage_type2.save()
self.api_key = self.create_apikey(
self.user.username,
self.user.api_key.key,
)
def _get_sample_service_usages_object(self, overwrite=None):
service_usages = ServiceUsageObject(
date=self.date,
service=self.service.symbol,
venture_usages=[
VentureUsageObject(
venture=self.venture1.symbol,
usages=[
UsageObject(symbol=self.usage_type1.symbol, value=123),
UsageObject(symbol=self.usage_type2.symbol, value=1.2),
]
),
VentureUsageObject(
venture=self.venture2.symbol,
usages=[
UsageObject(symbol=self.usage_type1.symbol, value=3.3),
UsageObject(symbol=self.usage_type2.symbol, value=44),
]
)
]
)
if overwrite is not None:
service_usages.overwrite = overwrite
return service_usages
def test_save_usages(self):
service_usages = self._get_sample_service_usages_object()
ServiceUsageResource.save_usages(service_usages)
self.assertEquals(DailyUsage.objects.count(), 4)
daily_usage_1 = DailyUsage.objects.order_by('id')[0]
self.assertEquals(daily_usage_1.pricing_venture, self.venture1)
self.assertEquals(daily_usage_1.date, self.datetime)
self.assertEquals(daily_usage_1.type, self.usage_type1)
self.assertEquals(daily_usage_1.value, 123)
def test_to_dict(self):
service_usages = self._get_sample_service_usages_object()
self.assertEquals(service_usages.to_dict(), {
'service': self.service.symbol,
'date': self.date,
'overwrite': 'no',
'venture_usages': [
{
'venture': self.venture1.symbol,
'usages': [
{
'symbol': self.usage_type1.symbol,
'value': 123,
},
{
'symbol': self.usage_type2.symbol,
'value': 1.2,
},
]
},
{
'venture': self.venture2.symbol,
'usages': [
{
'symbol': self.usage_type1.symbol,
'value': 3.3,
},
{
'symbol': self.usage_type2.symbol,
'value': 44,
},
]
}
]
})
def test_api(self):
service_usages = self._get_sample_service_usages_object()
data = service_usages.to_dict()
resp = self.api_client.post(
'/pricing/api/v0.9/{0}/'.format(self.resource),
format='json',
authentication=self.api_key,
data=data
)
self.assertEquals(resp.status_code, 201)
self.assertEquals(DailyUsage.objects.count(), 4)
daily_usage_1 = DailyUsage.objects.order_by('id')[0]
self.assertEquals(daily_usage_1.pricing_venture, self.venture1)
self.assertEquals(daily_usage_1.date, self.datetime)
self.assertEquals(daily_usage_1.type, self.usage_type1)
self.assertEquals(daily_usage_1.value, 123)
def test_api_invalid_service(self):
service_usages = self._get_sample_service_usages_object()
data = service_usages.to_dict()
service_symbol = 'invalid_service'
data['service'] = service_symbol
resp = self.api_client.post(
'/pricing/api/v0.9/{0}/'.format(self.resource),
format='json',
authentication=self.api_key,
data=data
)
self.assertEquals(resp.status_code, 400)
self.assertEquals(
resp.content,
'Invalid service symbol: {}'.format(service_symbol)
)
self.assertEquals(DailyUsage.objects.count(), 0)
def test_api_invalid_venture(self):
service_usages = self._get_sample_service_usages_object()
data = service_usages.to_dict()
venture_symbol = 'invalid_venture'
data['venture_usages'][1]['venture'] = venture_symbol
resp = self.api_client.post(
'/pricing/api/v0.9/{0}/'.format(self.resource),
format='json',
authentication=self.api_key,
data=data
)
self.assertEquals(resp.status_code, 400)
self.assertEquals(
resp.content,
'Invalid or inactive venture (symbol: {})'.format(venture_symbol)
)
self.assertEquals(DailyUsage.objects.count(), 0)
def test_api_invalid_usage(self):
service_usages = self._get_sample_service_usages_object()
data = service_usages.to_dict()
usage_type_symbol = 'invalid_usage'
data['venture_usages'][1]['usages'][1]['symbol'] = usage_type_symbol
resp = self.api_client.post(
'/pricing/api/v0.9/{0}/'.format(self.resource),
format='json',
authentication=self.api_key,
data=data
)
self.assertEquals(resp.status_code, 400)
self.assertEquals(
resp.content,
'Invalid usage type symbol: {}'.format(usage_type_symbol)
)
self.assertEquals(DailyUsage.objects.count(), 0)
def test_api_inactive_venture(self):
service_usages = self._get_sample_service_usages_object()
service_usages.venture_usages[0].venture = self.inactive_venture.symbol
data = service_usages.to_dict()
resp = self.api_client.post(
'/pricing/api/v0.9/{0}/'.format(self.resource),
format='json',
authentication=self.api_key,
data=data
)
self.assertEquals(resp.status_code, 400)
self.assertEquals(
resp.content,
'Invalid or inactive venture (symbol: {})'.format(
self.inactive_venture.symbol,
)
)
self.assertEquals(DailyUsage.objects.count(), 0)
def _basic_call(self):
service_usages = self._get_sample_service_usages_object()
data = service_usages.to_dict()
resp = self.api_client.post(
'/pricing/api/v0.9/{0}/'.format(self.resource),
format='json',
authentication=self.api_key,
data=data
)
self.assertEquals(resp.status_code, 201)
usages = DailyUsage.objects.filter(type=self.usage_type1).values_list(
'pricing_venture__symbol',
'value',
)
self.assertEquals(dict(usages), {
self.venture1.symbol: 123.0,
self.venture2.symbol: 3.3,
})
def test_overwrite_values_only(self):
self._basic_call()
service_usages = self._get_sample_service_usages_object(
overwrite='values_only'
)
service_usages.venture_usages[0].venture = self.venture3.symbol
data = service_usages.to_dict()
resp = self.api_client.post(
'/pricing/api/v0.9/{0}/'.format(self.resource),
format='json',
authentication=self.api_key,
data=data
)
self.assertEquals(resp.status_code, 201)
usages = DailyUsage.objects.filter(type=self.usage_type1).values_list(
'pricing_venture__symbol',
'value',
)
self.assertEquals(dict(usages), {
self.venture1.symbol: 123.0,
self.venture2.symbol: 3.3,
self.venture3.symbol: 123.0,
})
def test_overwrite_delete_all_previous(self):
self._basic_call()
service_usages = self._get_sample_service_usages_object(
overwrite='delete_all_previous'
)
service_usages.venture_usages[0].venture = self.venture3.symbol
data = service_usages.to_dict()
resp = self.api_client.post(
'/pricing/api/v0.9/{0}/'.format(self.resource),
format='json',
authentication=self.api_key,
data=data
)
self.assertEquals(resp.status_code, 201)
usages = DailyUsage.objects.filter(type=self.usage_type1).values_list(
'pricing_venture__symbol',
'value',
)
self.assertEquals(dict(usages), {
self.venture2.symbol: 3.3,
self.venture3.symbol: 123.0,
})
def test_not_overwriting(self):
self._basic_call()
service_usages = self._get_sample_service_usages_object(
overwrite='no'
)
service_usages.venture_usages[0].venture = self.venture3.symbol
data = service_usages.to_dict()
resp = self.api_client.post(
'/pricing/api/v0.9/{0}/'.format(self.resource),
format='json',
authentication=self.api_key,
data=data
)
self.assertEquals(resp.status_code, 201)
usages = DailyUsage.objects.filter(type=self.usage_type1).values_list(
'pricing_venture__symbol',
'value',
)
self.assertEquals([a for a in usages], [
(self.venture1.symbol, 123.0),
(self.venture2.symbol, 3.3),
(self.venture3.symbol, 123.0),
(self.venture2.symbol, 3.3)]
)
