File tree Expand file tree Collapse file tree 1 file changed +16
-0
lines changed Expand file tree Collapse file tree 1 file changed +16
-0
lines changed Original file line number Diff line number Diff line change 1
1
import unittest
2
2
from loklak import Loklak
3
+ import os
3
4
4
5
5
6
class TestLoklak (unittest .TestCase ):
6
7
def setUp (self ):
7
8
self .loklak = Loklak ()
8
9
10
+ def tearDown (self ):
11
+ try :
12
+ os .remove (self .map_file )
13
+ except OSError as error :
14
+ print (error )
15
+
9
16
def test_hello (self ):
10
17
result = self .loklak .hello ()
11
18
self .assertEqual (result , {u'status' : u'ok' })
@@ -77,6 +84,15 @@ def test_user(self):
77
84
self .assertTrue ('name' in result ['user' ])
78
85
self .assertTrue ('screen_name' in result ['user' ])
79
86
87
+ def test_get_map (self ):
88
+ self .map_file = os .path .join (os .getcwd (), 'markdown.png' )
89
+ data = self .loklak .get_map (17.582729 , 79.118320 )
90
+ with open (self .map_file , 'wb' ) as f :
91
+ f .write (data )
92
+ with open (self .map_file , 'rb' ) as f :
93
+ file_contents = f .read ()
94
+ self .assertTrue (os .path .exists (self .map_file ))
95
+ self .assertEqual (data , file_contents )
80
96
81
97
if __name__ == '__main__' :
82
98
unittest .main ()
You can’t perform that action at this time.
0 commit comments