def test_null_station_id(self):
username = '******'
password = '******'
user = User(username=username)
user.set_password(password)
user.kind = User.STATION
user.save()
station = Station()
station.name = 'no name'
station.user = user
station.max_sessions = 3
station.save()
# login
data = {
'username': username,
'password': password,
'api_key': settings.API_KEY
}
client = APIClient()
client.post(reverse('general:register'), data, format='json')
try:
session = Session.objects.get(user=user)
except:
session = None
data = {'token': session.token, 'api_key': settings.API_KEY}
response = self.client.post(self.url, data)
self.assertEqual(response.data, {
'status': 'error',
'reason': 'station_error'
})
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
def test_null_station_id(self):
username = '******'
password = '******'
user = User(username=username)
user.set_password(password)
user.kind = User.STATION
user.save()
station = Station()
station.name = 'no name'
station.user = user
station.max_sessions = 3
station.save()
# login
data = {'username': username, 'password': password,
'api_key': settings.API_KEY}
client = APIClient()
client.post(reverse('general:register'), data, format='json')
try:
session = Session.objects.get(user=user)
except:
session = None
data = {'token': session.token,
'api_key': settings.API_KEY}
response = self.client.post(self.url, data)
self.assertEqual(response.data, {'status': 'error', 'reason': 'station_error'})
self.assertEqual(response.status_code, status.HTTP_400_BAD_REQUEST)
