test_twitter_neuron.py 2.1 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. import unittest
  2. from core.NeuronModule import InvalidParameterException
  3. from neurons.twitter.twitter import Twitter
  4. class TestTwitter(unittest.TestCase):
  5. def setUp(self):
  6. self.consumer_key="kalliokey"
  7. self.consumer_secret = "kalliosecret"
  8. self.access_token_key = "kalliotokenkey"
  9. self.access_token_secret = "kalliotokensecret"
  10. self.tweet = "kalliotweet"
  11. def testParameters(self):
  12. def run_test(parameters_to_test):
  13. with self.assertRaises(InvalidParameterException):
  14. Twitter(**parameters_to_test)
  15. # empty
  16. parameters = dict()
  17. run_test(parameters)
  18. # missing tweet
  19. parameters = {
  20. "consumer_key": self.consumer_key,
  21. "consumer_secret": self.consumer_secret,
  22. "access_token_key": self.access_token_key,
  23. "access_token_secret": self.access_token_secret
  24. }
  25. run_test(parameters)
  26. # missing consumer_key
  27. parameters = {
  28. "consumer_secret": self.consumer_secret,
  29. "access_token_key": self.access_token_key,
  30. "access_token_secret": self.access_token_secret,
  31. "tweet": self.tweet
  32. }
  33. run_test(parameters)
  34. # missing consumer_secret
  35. parameters = {
  36. "consumer_key": self.consumer_key,
  37. "access_token_key": self.access_token_key,
  38. "access_token_secret": self.access_token_secret,
  39. "tweet": self.tweet
  40. }
  41. run_test(parameters)
  42. # missing access_token_key
  43. parameters = {
  44. "consumer_key": self.consumer_key,
  45. "consumer_secret": self.consumer_secret,
  46. "access_token_secret": self.access_token_secret,
  47. "tweet": self.tweet
  48. }
  49. run_test(parameters)
  50. # missing access_token_secret
  51. parameters = {
  52. "consumer_key": self.consumer_key,
  53. "consumer_secret": self.consumer_secret,
  54. "access_token_key": self.access_token_key,
  55. "tweet": self.tweet
  56. }
  57. run_test(parameters)